o
=Œ_íI ã @ s d Z ddlZz ddlm mZ W n ey# ddlm mZ Y nw ddlm Z m
Z
m Z m Z m
Z
ddlmZmZ ddlmZ ddlmZ G dd„ deƒZdS ) z CLass to edit XDG Menus é N)ÚMenuÚ MenuEntryÚLayoutÚ SeparatorÚXMLMenuBuilder)Úxdg_config_dirsÚ
xdg_data_dirs)Ú ParsingError)Ú setRootModec @ s\ e Zd ZdQdd„ZdQdd„Zdd„ ZdRd d
„ZdSd d „ZdTd
d„ZdTdd„Z dTdd„Z
dTdd„Z dTdd„Z dUdd„Z
dVdd„Zdd „ Zd d „ Zd d „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Z d-d.„ Zd/d0„ Zd1d2„ Zd3d4„ ZdWd6d7„Z d8d9„ Z d:d;„ Z dXd=d>„Z d?d@„ Z dAdB„ Z!dCdD„ Z"dTdEdF„Z#dTdGdH„Z$dIdJ„ Z%dKdL„ Z&dMdN„ Z'dOdP„ Z(dS )YÚ
MenuEditorNFc C s2 d | _ d | _d | _tƒ | _| |||¡ g | _d S ©N)ÚmenuÚfilenameÚtreer ÚparserÚparseÚ filenames©Úselfr
r Úroot© r ú0/usr/lib/python3/dist-packages/xdg/MenuEditor.pyÚ__init__ s
zMenuEditor.__init__c C sì |rt dƒ t|tƒr|| _n|r| j |¡| _n| j ¡ | _|r'| jj| _n|r-|| _ntj
t d dtj | jj¡d ¡| _z t
| j¡| _W n$ tya t
d| jj ¡}t
|¡| _Y n
tym td| jƒ‚w | | j¡ d S )NTr Úmenusé zÞ
zNot a valid .menu file)r
Ú
isinstancer r
r r ÚFilenamer ÚosÚpathÚjoinr ÚsplitÚetreer ÚIOErrorÚ
fromstringÚ ElementTreer Ú$_MenuEditor__remove_whitespace_nodesr r r r r $ s.
& ú ÿzMenuEditor.parsec C s | | j¡ | ¡ d S r )Ú_MenuEditor__saveEntriesr
Ú_MenuEditor__saveMenu©r r r r ÚsaveG s zMenuEditor.savec
C sD t | |d¡ƒ}
| |
||||||¡}
| ||
|| ¡ | j ¡ |
S )Nú.desktop)r Ú_MenuEditor__getFileNameÚ
editMenuEntryÚ_MenuEditor__addEntryr
Úsort) r ÚparentÚnameÚcommandÚ genericnameÚcommentÚiconÚterminalÚafterÚbeforeÚ menuentryr r r ÚcreateMenuEntryK s
zMenuEditor.createMenuEntryc C sX t ƒ }||_|jd |_|j|_|j|_| |||||¡}| ||||¡ | j ¡ |S )Nr ) r ÚParentÚDepthÚ
DefaultLayoutr ÚeditMenur- r
r. ) r r/ r0 r2 r3 r4 r6 r7 r
r r r Ú
createMenuU s
zMenuEditor.createMenuc C s& t |ƒ}| ||||¡ | j ¡ |S r )r r- r
r. )r r/ r6 r7 Ú separatorr r r ÚcreateSeparatore s
zMenuEditor.createSeparatorc C s. | ||||¡ | ||||¡ | j ¡ |S r ©Ú_MenuEditor__deleteEntryr- r
r. ©r r8 Ú oldparentÚ newparentr6 r7 r r r Ú
moveMenuEntryn ó
zMenuEditor.moveMenuEntryc C s‚ | ||||¡ | ||||¡ | | jj¡}| d¡| d¡kr:| |tj | d¡|j¡tj | d¡|j¡¡ | j
¡ |S )NT) rB r- Ú_MenuEditor__getXmlMenur
ÚNameÚgetPathÚ_MenuEditor__addXmlMover r r r. )r r
rD rE r6 r7 Ú root_menur r r ÚmoveMenuv s 2
zMenuEditor.moveMenuc C s. | ||||¡ | ||||¡ | j ¡ |S r rA )r r? r/ r6 r7 r r r Ú
moveSeparator‚ rG zMenuEditor.moveSeparatorc C s | ||||¡ | j ¡ |S r )r- r
r. rC r r r Ú
copyMenuEntryŠ s
zMenuEditor.copyMenuEntryc
C s> |j }
|r|
d¡s|
d|¡ |
jd|dd |r-|
d¡s%|
d|¡ |
jd|dd |rB|
d¡s:|
d|¡ |
jd|dd |rJ|
d|¡ |rR|
d|¡ |r[|
dd ¡ n|sc|
dd
¡ |du rn|
d d ¡ n
|d u rx|
d d
¡ | du rƒ|
d
d ¡ n
| d u r|
d
d
¡ | ¡ t|jƒdkr| j ¡ |S )NrI T©ÚlocaleÚCommentÚ GenericNameÚExecÚIconÚTerminalÚtrueÚfalseÚ NoDisplayFÚHiddenr )Ú DesktopEntryÚhasKeyÚsetÚupdateAttributesÚlenÚParentsr
r. ) r r8 r0 r2 r3 r1 r4 r5 Ú nodisplayÚhiddenÚ deskentryr r r r, ‘ sB
zMenuEditor.editMenuEntryc C sÈ t |jtƒr)|jjdkr)| | dd¡¡}| |d|jd ¡ |j |jd ¡ n4t |jtƒs]|s4|j}| |d¡
dd¡} |jsH|
dd¡|_| | dd¡¡}| |d| ¡ t| ƒ|_|jj }
|rv|
d¡sn|
d|¡ |
j
d|dd |r‹|
d¡sƒ|
d|¡ |
j
d|dd |r |
d ¡s˜|
d |¡ |
j
d |dd |r¨|
d
|¡ |du r³|
d d ¡ n
|d
u r½|
d d¡ |du rÈ|
dd ¡ n
|d
u rÒ|
dd¡ |j ¡ t |jtƒrâ| j ¡ |S )Nú
.directoryTÚ Directoryú/Ú rI rP rS rR rU rY rW FrX rZ )r re r r rH rJ Ú _MenuEditor__addXmlTextElementrI Ú
setAttributesr+ Úreplacer[ r\ r] r^ r: r r
r. ) r r
r0 r2 r3 r4 ra rb Úxml_menur rc r r r r= » sP
zMenuEditor.editMenuc C ó | j |dd d S ©NT)ra ©r, ©r r8 r r r Ú
hideMenuEntryî ó zMenuEditor.hideMenuEntryc C s | j |ddd d S )NF©ra rb rn ro r r r ÚunhideMenuEntryñ s zMenuEditor.unhideMenuEntryc C rl rm )r= ©r r
r r r ÚhideMenuô rq zMenuEditor.hideMenuc C sT | j |ddd | | dd¡d¡}| d¡}| d¡}|| D ]}| |¡ q d S )NFrr TÚDeletedÚ
NotDeleted)r= rH rJ ÚfindallÚremove)r r
rk ÚdeletedÚ not_deletedÚnoder r r Ú
unhideMenu÷ s
ÿzMenuEditor.unhideMenuc C sB | |¡dkr | |jj¡ |jD ]}| ||¡ q| j ¡ |S )NÚdelete)Ú getActionÚ_MenuEditor__deleteFiler[ r r` rB r
r. )r r8 r/ r r r ÚdeleteMenuEntryÿ s
zMenuEditor.deleteMenuEntryc C s| | |¡dkr<| |jj¡ g |j_|jD ]!}|j |¡}|j|j|< |j |¡}|j|j|< |jj |¡ q| j
¡ |S ©NÚrevert) r r€ r[ r ÚOriginalr` ÚEntriesÚindexÚ MenuEntriesÚappendr
r. )r r8 r/ r† r r r ÚrevertMenuEntry s
zMenuEditor.revertMenuEntryc C s` | |¡dkr.| |jjj¡ | |j|¡ | | dd¡¡}| |¡}|
|¡ | j ¡ |S )Nr~ T)
r r€ re r[ r rB r: rH rJ Ú _MenuEditor__get_parent_nodery r
r. )r r
rk r/ r r r Ú
deleteMenu s
zMenuEditor.deleteMenuc C s6 | |¡dkr| |jjj¡ |jj|_| j ¡ |S r‚ )r r€ re r[ r r„ r
r. rt r r r Ú
revertMenu s
zMenuEditor.revertMenuc C s | j |j|dd | j ¡ |S )NT)r6 )rB r: r
r. )r r? r r r ÚdeleteSeparator% s
zMenuEditor.deleteSeparatorc C sŒ t |tƒr-t |jtƒs
dS |j ¡ dkrdS |j ¡ dkr+t|jƒt|jƒ dkr+dS dS t |tƒrD| ¡ dkr:dS | ¡ dkrBdS dS dS )NÚnoneÚBothrƒ ÚUserr r~ )r r re r ÚgetTyper_ ÚSubmenusr‡ )r Úentryr r r r - s$
þ
ø zMenuEditor.getActionc C s^ |s| j }t|jtƒr|j ¡ |jddD ]}t|tƒr"| ¡ qt|tƒr,| |¡ qd S )NT)rb )r
r re r r) Ú
getEntriesr r& )r r
r“ r r r Ú
__saveEntriesB s
€üzMenuEditor.__saveEntriesc C s@ t j t j | j¡¡st t j | j¡¡ | jj| jdd d S )Nzutf-8)Úencoding)r r ÚisdirÚdirnamer Úmakedirsr Úwriter( r r r Ú
__saveMenuM s zMenuEditor.__saveMenuc C sˆ d} |dkr || }n
|d t |ƒ | }|dkr d}n|dkr#d}|| jvr?tj tj td ||¡¡s?| j |¡ |S |d7 }q)Nr r ú-r* Ú applicationsrd zdesktop-directories)Ústrr r r Úisfiler r rˆ )r r0 Ú extensionÚpostfixr Údirr r r Ú
__getFileNameR s"
ÿ þñzMenuEditor.__getFileNameTc C s¦ |s| j }d|v r| dd¡\}}n|}d}d }| d¡D ] }| d¡}|j|kr5|r3| |||¡}n|}|r9 nq |sQ|rQ| ||¡}|rO| |||¡}|S |}|S )Nrf r rg r rI )r r rx ÚfindÚtextrH Ú _MenuEditor__addXmlMenuElement)r r ÚcreateÚelementr0 Úfoundr| Ú name_noder r r Ú __getXmlMenug s.
ÿ þzMenuEditor.__getXmlMenuc C s" t d|¡}t d|¡}||_|S )Nr rI ©r! Ú
SubElementr¥ )r r¨ r0 Ú menu_noderª r r r Ú__addXmlMenuElement‡ s z MenuEditor.__addXmlMenuElementc C s t ||¡}||_|S r r¬ )r r¨ r0 r¥ r| r r r Ú__addXmlTextElement s z MenuEditor.__addXmlTextElementÚIncludec C sj | d¡}| d¡}|| }|D ]}|d jdkr%|d j|kr%| |¡ qt ||¡}| |d|¡ |S )Nr± ÚExcluder r )rx Útagr¥ ry r! r rh ) r r¨ r Útype_ÚincludesÚexcludesÚrulesÚruler| r r r Ú__addXmlFilename’ s
€
zMenuEditor.__addXmlFilenamec C s, t d|¡}| |d|¡ | |d|¡ |S )NÚMoveÚOldÚNew)r! r rh )r r¨ ÚoldÚnewr| r r r Ú __addXmlMove¨ s zMenuEditor.__addXmlMovec C s¶ | d¡D ]}| |¡ qt d|¡}|jD ]B}|d dkr%t d|¡}q|d dkr5| |d|d ¡}q|d dkrE| |d|d ¡}q|d dkrXt d|¡}|d |jd< q|S ) Nr r r r r ÚMenunameÚMergeÚtype)rx ry r! r Úorderrh Úattrib)r r¨ Úlayoutr| rà Úchildr r r Ú__addXmlLayout® s
€zMenuEditor.__addXmlLayoutc C s¶ t ƒ }g |_|j j|_|j j|_|j j|_|j j|_|j j|_|j ddg¡ |jD ]#}t |t
ƒr8| |j ¡ q*t |t
ƒrD| |j¡ q*t |tƒrM| ¡ q*|j ddg¡ ||_ |S )NrÁ r Úfiles)r rà Ú
show_emptyÚinlineÚ
inline_headerÚ inline_aliasÚ inline_limitrˆ r… r r Ú
parseMenunamerI r Ú
parseFilenameÚ
DesktopFileIDr ÚparseSeparator)r r/ rÅ r“ r r r Ú __addLayoutÁ s&
€zMenuEditor.__addLayoutc C sÌ |s|r |r|j |¡d }n|r|j |¡}|j ||¡ n|j |¡ | | dd¡¡}t|tƒrH|j |¡ |j |¡ |
||j d¡ n
t|t ƒrR|
|¡ |sV|rd| |¡ | ||j¡ d S d S )Nr Tr± )r… r† Úinsertrˆ rH rJ r r r‡ r` Ú_MenuEditor__addXmlFilenamerÐ r Ú
addSubmenuÚ_MenuEditor__addLayoutÚ_MenuEditor__addXmlLayoutr )r r/ r“ r6 r7 r† Ú
xml_parentr r r Ú
__addEntryØ s$
þzMenuEditor.__addEntryc C s |j |¡ | | dd¡¡}t|tƒr)|j |¡ |j |¡ | ||j d¡ n t|t
ƒr4|j |¡ |s8|rF| |¡ |
||j¡ d S d S )NTr² )r… ry rH rJ r r r` r‡ rÔ rÐ r r’ rÖ r× r )r r/ r“ r6 r7 rØ r r r Ú
__deleteEntryï s
þzMenuEditor.__deleteEntryc C sJ zt |¡ W n ty Y nw z | j |¡ W d S ty$ Y d S w r )r ry ÚOSErrorr Ú
ValueError)r r r r r Ú __deleteFileÿ s ÿ ÿzMenuEditor.__deleteFilec C sH |D ] }|j ¡ }|sd|_ |j ¡ }|sd|_t|ƒr!| |¡ qd S )Nrg )r¥ ÚstripÚtailr_ Ú$_MenuEditor__remove_whilespace_nodes)r r| rÆ r¥ rß r r r Ú__remove_whitespace_nodes s
€øz$MenuEditor.__remove_whitespace_nodesc C s& | ¡ D ] \}}||u r| S qd S r )Ú_MenuEditor__iter_parent)r r| r/ rÆ r r r Ú__get_parent_node s
ÿÿz MenuEditor.__get_parent_nodec c s* | j ¡ D ] }|D ]}||fV q
qd S r )r Ú getiterator)r r/ rÆ r r r Ú
__iter_parent s € ÿÿzMenuEditor.__iter_parent)NNF)NNNNNNN)NNNNN)NN)NNNNNNNN)NNNNNN)TN)r± ))Ú__name__Ú
__module__Ú __qualname__r r r) r9 r> r@ rF rM rN rO r, r= rp rs ru r} r r‰ r‹ rŒ r r r& r' r+ rH r¦ rh rÔ rK r× rÖ r- rB r€ r% rŠ râ r r r r r sP
#
*3
r )Ú__doc__r Úxml.etree.cElementTreer! Ú cElementTreeÚ ImportErrorÚxml.etree.ElementTreer$ Úxdg.Menur r r r r Úxdg.BaseDirectoryr r Úxdg.Exceptionsr Ú
xdg.Configr
Úobjectr r r r r Ú s ÿ