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 deZdS ) z CLass to edit XDG Menus N)Menu MenuEntryLayout SeparatorXMLMenuBuilder)xdg_config_dirs
xdg_data_dirs) ParsingError) setRootModec @ s\ e Zd ZdQddZdQddZdd ZdRd d
ZdSd d ZdTd
dZdTddZ dTddZ
dTddZ dTddZ dUddZ
dVddZdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Z d-d. Zd/d0 Zd1d2 Zd3d4 ZdWd6d7Z d8d9 Z d:d; Z dXd=d>Z d?d@ Z dAdB Z!dCdD Z"dTdEdFZ#dTdGdHZ$dIdJ Z%dKdL Z&dMdN Z'dOdP Z(dS )Y
MenuEditorNFc C s2 d | _ d | _d | _t | _| ||| g | _d S N)menufilenametreer parserparse filenamesselfr
r root r 0/usr/lib/python3/dist-packages/xdg/MenuEditor.py__init__ s
zMenuEditor.__init__c C s |rt d t|tr|| _n|r| j|| _n| j | _|r'| jj| _n|r-|| _ntj
t d dtj | jjd | _z t
| j| _W n$ tya t
d| jj }t
|| _Y n
tym td| jw | | 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 r r r r $ s.
& zMenuEditor.parsec C s | | j | d S r )_MenuEditor__saveEntriesr
_MenuEditor__saveMenur r r r saveG s zMenuEditor.savec
C sD t | |d}
| |
||||||}
| ||
|| | j |
S )N.desktop)r _MenuEditor__getFileName
editMenuEntry_MenuEditor__addEntryr
sort) r parentnamecommand genericnamecommenticonterminalafterbefore menuentryr r r createMenuEntryK s
zMenuEditor.createMenuEntryc C sX t }||_|jd |_|j|_|j|_| |||||}| |||| | j |S )Nr ) r ParentDepth
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|dkr:| |tj |d|jtj |d|j | j
|S )NT) rB r- _MenuEditor__getXmlMenur
NamegetPath_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|
ds|
d| |
jd|dd |r-|
ds%|
d| |
jd|dd |rB|
ds:|
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|jdkr| j |S )NrI TlocaleComment GenericNameExecIconTerminaltruefalse NoDisplayFHiddenr ) DesktopEntryhasKeysetupdateAttributeslenParentsr
r. ) r r8 r0 r2 r3 r1 r4 r5 nodisplayhidden deskentryr r r r, sB
zMenuEditor.editMenuEntryc C s t |jtr)|jjdkr)| |dd}| |d|jd |j|jd n4t |jts]|s4|j}| |d
dd} |jsH|
dd|_| |dd}| |d| t| |_|jj }
|rv|
dsn|
d| |
j
d|dd |r|
ds|
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 |jtr| 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 )NFra 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 | |ddd}|d}|d}|| D ]}|| q d S )NFrr TDeleted
NotDeleted)r= rH rJ findallremove)r r
rk deleted not_deletednoder r r
unhideMenu s
zMenuEditor.unhideMenuc C sB | |dkr | |jj |jD ]}| || q| j |S )Ndelete) 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 Nrevert) r r r[ r Originalr` Entriesindex MenuEntriesappendr
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 |tr-t |jts
dS |j dkrdS |j dkr+t|jt|j dkr+dS dS t |trD| dkr:dS | dkrBdS dS dS )NnoneBothr Userr r~ )r r re r getTyper_ Submenusr )r entryr r r r - s$
zMenuEditor.getActionc C s^ |s| j }t|jtr|j |jddD ]}t|tr"| qt|tr,| | qd S )NT)rb )r
r re r r)
getEntriesr r& )r r
r r r r
__saveEntriesB s
zMenuEditor.__saveEntriesc C s@ t jt j| jst t j| j | jj| jdd d S )Nzutf-8)encoding)r r isdirdirnamer makedirsr writer( r r r
__saveMenuM s zMenuEditor.__saveMenuc C s d} |dkr || }n
|d t | | }|dkr d}n|dkr#d}|| jvr?tjtjtd ||s?| j| |S |d7 }q)Nr r -r* applicationsrd zdesktop-directories)strr r r isfiler r r )r r0 extensionpostfixr dirr r r
__getFileNameR s"
zMenuEditor.__getFileNameTc C s |s| j }d|v r|dd\}}n|}d}d }|dD ] }|d}|j|kr5|r3| |||}n|}|r9 nq |sQ|rQ| ||}|rO| |||}|S |}|S )Nrf r rg r rI )r r rx findtextrH _MenuEditor__addXmlMenuElement)r r createelementr0 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.__addXmlTextElementIncludec 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_includesexcludesrulesruler| r r r __addXmlFilename s
zMenuEditor.__addXmlFilenamec C s, t d|}| |d| | |d| |S )NMoveOldNew)r! r rh )r r oldnewr| r r r __addXmlMove s zMenuEditor.__addXmlMovec C s | dD ]}|| qtd|}|jD ]B}|d dkr%td|}q|d dkr5| |d|d }q|d dkrE| |d|d }q|d dkrXtd|}|d |jd< q|S ) Nr r r r r MenunameMergetype)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|_|jddg |jD ]#}t |t
r8| |j q*t |t
rD||j q*t |trM| q*|jddg ||_ |S )Nr r files)r r
show_emptyinline
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|trH|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|tr)|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 striptailr_ $_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 ImportErrorxml.etree.ElementTreer$ xdg.Menur r r r r xdg.BaseDirectoryr r xdg.Exceptionsr
xdg.Configr
objectr r r r r s