View file File name : MenuEditor.cpython-36.pyc Content :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