Mini Shell
3
��P�2 � @ sv d Z ddlZddlZddlZddlZddlmZmZmZm Z m
Z
m Z m Z ddl
ZddlmZ dd� ZG dd� d�ZdS ) z5
Base Class for DesktopEntry, IconTheme and IconData
� N)� ParsingError�DuplicateGroupError� NoGroupError�
NoKeyError�DuplicateKeyError�ValidationError�debug)�uc C s* y| j dd� dS tk
r$ dS X dS )z>Return True if a string consists entirely of ASCII characters.�ascii�strictTFN)�encode� UnicodeError)�s� r � /usr/lib/python3.6/IniFile.py�is_ascii s
r c @ s e Zd ZdZdZdZdZd=dd�Zdd� Zd>dd �Z d?d d �Z
d
d� Z dd� Z d@dd�Z
dAdd�Zdd� Zdd� ZdBdd�Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� ZdCd*d+�ZdDd,d-�Zd.d/� Zd0d1� Z dEd3d4�Z d5d6� Z d7d8� Z dFd9d:�Z d;d<� Z!dS )G�IniFile� FNc C s t � | _|r| j|� d S )N)�dict�content�parse)�self�filenamer r r �__init__ s zIniFile.__init__c C s t | j|j�S )N)Zcmpr )r �otherr r r �__cmp__! s zIniFile.__cmp__c # C s� | j }tjj|�s td|��ytj|dddd�}W n0 tk
rb } ztrN|�ndS W Y dd}~X nX �x |D �]}|j � }|s�qlql|d dkr�qlql|d d kr�|j
d �j d
�}tr�| j |�r�t
||��ni ||< qly|jd d �\}} W n$ tk
�r
td
| |��Y nX |j � }y6t�r8| j||��r8t|||��n| j � || |< W ql ttfk
�rn td|��Y qlX qlW |j� || _d| _|�r�x2|D ]}
|
|k�r�|
| _P �q�W td|d |��dS )zyParse an INI file.
headers -- list of headers the parser will try to select as a default header
zFile not found�rzutf-8�replace)�encoding�errorsNr �#�[�]�=� zInvalid line: z#Parsing error on key, group missingFz[%s]-Header missing)r �os�path�isfiler �io�open�IOErrorr �strip�lstrip�rstrip�hasGroupr �split�
ValueError�hasKeyr �
IndexError�UnboundLocalError�closer �tainted� defaultGroup) r r Zheadersr �fd�e�lineZ currentGroup�key�value�headerr r r r $ sP
z
IniFile.parse�stringc C s� |s
| j }|| jkrN|| j| krN|r>| j| | j||� }q�| j| | }n>tr�|| jkrjt|| j��q�|| j| kr�t||| j��nd}|dkr�| j|�}g }n|g}x�|D ]�}|dkr�| j|�}n�|dkr�y t |�}W n t
k
r� d}Y nX nr|dk�r,y t |�}W n t
k
�r( d}Y nX n>|dk�rBt j
|�}n(|d k�rj|jd
�\} }
t | �t |
�f}|dk�r�|j|� q�|}q�W |S ) Nr T�boolean�integerr �numericg �regex�point�,)r6 r �_IniFile__addLocaler r r r �getList�_IniFile__getBoolean�intr0 �float�re�compiler/ �append) r r: �group�locale�type�listr; �values�result�x�yr r r �geti sL
z IniFile.getc C sl t jd|�rt jd|�}n:t jd|�r4t jd|�}n t jd|�rNt jd|�}n|g}|d dkrh|j� |S )Nz (?<!\\)\;z(?<!\\);z (?<!\\)\|z(?<!\\),r$ r ���)rI �searchr/ �pop)r r= rO r r r rE � s zIniFile.getListc C s< |dks|dks|dkr dS |dks4|dks4|dkr8dS dS ) Nr$ �true�TrueTr �false�FalseFr )r r> r r r Z __getBoolean� s
zIniFile.__getBooleanc C s> |s
| j }x.tjjD ]"}d||f }|| j| kr|S qW |S )z3add locale to key according the current lc_messagesz%s[%s])r6 �xdg�Locale�langsr )r r: rL ZlangZlangkeyr r r Z __addLocale� s zIniFile.__addLocale�Allc C s g | _ g | _tjj| j�d | _| j� xf| jD ]\}| j |� xL| j| D ]>}| j
|| j| | |� | j| | dkrJ| j j d| � qJW q0W d}|dks�|dkr�x| j D ]}|d| 7 }q�W |dks�|dkr�x| jD ]}|d| 7 }q�W |r�t || j��dS ) z�Validate the contents, raising ``ValidationError`` if there
is anything amiss.
report can be 'All' / 'Warnings' / 'Errors'
r$ r zValue of Key '%s' is emptyr_ ZWarningsz
- ZErrorsN)
�warningsr r% r& �splitextr �
fileExtension� checkExtrasr �
checkGroup�checkKeyrK r )r ZreportrL r: �msgr9 r r r �validate� s&
zIniFile.validatec C s d S )Nr )r rL r r r rd � s zIniFile.checkGroupc C s d S )Nr )r r: r; rL r r r re � s zIniFile.checkKeyc C s� |dkr| j |�}n|g}x�|D ]�}|dkr6| j|�}|dkrBq nb|dkrV| j|�}nN|dkrj| j|�}n:|dkr~| j|�}n&|dkr�| j|�}n|dkr�| j|�}|d kr�| jjd
||f � q |d kr | j jd | � q W d S )
NTr= Z localestringr> r@ r? rA rB r$ z'%s' is not a valid %s� z Value of key '%s' is deprecated)
rE � checkString� checkBoolean� checkNumber� checkInteger�
checkRegex�
checkPointr rK r` )r r: r; rN rO rP �coder r r �
checkValue� s,
zIniFile.checkValuec C s d S )Nr )r r r r rc s zIniFile.checkExtrasc C s, |dks|dkrdS |dkp"|dks(dS d S )N�1�0rh rX rZ r$ r )r r; r r r rj s zIniFile.checkBooleanc C s y t |� W n
dS d S )Nr$ )rH )r r; r r r rk s zIniFile.checkNumberc C s y t |� W n
dS d S )Nr$ )rG )r r; r r r rl s zIniFile.checkIntegerc C s t jd|�sdS d S )Nz^[0-9]+,[0-9]+$r$ )rI �match)r r; r r r rn % s zIniFile.checkPointc C s t |�r dS dS )Nr r$ )r )r r; r r r ri ) s zIniFile.checkStringc C s yt j|� W n
dS d S )Nr$ )rI rJ )r r; r r r rm , s zIniFile.checkRegexc
C s� | r| j rtdd��|r$|| _ n| j }tjj|�r^tjjtjj|�� r^tjtjj|�� tj|ddd���}|r�|j t
d�� | j r�|j t
d�| j � x2| j | j j
� D ] \}}|j t
d�||f � q�W |j t
d �� xn| j j
� D ]`\}}|| j kr�|j t
d�| � x,|j
� D ] \}}|j t
d�||f � �qW |j t
d �� q�W W d Q R X |�r�tj|�j}|tjB tjB tjB } tj|| � d
| _d S ) NzFile not foundr �wzutf-8)r z#!/usr/bin/env xdg-open
z[%s]
z%s=%s
�
F)r r r% r& �dirname�isdir�makedirsr( r) �writer r6 r �items�stat�st_mode�S_IXUSR�S_IXGRP�S_IXOTH�chmodr5 )
r r Ztrusted�fpr: r; �namerL Zoldmode�moder r r ry 3 s4
"
z
IniFile.writec C s� |s
| j }|dkr:ttjj�dkr:|d tjjd d }y|| j| |< W n tk
rl t|| j��Y nX || j ||�k| _
d S )NTr r! r" ) r6 �lenr\ r] r^ r �KeyErrorr r rT r5 )r r: r; rL rM r r r �set\ s z IniFile.setc C s0 | j |�r tr,t|| j��ni | j|< d| _d S )NT)r. r r r r r5 )r rL r r r �addGroupk s
zIniFile.addGroupc C s2 || j k}|r | j |= d| _ntr.t|| j��|S )NT)r r5 r r r )r rL Zexistedr r r � removeGroups s
zIniFile.removeGroupTc C s� |s
| j }yf|rXxFt| j| �D ]4}tjd| tjj d |�r ||kr | j| |= q W | j| j|�}d| _ |S t
k
r� } z4t r�||kr�t || j
��q�t||| j
��ndS W Y d d }~X nX d S )N�^�$Tr )r6 rO r rI rs r\ r] rA rW r5 r� r r r r )r r: rL Zlocalesr� r; r8 r r r � removeKey} s $zIniFile.removeKeyc C s
| j j� S )N)r �keys)r r r r �groups� s zIniFile.groupsc C s
|| j kS )N)r )r rL r r r r. � s zIniFile.hasGroupc C s |s
| j }|| j| kS )N)r6 r )r r: rL r r r r1 � s zIniFile.hasKeyc C s | j S )N)r )r r r r � getFileName� s zIniFile.getFileName)N)N)NFr= F)N)r_ )r= F)NF)NF)NT)N)"�__name__�
__module__� __qualname__r6 rb r r5 r r r rT rE rF rD rg rd re rp rc rj rk rl rn ri rm ry r� r� r� r� r� r. r1 r� r r r r r s>
E
6
(
)
r )�__doc__rI r% r{ r( Zxdg.Exceptionsr r r r r r r Z
xdg.Localer\ Zxdg.utilr r r r r r r �<module> s $