3
,6`                 @   s   d Z ddlZddlZddgZdd Zdd Zdd	 Zd
d Zdd Zdd Z	dd Z
ddg fddZd ddZg fddZdd Zdd Zdd Zedkre  dS )!z%Mailcap file handling.  See RFC 1524.    Ngetcaps	findmatchc             C   s   d| krd| d fS dS d S )Nlinenor      )r   r    )entryr   r   /usr/lib/python3.6/mailcap.pylineno_sort_key	   s    r	   c              C   s   i } d}xt  D ]}yt|d}W n tk
r8   wY nX | t||\}}W dQ R X x6|j D ]*\}}|| kr||| |< qb| | | | |< qbW qW | S )a  Return a dictionary containing the mailcap database.

    The dictionary maps a MIME type (in all lowercase, e.g. 'text/plain')
    to a list of dictionaries corresponding to mailcap entries.  The list
    collects all the entries for that MIME type from all available mailcap
    files.  Each dictionary contains key-value pairs for that MIME type,
    where the viewing command is stored with the key "view".

    r   rN)listmailcapfilesopenOSError_readmailcapfileitems)capsr   ZmailcapfpZmorecapskeyvaluer   r   r   r      s    

c              C   sP   dt jkr"t jd } | jt j}n*dt jkr8t jd }nd}|d dddg}|S )z7Return a list of all mailcap files found on the system.ZMAILCAPSHOME.z	/.mailcapz/etc/mailcapz/usr/etc/mailcapz/usr/local/etc/mailcap)osenvironsplitpathsep)ZpathstrZmailcapsZhomer   r   r   r   -   s    


r   c             C   s    t jdtd t| d\}}|S )z?Read a mailcap file and return a dictionary keyed by MIME type.z2readmailcapfile is deprecated, use getcaps instead   N)warningswarnDeprecationWarningr   )r   r   _r   r   r   readmailcapfile?   s    r   c       	      C   s  i }x| j  }|sP |d dks|j dkr.q|}x4|dd dkrf| j  }|sTd}|dd | }q4W t|\}}|oz|s~q|dk	r||d< |d	7 }|jd
}x$tt|D ]}|| j ||< qW d
j|j }||kr|| j| q|g||< qW ||fS )a  Read a mailcap file and return a dictionary keyed by MIME type.

    Each MIME type is mapped to an entry consisting of a list of
    dictionaries; the list will contain more than one such dictionary
    if a given MIME type appears more than once in the mailcap file.
    Each dictionary contains key-value pairs for that MIME type, where
    the viewing command is stored with the key "view".
    r   # r   Nz\

r   r   /r$   )	readlinestrip	parseliner   rangelenjoinlowerappend)	r   r   r   lineZnextliner   fieldstypesjr   r   r   r   G   s6    	  
r   c       
      C   s   g }dt |  }}x.||k r@t| ||\}}|j| |d }qW t |dk rRdS |d |d |dd   }}}d|i}x^|D ]V}|jd}|dk r|}d}	n$|d| j }||d d j }	||krq|	||< qW ||fS )	zParse one entry in a mailcap file and return a dictionary.

    The viewing command is stored as the value with the key "view",
    and the rest of the fields produce key-value pairs in the dict.
    r   r   r   Nview=r!   )NN)r)   
parsefieldr,   findr&   )
r-   r.   infieldr   r1   restZfkeyZfvaluer   r   r   r'   p   s*    

 

r'   c             C   sT   |}x:||k r>| | }|dkr"P q|dkr4|d }q|d }qW | || j  |fS )z/Separate one key-value pair in a mailcap entry.;\r   r   )r&   )r-   r5   r6   startcr   r   r   r3      s    

r3   r1   z	/dev/nullc       	      C   sb   t | ||}xP|D ]H}d|krBt|d ||}|rBtj|dkrBqt|| |||}||fS W dS )a  Find a match for a mailcap entry.

    Return a tuple containing the command line, and the mailcap entry
    used; (None, None) if no match is found.  This may invoke the
    'test' command of several matching entries before deciding which
    entry to use.

    testr   N)NN)lookupsubstr   system)	r   MIMEtyper   filenameplistentrieser=   commandr   r   r   r      s    	

c                sl   g }|| kr|| |  }|j d}|d d }|| krB|| |  } d k	r\ fdd|D }t|td}|S )Nr#   r   z/*c                s   g | ]} |kr|qS r   r   ).0rE   )r   r   r   
<listcomp>   s    zlookup.<locals>.<listcomp>)r   )r   sortedr	   )r   rA   r   rD   Z	MIMEtypesr   )r   r   r>      s    
r>   c       
      C   s  d}dt |  }}x||k r| | }|d }|dkr`|dkrV| ||d  }|d }|| }q| | }|d }|dkr|| }q|dkr|| }q|dkr|| }q|dkr|}x ||k r| | d	kr|d }qW | || }	|d }|t|	| }q|d | }qW |S )
Nr!   r   r   %r:   st{})r)   	findparam)
r7   rA   rB   rC   Zresr5   r6   r<   r;   namer   r   r   r?      s6      
 


r?   c             C   sF   | j  d } t| }x,|D ]$}|d | j  | kr||d  S qW dS )Nr2   r!   )r+   r)   )rP   rC   r6   pr   r   r   rO      s    
rO   c        	      C   s   dd l } t }| jdd  s(t| d S xtdt| jdD ]}| j||d  }t|dk rjtd d S |d }|d }t||d|\}}|stdt q<td| t	j
|}|r<td| q<W d S )	Nr   r   r   z"usage: mailcap [MIMEtype file] ...r1   zNo viewer found forz
Executing:zExit status:)sysr   argvshowr(   r)   printr   typer   r@   )	rR   r   r5   argsrA   filerF   rE   stsr   r   r   r=      s&    

r=   c             C   s   t d xt D ]}t d|  qW t   | s4t } t d t   t| }xX|D ]P}t | | | }x:|D ]2}t|}x|D ]}t d| ||  q|W t   qjW qPW d S )NzMailcap files:	zMailcap entries:z  %-15s)rU   r   r   rI   )r   fnZckeysrV   rD   rE   keyskr   r   r   rT     s"      


rT   __main__)N)__doc__r   r   __all__r	   r   r   r   r   r'   r3   r   r>   r?   rO   r=   rT   __name__r   r   r   r   <module>   s$   
)
 