ucI d Z ddlZ ddlmc mZ n# e$ r ddlmc mZ Y nw xY wddlm Z m
Z
m Z m Z m
Z
ddlmZmZ ddlmZ ddlmZ G d de ZdS ) z CLass to edit XDG Menus N)Menu MenuEntryLayout SeparatorXMLMenuBuilder)xdg_config_dirs
xdg_data_dirs) ParsingError) setRootModec e Zd Zd+dZd+dZd Zd,dZd-dZd.dZd.d Z d.d
Z
d.d Z d.d Z d/d
Z
d0dZd Zd Zd Zd Zd Zd Zd Zd Zd Z d Zd Zd Zd Zd1d Z d Z d Z d2d!Z d" Z d# Z!d$ Z"d.d%Z#d.d&Z$d' Z%d( Z&d) Z'd* Z(dS )3
MenuEditorNFc d | _ d | _ d | _ t | _ | ||| g | _ d S N)menufilenametreer parserparse filenamesselfr r roots 0/usr/lib/python3/dist-packages/xdg/MenuEditor.py__init__zMenuEditor.__init__ sG
$&&
44((( c |rt d t |t r|| _ n@|r | j | | _ n | j | _ |r| j j | _ nh|r|| _ n^t j
t d dt j | j j d | _ t j | j | _ nh# t $ r= t j d| j j z }t j | | _ Y n"t$ $ r t% d| j w xY w| | j d S )NTr menus z
zNot a valid .menu file)r
isinstancer r r r Filenamer ospathjoinr splitetreer IOError
fromstring ElementTreer
$_MenuEditor__remove_whitespace_nodesr s r r zMenuEditor.parse$ s dD ! ! , DII
, ))$//DII ))++DI l I .DMM
l $DMM GLL );WbgmmTXT]TfFgFghiFj k kDM H DM22DII 0 0 0# % i
% D )$//DIII H H H 7 GG G H
&&ty11111s , D AE0 E0c b | | j | d S r )_MenuEditor__saveEntriesr _MenuEditor__saveMenur s r savezMenuEditor.saveG s. 49%%% r c
t | |d }
| |
|||||| }
| ||
|| | j |
S )N.desktop)r _MenuEditor__getFileName
editMenuEntry_MenuEditor__addEntryr sort) r parentnamecommand genericnamecommenticonterminalafterbefore menuentrys r createMenuEntryzMenuEditor.createMenuEntryK sp d 0 0z B BCC &&y$ WgW[]eff 5&999 r c t }||_ |j dz |_ |j |_ |j |_ | ||||| }| |||| | j |S )Nr ) r ParentDepth
DefaultLayoutr editMenur3 r r4 ) r r5 r6 r8 r9 r: r< r= r s r
createMenuzMenuEditor.createMenuU s} vv \A%
* # 1}}T4gtDD eV444 r c t | }| |||| | j |S r )r r3 r r4 )r r5 r< r= separators r createSeparatorzMenuEditor.createSeparatore s? f%% 5&999 r c | |||| | |||| | j |S r _MenuEditor__deleteEntryr3 r r4 r r> oldparent newparentr< r= s r
moveMenuEntryzMenuEditor.moveMenuEntryn sL 9i??? 9eV<<< r c 4 | |||| | |||| | | j j }| d | d k r| |t j | d |j t j | d |j | j
|S )NT) rK r3 _MenuEditor__getXmlMenur NamegetPath_MenuEditor__addXmlMover! r" r# r4 )r r rM rN r< r= root_menus r moveMenuzMenuEditor.moveMenuv s 9dE6::: 4777%%din55 T " "i&7&7&=&= = = i i6G6G6M6Mty)Y)Y[][b[g[ghqhyhyz~hh BF BK \L \L
M
M
M r c | |||| | |||| | j |S r rJ )r rG r5 r< r= s r
moveSeparatorzMenuEditor.moveSeparator sL 69eV<<< 5&999 r c h | |||| | j |S r )r3 r r4 rL s r
copyMenuEntryzMenuEditor.copyMenuEntry s2 9eV<<< r c
|j }
|rC|
d s|
d| |
d|d |rC|
d s|
d| |
d|d |rC|
d s|
d| |
d|d |r|
d| |r|
d| |r|
dd n|s|
dd
|du r|
d d n|d u r|
d d
| du r|
d
d n| d u r|
d
d
| t |j dk r| j |S )NrR TlocaleComment GenericNameExecIconTerminaltruefalse NoDisplayFHiddenr ) DesktopEntryhasKeysetupdateAttributeslenParentsr r4 ) r r> r6 r8 r9 r7 r: r; nodisplayhidden deskentrys r r2 zMenuEditor.editMenuEntry s * 5 ##F++
,
fd+++ MM&$tM 4 4 4 ; ##I..
2
i111 MM)WTM : : : C ##M22
:
m[999 MM-TM B B B + MM&' * * * ( MM&$ ' ' ' / MM*f - - - - / MM*g . . . MM+v . . . .
%
MM+w / / / T>> MM(F + + + +
u__ MM(G , , ,""$$$ y ! !A % % INN r c t |j t r{|j j dk rk| | dd }| |d|j dz |j |j dz nt |j t s|s|j }| |d
dd } |j s|
dd |_ | | dd }| |d| t | |_ |j j }
|rC|
d s|
d| |
d|d |rC|
d s|
d| |
d|d |rC|
d s|
d | |
d |d |r|
d
| |du r|
d d n|d
u r|
d d |du r|
dd n|d
u r|
dd |j t |j t r| j |S )N
.directoryT Directory/ rR r\ r_ r^ ra re rc Frd rf )r rr r r rQ rS _MenuEditor__addXmlTextElementrR
setAttributesr1 replacerg rh ri rj rA r r r4 ) r r r6 r8 r9 r: rm rn xml_menur ro s r rD zMenuEditor.editMenu s dni 0 0
1T^5LP\5\5\(( dD)A)ABBH $ $X{DI > (( dD)A)ABBH $ $X{H E E E &x 0 0DNN/ 5 ##F++
,
fd+++ MM&$tM 4 4 4 C ##M22
:
m[999 MM-TM B B B ; ##I..
2
i111 MM)WTM : : : ( MM&$ ' ' ' MM+v . . . .
%
MM+w / / / T>> MM(F + + + +
u__ MM(G , , , ''))) dk4 ( ( INN r c 4 | |d d S NT)rm r2 r r> s r
hideMenuEntryzMenuEditor.hideMenuEntry s! 955555r c 6 | |dd d S )NFrm rn r{ r| s r unhideMenuEntryzMenuEditor.unhideMenuEntry s# 9eDDDDDr c 4 | |d d S rz )rD r r s r hideMenuzMenuEditor.hideMenu s
dd
+++++r c | |dd | | dd d }| d }| d }||z D ]}| | d S )NFr TDeleted
NotDeleted)rD rQ rS findallremove)r r rx deleted not_deletednodes r
unhideMenuzMenuEditor.unhideMenu s
deE
:::$$T\\$%=%=uEE""9-- &&|44 k) " "D OOD ! ! ! ! " "r c | | dk rX| |j j |j D ]}| || | j |S )Ndelete) getAction_MenuEditor__deleteFilerg r rl rK r r4 )r r> r5 s r deleteMenuEntryzMenuEditor.deleteMenuEntry sv >>) $ $ 0 0 i 4 = > > >#+
6
6""695555 INN r c | | dk r| |j j g |j _ |j D ]s}|j | }|j |j |< |j | }|j |j |< |j j | t| j
|S Nrevert) r r rg r Originalrl Entriesindex MenuEntriesappendr r4 )r r> r5 r s r revertMenuEntryzMenuEditor.revertMenuEntry s >>) $ $ 0 0 i 4 = > > >)+I &#+
:
: ,,Y77(1(:u% *00;;,5,>"5)"*11&9999 INN r c | | dk r| |j j j | |j | | | dd }| | }|
| | j |S )Nr T)
r r rr rg r rK rA rQ rS _MenuEditor__get_parent_noder r r4 )r r rx r5 s r
deleteMenuzMenuEditor.deleteMenu s >>$ 8 + + dn 9 B C C C t{D 1 1 1(( dD)A)ABBH++H55F MM( # # # INN r c | | dk rN| |j j j |j j |_ | j |S r )r r rr rg r r r r4 r s r
revertMenuzMenuEditor.revertMenu sX >>$ 8 + + dn 9 B C C C !^ 4DN INN r c r | |j |d | j |S )NT)r< )rK rA r r4 )r rG s r deleteSeparatorzMenuEditor.deleteSeparator% s8 9+YdCCC r c t |t rt |j t sdS |j dk rdS |j dk r/t |j t |j z dk rdS nKt |t r6| dk rdS | dk rdS dS dS )NnoneBothr Userr r )r r rr r getTyperk Submenusr )r entrys r r zMenuEditor.getAction- s eT " " eoy99
v ((**f44 x ((**f44EN##c%*;&<&<<: : x
y
)
) }}&(( x F** x vvr c X |s| j }t |j t r|j | d D ]V}t |t r| ,t |t r| | Wd S )NT)rn )r r rr r r.
getEntriesr r+ )r r r s r
__saveEntrieszMenuEditor.__saveEntriesB s 9D dni 0 0 " N ! ! !__D_11 * *E%++
*
E4((
*""5))) * *r c 6 t j t j | j s6t j t j | j | j | j d d S )Nzutf-8)encoding)r! r" isdirdirnamer makedirsr writer- s r
__saveMenuzMenuEditor.__saveMenuM sh w}}RW__T] ; ;<< 8 K
66 7 7 7
88888r c \ d} |dk r||z }n|dz t | z |z }|dk rd}n|dk rd}|| j vrdt j t j t d || s| j | n|dz
}|S )Nr r -r0 applicationsrq zdesktop-directories)strr r! r" isfiler# r r )r r6 extensionpostfixr dirs r
__getFileNamezMenuEditor.__getFileNameR s !|| )+ #:G 4y@J&&$l**+t~--bgnn ]1 -sH==7 7- %%h///1 " r Tc | |s| j }d|v r| dd \ }}n|}d}d }| d D ]B}| d }|j |k r |r| ||| }n|}|r nC|s4|r2| || }|r| ||| }n|}|S )Nrs r rt r rR )r r$ r findtextrQ _MenuEditor__addXmlMenuElement)r r" createelementr6 foundr name_nodes r __getXmlMenuzMenuEditor.__getXmlMenug s iG $;; ::c1--LT44DDOOF++ D &))I~%% ! - -dFD A AEE E
++GT::D
))$== r c h t j d| }t j d| }||_ |S )Nr rR r%
SubElementr )r r r6 menu_noder s r __addXmlMenuElementz MenuEditor.__addXmlMenuElement s3 $VW55 $VY77 r c > t j || }||_ |S r r )r r r6 r r s r __addXmlTextElementz MenuEditor.__addXmlTextElement s g.. r Includec 4 | d }| d }||z }|D ]9}|d j dk r&|d j |k r| | :t j || }| |d| |S )Nr Excluder r )r tagr r r% r ru ) r r r type_includesexcludesrulesruler s r __addXmlFilenamezMenuEditor.__addXmlFilename s ??9--??9--8# % %DAw{j((T!W\X-E-Et$$$ w// z8<<< r c t j d| }| |d| | |d| |S )NMoveOldNew)r% r ru )r r oldnewr s r __addXmlMovezMenuEditor.__addXmlMove sI 00 uc222 uc222 r c | d D ]}| | t j d| }|j D ]}|d dk rt j d| }$|d dk r | |d|d }N|d dk r | |d|d }x|d dk r%t j d| }|d |j d< |S ) Nr r r r r MenunameMergetype)r r r% r orderru attrib)r r layoutr r childs r __addXmlLayoutzMenuEditor.__addXmlLayout s OOH-- ! !D NN4 '22\ 0 0EQx;&& (d;;qZ'' 00z58LLqZ'' 00z58LLqW$$ ($77',Qx V$ r c ~ t }g |_ |j j |_ |j j |_ |j j |_ |j j |_ |j j |_ |j ddg |j D ]}t |t r| |j 2t |t r| |j bt |t r| |j ddg ||_ |S )Nr r files)r r
show_emptyinline
inline_header inline_alias inline_limitr r r r
parseMenunamerR r
parseFilename
DesktopFileIDr parseSeparator)r r5 r r s r __addLayoutzMenuEditor.__addLayout s! "M 4
,
%} : $m 8 $m 8 Wg .///^ ( (E%&&
($$UZ0000E9--
($$U%89999E9--
(%%''' Wg .///
r c |s|rX|r |j | dz }n |r|j | }|j || n|j | | | dd }t
|t rQ|j | |j | |
||j d n*t
|t r|
| |s|r2| | | ||j d S d S )Nr Tr )r r insertr rQ rS r r r rl _MenuEditor__addXmlFilenamer r
addSubmenu_MenuEditor__addLayout_MenuEditor__addXmlLayoutr )r r5 r r< r= r
xml_parents r
__addEntryzMenuEditor.__addEntry sr )F )
5 ,,U33a7
5 ,,V44 N ! !% / / / / N ! !% ( ( (&&v~~dD'A'ABB
eY ' ' % % %e , , , M ( ( ( ! !*e.A9 M M M M
t
$
$ % e $ $ $ ;F ; V $ $ $
FM : : : : : ; ;r c " |j | | | dd }t |t
rQ|j | |j | | ||j d n/t |t r|j | |s|r2| | |
||j d S d S )NTr )r r rQ rS r r rl r r r r r r r r )r r5 r r< r= r s r
__deleteEntryzMenuEditor.__deleteEntry s e$$$&&v~~dD'A'ABB
eY ' ' * M ( ( ( % %e , , , ! !*e.A9 M M M M
t
$
$ * O " "5 ) ) ) ;F ; V $ $ $
FM : : : : : ; ;r c t j | n# t $ r Y nw xY w | j | d S # t $ r Y d S w xY wr )r! r OSErrorr
ValueError)r r s r __deleteFilezMenuEditor.__deleteFile s} Ih D N ! !( + + + + + DD s
$$A
AAc |D ]j}|j }|sd|_ |j }|sd|_ t | r| | kd S )Nrt )r striptailrk $_MenuEditor__remove_whilespace_nodes)r r r r r s r __remove_whitespace_nodesz$MenuEditor.__remove_whitespace_nodes s 6 6E:##%%D
:##%%D
5zz
6..u555 6 6r c J | D ]
\ }}||u r|c S d S r )_MenuEditor__iter_parent)r r r5 r s r __get_parent_nodez MenuEditor.__get_parent_node sB ! / / 1 1 MFE}} r c # X K | j D ]
}|D ]}||fV d S r )r getiterator)r r5 r s r
__iter_parentzMenuEditor.__iter_parent sT i++-- $ $F
$
$ em####
$ $ $r )NNF)NNNNNNN)NNNNN)NN)NNNNNNNN)NNNNNN)TN)r ))__name__
__module__ __qualname__r r r. r? rE rH rO rV rX rZ r2 rD r} r r r r r r r r r r+ r, r1 rQ r ru r rT r r r3 rK r r) r r r r r
r
s !2 !2 !2 !2F
( ( ( (T1 1 1 1f6 6 6E E E, , ," " " * * * *9 9 9
* @
, & .; ; ; ;.; ; ; ; 6 6 6 $ $ $ $ $r r
)__doc__r! xml.etree.cElementTreer% cElementTree ImportErrorxml.etree.ElementTreer( xdg.Menur r r r r xdg.BaseDirectoryr r xdg.Exceptionsr
xdg.Configr objectr
r r r r s *********** * * *)))))))))))* H G G G G G G G G G G G G G < < < < < < < < ' ' ' ' ' ' " " " " " "E$ E$ E$ E$ E$ E$ E$ E$ E$ E$s ##