Mini Shell
3
�Y�]�> � @ sd d Z ddlZddlZddlZddlZddlmZ ddlZddlm Z m
Z
ddl m Z dZ
i Zdadadadaejd dkZdd � Zd1d
d �ZG d d
� d
�ZG dd� d�ZG dd� d�ZG dd� d�Zedd�Zedd�Z edd�Z edd�Z edd�Z edd�Z edd �Z!edd �Z"ed d �Z#d a$d!d"� Z%d#d$� Z&d%d&� Z'd2d(d)�Z(d3d*d+�Z)d4d-d.�Z*d/d0� Z+dS )5a�
This module is based on a rox module (LGPL):
http://cvs.sourceforge.net/viewcvs.py/rox/ROX-Lib2/python/rox/mime.py?rev=1.21&view=log
This module provides access to the shared MIME database.
types is a dictionary of all known MIME types, indexed by the type name, e.g.
types['application/x-python']
Applications can install information about MIME types by storing an
XML file as <MIME>/packages/<application>.xml and running the
update-mime-database command, which is provided by the freedesktop.org
shared mime database package.
See http://www.freedesktop.org/standards/shared-mime-info-spec/ for
information about the format of these files.
(based on version 0.13)
� N)�
BaseDirectory)�minidom�
XML_NAMESPACE)� defaultdictz5http://www.freedesktop.org/standards/shared-mime-info� c C s dj dd� | jD ��j� S )zGet text of XML node� c S s g | ]
}|j �qS � )Z nodeValue)�.0�nr r �/usr/lib/python3.6/Mime.py�
<listcomp>. s z"_get_node_data.<locals>.<listcomp>)�joinZ
childNodes�strip)Znoder r r �_get_node_data, s r c C sJ |dkr d| kr | j dd�\} }| |ftkr>t| |�t| |f< t| |f S )z�Get the MIMEtype object for this type, creating a new one if needed.
The name can either be passed as one part ('text/plain'), or as two
('text', 'plain').
N�/� )�split�types�MIMEtype)�media�subtyper r r �lookup0 s
r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )r z#Type holding data about a MIME typec C s || _ || _d| _dS )z?Don't use this constructor directly; use mime.lookup() instead.N)r r �_comment)�selfr r r r r �__init__> s zMIMEtype.__init__c C s� t jjd| j| jd �}x�tj|�D ]x}tj|�}|dkr<q$x^|j j
t d�D ]L}|j t
d�p^d}d|tjjk }|| jd kr�|t|�f| _|d krLdS qLW q$W dS )
z>Loads comment for current language. Use get_comment() instead.�mimez.xmlN�comment�langZenr r � )�os�pathr
r r r �load_data_pathsr �parseZdocumentElementZgetElementsByTagNameNS�FREE_NSZgetAttributeNSr �xdgZLocaleZlangsr r )r �resourcer �docr r Zgoodnessr r r �_loadH s
zMIMEtype._loadc C s* | j dkr dt| �f| _ | j� | j d S )z;Returns comment for current language, loading it if needed.Nr r )r �strr' )r r r r � get_commentW s
zMIMEtype.get_commentc C s& t � t| �}|tkr"tt| �S | S )z:Returns the canonical MimeType object if this is an alias.)� update_cacher( �aliasesr )r �sr r r � canonical_ s
zMIMEtype.canonicalc C s t � tdd� tt| � D ��S )z5Returns a set of Mime types which this inherits from.c s s | ]}t |�V qd S )N)r )r �tr r r � <genexpr>j s z)MIMEtype.inherits_from.<locals>.<genexpr>)r* �set� inheritancer( )r r r r �
inherits_fromg s zMIMEtype.inherits_fromc C s | j d | j S )Nr )r r )r r r r �__str__l s zMIMEtype.__str__c C s d| | j p df S )Nz<%s: %s>z(comment not loaded))r )r r r r �__repr__o s zMIMEtype.__repr__N) �__name__�
__module__� __qualname__�__doc__r r' r) r- r2 r3 r4 r r r r r < s
r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� MagicRulec C s� d | _ d | _d}x |jd�}|dkr&P ||7 }qW |s>d| _nt|jd��| _d}x |jd�}|dkrhP ||7 }qTW t|jd��| _|jd�}|jd�}t|�t|�d> | _|j| j�| _ |jd�}|dkr�|j| j�| _
|jd�}nd | _
|d k�rRd}xB|d
k�r>|d k�r>|jd�}|d
k�s0|d k�r2P ||7 }� q�W t|jd��| _ nd| _ |d
k�r�d}x.|d k�r�|jd�}|d k�r�P ||7 }�qhW t|jd��| _ nd| _ |d k�r�t
d ��d S )
N� r � >r �ascii� =� � &� ~� +�
zMalformed MIME magic line)�next�prev�read�nest�int�decode�start�ord�lenvalue�value�mask�word�range�
ValueError) r �fZind�crI ZhbZlb�w�rr r r r s s^
zMagicRule.__init__c C s | j | j | j S )N)rI rK rO )r r r r � getLength� s zMagicRule.getLengthc C s0 | j |j k r|| _| |_n| jr,| jj|� d S )N)rF rC rD �
appendRule)r �ruler r r rV � s
zMagicRule.appendRulec C s$ | j |�r | jr | jj|�S dS d S )NT)�match0rC �match)r �bufferr r r rY � s
zMagicRule.matchc C s� t |�}x�t| j�D ]�}| j| }|| j }||k r8dS | jr�d}xft| j�D ]J}trn||| | j| @ }n t||| �t| j| �@ }|t|�7 }qNW n |||� }|| jkrdS qW d S )NFr T) �lenrO rI rK rM �PY3rJ �chrrL ) r rZ �l�or, �eZtest�irR r r r rX � s
zMagicRule.match0c C s$ d| j | j| j| j| j| j| jf S )Nz!<MagicRule %d>%d=[%d]%r&%r~%d+%d>)rF rI rK rL rM rN rO )r r r r r4 � s zMagicRule.__repr__N) r5 r6 r7 r rU rV rY rX r4 r r r r r9 r s @r9 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� MagicTypec C s || _ g | _d | _d S )N)�mtype� top_rules� last_rule)r rc r r r r � s zMagicType.__init__c C s8 t |�}|jr"| jr"| jj|� n | jj|� || _|S )N)r9 rF re rV rd �append)r rQ Znruler r r �getLine� s zMagicType.getLinec C s$ x | j D ]}|j|�r| jS qW d S )N)rd rY rc )r rZ rW r r r rY � s
zMagicType.matchc C s
d| j S )Nz<MagicType %s>)rc )r r r r r4 � s zMagicType.__repr__N)r5 r6 r7 r rg rY r4 r r r r rb � s rb c @ s8 e Zd Zdd� Zdd� Zddd�Zdd d
�Zd d � Zd
S )�MagicDBc C s i | _ d| _d S )Nr )r �maxlen)r r r r r � s zMagicDB.__init__c C sF t |d���0}|j� }|dkr&td���x|j� jd�}|s>P |d dksZ|d
d � dkrbtd ��|d
d� jd �\}}t|�}t|�}y| j| }W n g }|| j|< Y nX t |�} |j
d
�}
|j dd
� xR|
o�|
d k�r"| j |�} | �r
| j
� | jk�r
| j
� | _|j
d
�}
|j dd
� q�W |j| � |
s*P q*W W d Q R X d S )N�rbs MIME-Magic
zNot a MIME magic filer< r �[r z]
zMalformed section headingr �:� [���rn ���ro )�open�readline�IOErrorrH rP r rG r r rb rE �seekrg rU ri rf ) r ZfnamerQ �lineZsheadZpriZtnamerc ZentsZ magictyperR rW r r r � mergeFile s<
zMagicDB.mergeFile�d r c C sZ xTt | jj� dd�D ]>}||kr"q||k r,P x$| j| D ]}|j|�}|r8|S q8W qW d S )NT)�reverse)�sortedr �keysrY )r �data�max_pri�min_priZpriority�type�mr r r �
match_data- s
zMagicDB.match_datac C sB y0t |d��}|j| j�}W d Q R X | j|||�S Y nX d S )Nrj )rp rE ri r )r r r{ r| rQ Zbufr r r rY 9 s z
MagicDB.matchc C s
d| j S )Nz <MagicDB %s>)r )r r r r r4 A s zMagicDB.__repr__N)rv r )rv r )r5 r6 r7 r ru r rY r4 r r r r rh � s
,
rh �textZplain�inodeZ blockdeviceZ
chardeviceZ directoryZfifoZsocket�symlinkZdoor� application�
executableFc C sP da i ag ai ai att�at� a dd� } x$t
j t j
jdd��D ] }| |� q@W x&t
j t j
jdd��D ]}t j|� qfW tjdd� d � x\t
j t j
jdd
��D ]D}t|d ��0}x(|D ] }|j� jd d �\}}|t|< q�W W d Q R X q�W xdt
j t j
jdd
��D ]L}t|d ��8}x0|D ](}|j� jd d �\}}t| j|� �qW W d Q R X q�W d S )NTc S s� t | ���}x�|D ]�}|jd�r q|dd
� }|jdd�\}}t|�}|jd�r�|dd� }d|kppd|kppd |ks�|tkr|t|< qd|ks�d|ks�d |kr�tj||f� q|t|< qW W dQ R X dS ) z6Loads name matching information from a MIME directory.�#Nr rl z*.r �*rk �?ro )rp �
startswithr r �exts�globsrf �literals)r rQ rt Z type_name�patternrc �restr r r �_import_glob_file^ s
z*_cache_database.<locals>._import_glob_filer r� �magicc S s t | d �S )Nr )r[ )�xr r r �<lambda>y s z!_cache_database.<locals>.<lambda>)�keyr+ rT r Z
subclasses)�_cache_uptodater� r� r� r+ r r0 r1 rh r� r r! r r r
ru �sortrp r r �add)r� r rQ rt �aliasr- �sub�parentr r r �_cache_databaseR s.
r� c C s t s
t� d S )N)r� r� r r r r r* � s r* c C s� t � tjj| �}|tkr"t| S |j� }|tkr:t| S |}x8|jd�}|dk rTP ||d d� }|tkr@t| S q@W |}x8|jd�}|dk r�P ||d d� }|tkr~t| S q~W x0tD ](\}}t j ||�r�|S t j ||�r�|S q�W dS )z6Returns type of file by its name, or None if not known�.r r N)
r* r r �basenamer� �lower�findr� r� �fnmatch)r ZleafZlleafZext�pZglobZ mime_typer r r �get_type_by_name� s:
r� rv c C s t � tj| ||�S )z:Returns type of file by its contents, or None if not known)r* r� rY )r r{ r| r r r �get_type_by_contents� s r� c C s t � tj| ||�S )z0Returns type of the data, which should be bytes.)r* r� r )rz r{ r| r r r �get_type_by_data� s r� Tc
C s� t � y |rtj| �}n
tj| �}W n t| �}|p:tS tj|j�r�t| |d�}|s`t| �}|spt| |d�}|dkr�tj |j�d@ r�t
S tS |S tj |j�r�t S tj
|j�r�tS tj|j�r�tS tj|j�r�tS tj|j�r�tS tj|j�� r�tS tS )a� Returns type of file indicated by path.
path :
pathname to check (need not exist)
follow :
when reading file, follow symbolic links
name_pri :
Priority to do name matches. 100=override magic
This tries to use the contents of the file, and falls back to the name. It
can also handle special filesystem objects like directories and sockets.
)r| )r{ N�I )r* r �stat�lstatr� r� �S_ISREG�st_moder� �S_IMODE�app_exe�S_ISDIR� inode_dir�S_ISCHR�
inode_char�S_ISBLK� inode_block�S_ISFIFO�
inode_fifo�S_ISLNK�
inode_symlink�S_ISSOCK� inode_socket�
inode_door)r ZfollowZname_pri�str. r r r �get_type� s@
r� c
C s� | d7 } t |�j� }tjjdd�}tjj|| �}x>tj|�D ]0}yt |�j� }W n w<Y nX ||kr<dS q<W datjjtj|�| �}t |d�j |� d}tj
tj ||tjd��r�tj |� t
d| ��dS ) a Copy 'package_file' as ``~/.local/share/mime/packages/<application>.xml.``
If package_file is None, install ``<app_dir>/<application>.xml``.
If already installed, does nothing. May overwrite an existing
file with the same name (if the contents are different)z.xmlr ZpackagesNFrS zupdate-mime-databasez�The '%s' command returned an error code!
Make sure you have the freedesktop.org shared MIME package:
http://standards.freedesktop.org/shared-mime-info/)rp rE r r r
r r! r� Zsave_data_path�write�spawnlp�P_WAIT�unlink� Exception) r� Z package_fileZnew_dataZ package_dirr% r� Zold_dataZnew_fileZcommandr r r �install_mime_info� s&
r� )N)rv r )rv r )Trv ),r8 r r� �sysr� r$ r Z
xdg.LocaleZxml.domr r � collectionsr r# r r� r� r� r� � version_infor\ r r r r9 rb rh r� r� r� r� r� r� r� r� r� r� r� r* r� r� r� r� r� r r r r �<module> sL
6pJ
7!
*