Mini Shell
3
0��G�G � @ sV d Z ddlT ddlT ddlT ddlT ddlT ddlZddlZddl Z G dd� d�Z
dS )z CLass to edit XDG Menus � )�*Nc @ sR e Zd ZdOdd�ZdPdd�Zdd� ZdQd d
�ZdRd d �ZdSd
d�ZdTdd�Z dUdd�Z
dVdd�Z dWdd�Z dXdd�Z
dYdd�Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� ZdZd6d7�Z d8d9� Z d:d;� Z d[d=d>�Z d?d@� Z dAdB� Z!dCdD� Z"dEdF� Z#d\dGdH�Z$d]dIdJ�Z%dKdL� Z&dMdN� Z'dS )^�
MenuEditorNFc C s* d | _ d | _d | _| j|||� g | _d S )N)�menu�filename�doc�parse� filenames)�selfr r �root� r � /usr/lib/python3.6/MenuEditor.py�__init__ s
zMenuEditor.__init__c C s� |dkrt d� t|t�r"|| _n|r2t|�| _nt� | _|dkrN| jj| _n2|rZ|| _n&tjj t
d dtjj | jj�d �| _yt j
jj| j�| _W nT tk
r� t j
jjd| jj d �| _Y n& t jjjk
r� td| j��Y nX | j| j� d S )NTr �menus� z�<!DOCTYPE Menu PUBLIC "-//freedesktop//DTD Menu 1.0//EN" "http://standards.freedesktop.org/menu-spec/menu-1.0.dtd"><Menu><Name>Applications</Name><MergeFile type="parent">z</MergeFile></Menu>zNot a valid .menu file)Z setRootMode�
isinstance�Menur r �Filenamer �os�path�joinZxdg_config_dirs�split�xml�dom�minidomr �IOErrorZ parseStringZparsersZexpatZ
ExpatErrorZ ParsingError�$_MenuEditor__remove_whilespace_nodes)r r r r
r r r r ! s&
& zMenuEditor.parsec C s | j | j� | j� d S )N)�_MenuEditor__saveEntriesr �_MenuEditor__saveMenu)r r r r �save< s zMenuEditor.savec
C sD t | j|d��}
| j|
||||||�}
| j||
|| � t| j� |
S )Nz.desktop)� MenuEntry�_MenuEditor__getFileName�
editMenuEntry�_MenuEditor__addEntry�sortr ) r �parent�name�command� genericname�comment�icon�terminal�after�before� menuentryr r r �createMenuEntry@ s
zMenuEditor.createMenuEntryc C sX t � }||_|jd |_|j|_|j|_| j|||||�}| j||||� t| j� |S )Nr ) r �ParentZDepthZ
DefaultLayout�Layout�editMenur" r# r ) r r$ r% r' r( r) r+ r, r r r r �
createMenuJ s
zMenuEditor.createMenuc C s& t |�}| j||||� t| j� |S )N)� Separatorr" r# r )r r$ r+ r, � separatorr r r �createSeparatorZ s
zMenuEditor.createSeparatorc C s. | j ||||� | j||||� t| j� |S )N)�_MenuEditor__deleteEntryr" r# r )r r- � oldparent� newparentr+ r, r r r �
moveMenuEntryc s
zMenuEditor.moveMenuEntryc C s� | j ||||� | j||||� | j| jj�}|jd�|jd�krt| j|tjj |jd�|j�tjj |jd�|j�� t
| j� |S )NT) r6 r" �_MenuEditor__getXmlMenur �Name�getPath�_MenuEditor__addXmlMover r r r# )r r r7 r8 r+ r, Z root_menur r r �moveMenuk s 2
zMenuEditor.moveMenuc C s. | j ||||� | j||||� t| j� |S )N)r6 r" r# r )r r4 r$ r+ r, r r r �
moveSeparatorw s
zMenuEditor.moveSeparatorc C s | j ||||� t| j� |S )N)r" r# r )r r- r7 r8 r+ r, r r r �
copyMenuEntry s
zMenuEditor.copyMenuEntryc
C sL |j }
|r0|
jd�s |
jd|� |
jd|dd� |rZ|
jd�sJ|
jd|� |
jd|dd� |r�|
jd�st|
jd|� |
jd|dd� |r�|
jd|� |r�|
jd|� |dkr�|
jd d
� n|d kr�|
jd d � |dkr�|
jd
d
� n|d kr�|
jd
d � | dk�r|
jdd
� n| d k�r&|
jdd � |j� t|j�dk�rHt| j� |S )Nr; T)�locale�CommentZGnericNe� GenericNameZExec�IconZTerminal�trueF�false� NoDisplay�Hiddenr )� DesktopEntry�hasKey�set�updateAttributes�len�Parentsr# r ) r r- r% r' r( r& r) r* � nodisplay�hidden� deskentryr r r r! � sB
zMenuEditor.editMenuEntryc C s� t |jt�rR|jjdkrR| j|jdd��}| j|d|jd � |jj|jd � nht |jt�s�|sh|j}| j |d�j
dd�} |js�| j
dd�|_| j|jdd��}| j|d| � t| �|_|jj }
|r�|
j d�s�|
j
d|� |
j
d|dd� |�r|
j d��s
|
j
d|� |
j
d|dd� |�rH|
j d ��s8|
j
d |� |
j
d |dd� |�rZ|
j
d
|� |dk�rr|
j
d d � n|d
k�r�|
j
d d� |dk�r�|
j
dd � n|d
k�r�|
j
dd� |jj� t |jt��r�t| j� |S )Nz
.directoryT� Directory�/� r; )rA rC rB rD rG rE FrF rH )r rR r r r: r< � _MenuEditor__addXmlTextElementr; Z
setAttributesr �replacerI rJ rK rL r/ r r# r ) r r r% r' r( r) rO rP �xml_menur rQ r r r r1 � sP
zMenuEditor.editMenuc C s | j |dd� d S )NT)rO )r! )r r- r r r �
hideMenuEntry� s zMenuEditor.hideMenuEntryc C s | j |ddd� d S )NF)rO rP )r! )r r- r r r �unhideMenuEntry� s zMenuEditor.unhideMenuEntryc C s | j |dd� d S )NT)rO )r1 )r r r r r �hideMenu� s zMenuEditor.hideMenuc C sN | j |ddd� | j|jdd�d�}x$| jddg|�D ]}|jj|� q6W d S )NF)rO rP TZDeletedZ
NotDeleted)r1 r: r< � _MenuEditor__getXmlNodesByName�
parentNode� removeChild)r r rW �noder r r �
unhideMenu� s zMenuEditor.unhideMenuc C sF | j |�dkrB| j|jj� x|jD ]}| j||� q$W t| j� |S )N�delete)� getAction�_MenuEditor__deleteFilerI r rN r6 r# r )r r- r$ r r r �deleteMenuEntry� s
zMenuEditor.deleteMenuEntryc C s� | j |�dkr|| j|jj� g |j_xL|jD ]B}|jj|�}|j|j|< |jj|�}|j|j|< |jjj |� q,W t
| j � |S )N�revert) ra rb rI r �OriginalrN �Entries�index� MenuEntries�appendr# r )r r- r$ rg r r r �revertMenuEntry� s
zMenuEditor.revertMenuEntryc C sX | j |�dkrT| j|jjj� | j|j|� | j|jdd��}|j j
|� t | j � |S )Nr` T)
ra rb rR rI r r6 r/ r: r< r\ r] r# r )r r rW r r r �
deleteMenu s
zMenuEditor.deleteMenuc C s6 | j |�dkr2| j|jjj� |jj|_t| j� |S )Nrd )ra rb rR rI r re r# r )r r r r r �
revertMenu s
zMenuEditor.revertMenuc C s | j |j|dd� t| j� |S )NT)r+ )r6 r/ r# r )r r4 r r r �deleteSeparator s
zMenuEditor.deleteSeparatorc C s� t |t�rXt |jt�sdS |jj� dkr,dS |jj� dkr�t|j�t|j� dkr�dS n.t |t�r�|j� dkrrdS |j� dkr�dS dS dS )NZnoneZBothrd ZUserr r` )r r rR r ZgetTyperM �Submenusrh )r �entryr r r ra s
zMenuEditor.getActionc C sb |s
| j }t|jt�r |jj� x<|jdd�D ],}t|t�rF|j� q.t|t�r.| j|� q.W d S )NT)rP )r r rR r r Z
getEntriesr r )r r ro r r r Z
__saveEntries3 s
zMenuEditor.__saveEntriesc C sd t jjt jj| j��s*t jt jj| j�� t| jd�}|jtj dd| j
j � j dd��� |j
� d S )N�wz
[\s]*([^
<]*)
[\s]*</z\1</z<?xml version="1.0" ?>
rT )r r �isdir�dirnamer �makedirs�open�write�re�subr Z toprettyxmlrV �close)r �fdr r r Z
__saveMenu> s
"zMenuEditor.__saveMenuc C s� d}x�|dkr|| }n|d t |� | }|dkr:d}n |dkrFd}|| jkr~tjjtjjtd ||�� r~| jj|� P q|d7 }qW |S )Nr �-z.desktopZ applicationsz
.directoryzdesktop-directoriesr )�strr r r �isfiler Z
xdg_data_dirsri )r r% � extensionZpostfixr �dirr r r Z
__getFileNameE s
zMenuEditor.__getFileNameTc C s� |s
| j }d|kr$|jdd�\}}n|}d}d }xZ| jd|�D ]J}x>| jd|�D ].}|jd j|krP|rx| j|||�}n|}P qPW |r>P q>W | r�|dkr�| j||�}|r�| j|||�}n|}|S )NrS r rT r r; r T)r r r[ �
childNodes� nodeValuer: � _MenuEditor__addXmlMenuElement)r r Zcreate�elementr% �foundr^ �childr r r Z __getXmlMenuY s, zMenuEditor.__getXmlMenuc C s$ | j jd�}| j|d|� |j|�S )Nr r; )r �
createElementrU � appendChild)r r� r% r^ r r r Z__addXmlMenuElementw s z MenuEditor.__addXmlMenuElementc C s, | j j|�}| j j|�}|j|� |j|�S )N)r r� ZcreateTextNoder� )r r� r% �textr^ r r r Z__addXmlTextElement| s
z MenuEditor.__addXmlTextElement�Includec C st xH| j ddg|�D ]4}|jd jdkr|jd jd j|kr|j|� qW | jj|�}|j| j|d|�� |j|�S )Nr� �Excluder r ) r[ r �nodeNamer� r] r r� r� rU )r r� r �typer^ r r r Z__addXmlFilename� s & zMenuEditor.__addXmlFilenamec C s> | j jd�}|j| j|d|�� |j| j|d|�� |j|�S )NZMoveZOldZNew)r r� r� rU )r r� �old�newr^ r r r Z __addXmlMove� s zMenuEditor.__addXmlMovec C s� x | j d|�D ]}|j|� qW | jjd�}x�|jD ]�}|d dkr\| jjd�}|j|� q4|d dkr|| j|d|d �}q4|d dkr�| j|d|d �}q4|d dkr4| jjd�}|jd|d � |j|� q4W |j|�S ) Nr0 r r3 r r ZMenuname�Merger� )r[ r] r r� �orderr� rU Z setAttribute)r r� �layoutr^ r� r� r r r Z__addXmlLayout� s zMenuEditor.__addXmlLayoutc c s4 x.|j D ]$}|jtjjjkr|j|kr|V qW d S )N)r �nodeTyper r �NodeZ ELEMENT_NODEr� )r r% r� r� r r r Z__getXmlNodesByName� s z MenuEditor.__getXmlNodesByNamec C s� t � }g |_|j j|_|j j|_|j j|_|j j|_|j j|_|jjddg� xP|jD ]F}t |t
�rr|j |j � qVt |t
�r�|j|j� qVt |t�rV|j� qVW |jjddg� ||_ |S )Nr� r �files)r0 r� Z
show_emptyZinlineZ
inline_headerZ inline_aliasZ inline_limitri rf r r Z
parseMenunamer; r Z
parseFilename�
DesktopFileIDr3 ZparseSeparator)r r$ r� ro r r r Z __addLayout� s$
zMenuEditor.__addLayoutc C s� |s|r>|r |j j|�d }n|r.|j j|�}|j j||� n |j j|� | j|jdd��}t|t�r�|jj|� |j j|� | j
||j d� nt|t �r�|j
|� |s�|r�| j|� | j||j� d S )Nr Tr� )rf rg �insertri r: r< r r rh rN �_MenuEditor__addXmlFilenamer� r Z
addSubmenu�_MenuEditor__addLayout�_MenuEditor__addXmlLayoutr0 )r r$ ro r+ r, rg �
xml_parentr r r Z
__addEntry� s"
zMenuEditor.__addEntryc C s� |j j|� | j|jdd��}t|t�rR|jj|� |jj|� | j||j d� nt|t
�rh|j j|� |sp|r�| j |� | j
||j� d S )NTr� )rf �remover: r< r r rN rh r� r� r rn r� r� r0 )r r$ ro r+ r, r� r r r Z
__deleteEntry� s
zMenuEditor.__deleteEntryc C sN yt j|� W n tk
r" Y nX y| jj|� W n tk
rH Y nX d S )N)r r� �OSErrorr �
ValueError)r r r r r Z __deleteFile� s zMenuEditor.__deleteFilec C sx g }xT|j D ]J}|jtjjjjkrD|jj� |_|jj� sV|j |� q |j
� r | j |� q W x|D ]}|j j
|� q`W d S )N)r r� r r r r� Z TEXT_NODE�data�stripri Z
hasChildNodesr r\ r] )r r^ Z remove_listr� r r r Z__remove_whilespace_nodes� s
z$MenuEditor.__remove_whilespace_nodes)NNF)NNF)NNNNNNN)NNNNN)NN)NN)NN)NN)NN)NNNNNNNN)NNNNNN)TN)r� )NN)NN)(�__name__�
__module__� __qualname__r
r r r. r2 r5 r9 r> r? r@ r! r1 rX rY rZ r_ rc rj rk rl rm ra r r r r: r� rU r� r= r� r[ r� r" r6 rb r r r r r r sJ
*
3
r ) �__doc__Zxdg.MenuZxdg.BaseDirectoryZxdg.ExceptionsZxdg.DesktopEntryZ
xdg.ConfigZxml.dom.minidomr r rv r r r r r �<module> s