Your IP : 216.73.217.13
f
�hR2 � @ s0 d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z d d l Z d d l Z y d d l
Z
Wn e k
r� d Z
Yn Xd d d d d d d d
d d d
d g Z e j j d � Z Gd d � d � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e j j � Z Gd d
� d
e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d
� d
e � Z Gd! d � d e � Z! Gd" d# � d# � Z" Gd$ d% � d% e" � Z# d& d' d( � Z$ d) d* � Z% d+ d, � Z& d- d. � Z' d/ d0 � Z( d1 d2 � Z) Gd3 d4 � d4 e* � Z+ Gd5 d6 � d6 e+ � Z, Gd7 d8 � d8 e+ � Z- Gd9 d: � d: e+ � Z. Gd; d< � d< e+ � Z/ d S)=zDRead/write support for Maildir, mbox, MH, Babyl, and MMDF mailboxes.� N�Mailbox�Maildir�mbox�MH�Babyl�MMDF�Message�MaildirMessage� mboxMessage� MHMessage� BabylMessage� MMDFMessage�asciic @ s� e Z d Z d Z d d d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d. d/ � Z d d0 d1 � Z d2 d3 � Z d d4 d5 � Z d6 d7 � Z d8 d9 � Z d: d; � Z d< d= � Z d> d? � Z! d@ Z" d@ dA dB � Z# d S)Cr z*A group of messages in a particular place.NTc C s. t j j t j j | � � | _ | | _ d S)z Initialize a Mailbox instance.N)�os�path�abspath�
expanduser�_path�_factory)�selfr �factory�create� r � /usr/lib/python3.5/mailbox.py�__init__# s !zMailbox.__init__c C s t d � � d S)z$Add message and return assigned key.z&Method must be implemented by subclassN)�NotImplementedError)r �messager r r �add( s z Mailbox.addc C s t d � � d S)z=Remove the keyed message; raise KeyError if it doesn't exist.z&Method must be implemented by subclassN)r )r �keyr r r �remove, s zMailbox.removec C s | j | � d S)N)r )r r r r r � __delitem__0 s zMailbox.__delitem__c C s* y | j | � Wn t k
r% Yn Xd S)z'If the keyed message exists, remove it.N)r �KeyError)r r r r r �discard3 s
zMailbox.discardc C s t d � � d S)z>Replace the keyed message; raise KeyError if it doesn't exist.z&Method must be implemented by subclassN)r )r r r r r r � __setitem__: s zMailbox.__setitem__c C s. y | j | � SWn t k
r) | SYn Xd S)z9Return the keyed message, or default if it doesn't exist.N)� __getitem__r! )r r �defaultr r r �get> s
z Mailbox.getc C sI | j s | j | � St j | j | � � � } | j | � SWd QRXd S)z=Return the keyed message; raise KeyError if it doesn't exist.N)r � get_message�
contextlib�closing�get_file)r r �filer r r r$ E s
zMailbox.__getitem__c C s t d � � d S)z4Return a Message representation or raise a KeyError.z&Method must be implemented by subclassN)r )r r r r r r' M s zMailbox.get_messagec C s t j | j | � � j � S)z�Return a string representation or raise a KeyError.
Uses email.message.Message to create a 7bit clean string
representation of the message.)�email�message_from_bytes� get_bytes� as_string)r r r r r �
get_stringQ s zMailbox.get_stringc C s t d � � d S)z8Return a byte string representation or raise a KeyError.z&Method must be implemented by subclassN)r )r r r r r r. X s zMailbox.get_bytesc C s t d � � d S)z6Return a file-like representation or raise a KeyError.z&Method must be implemented by subclassN)r )r r r r r r* \ s zMailbox.get_filec C s t d � � d S)z Return an iterator over keys.z&Method must be implemented by subclassN)r )r r r r �iterkeys` s zMailbox.iterkeysc C s t | j � � S)zReturn a list of keys.)�listr1 )r r r r �keysd s z Mailbox.keysc c sF x? | j � D]1 } y | | } Wn t k
r8 w
Yn X| Vq
Wd S)z%Return an iterator over all messages.N)r1 r! )r r �valuer r r �
itervaluesh s
zMailbox.itervaluesc C s
| j � S)N)r5 )r r r r �__iter__q s zMailbox.__iter__c C s t | j � � S)z,Return a list of messages. Memory intensive.)r2 r5 )r r r r �valuest s zMailbox.valuesc c sL xE | j � D]7 } y | | } Wn t k
r8 w
Yn X| | f Vq
Wd S)z.Return an iterator over (key, message) tuples.N)r1 r! )r r r4 r r r � iteritemsx s
zMailbox.iteritemsc C s t | j � � S)z9Return a list of (key, message) tuples. Memory intensive.)r2 r8 )r r r r �items� s z
Mailbox.itemsc C s t d � � d S)z9Return True if the keyed message exists, False otherwise.z&Method must be implemented by subclassN)r )r r r r r � __contains__� s zMailbox.__contains__c C s t d � � d S)z*Return a count of messages in the mailbox.z&Method must be implemented by subclassN)r )r r r r �__len__� s zMailbox.__len__c C s( x! | j � D] } | j | � q
Wd S)zDelete all messages.N)r3 r" )r r r r r �clear� s z
Mailbox.clearc C s8 y | | } Wn t k
r& | SYn X| j | � | S)z3Delete the keyed message and return it, or default.)r! r" )r r r% �resultr r r �pop� s
z Mailbox.popc C s7 x0 | j � D] } | | j | � f SWt d � � d S)z6Delete an arbitrary (key, message) pair and return it.zNo messages in mailboxN)r1 r> r! )r r r r r �popitem� s zMailbox.popitemc C s� t | d � r | j � } n$ t | d � r< | j � } n | } d } x= | D]5 \ } } y | | | <WqO t k
r� d } YqO XqO W| r� t d � � d S)z4Change the messages that correspond to certain keys.r8 r9 FTzNo message with key(s)N)�hasattrr8 r9 r! )r �arg�sourceZbad_keyr r r r r �update� s
zMailbox.updatec C s t d � � d S)z&Write any pending changes to the disk.z&Method must be implemented by subclassN)r )r r r r �flush� s z
Mailbox.flushc C s t d � � d S)zLock the mailbox.z&Method must be implemented by subclassN)r )r r r r �lock� s z Mailbox.lockc C s t d � � d S)z#Unlock the mailbox if it is locked.z&Method must be implemented by subclassN)r )r r r r �unlock� s zMailbox.unlockc C s t d � � d S)z Flush and close the mailbox.z&Method must be implemented by subclassN)r )r r r r �close� s z
Mailbox.closec C s6 y | j d � SWn t k
r1 t d � � Yn Xd S)Nr z?String input must be ASCII-only; use bytes or a Message instead)�encode� UnicodeError�
ValueError)r r r r r �_string_to_bytes� s
zMailbox._string_to_bytesFc C s� t | t j j � r� t j � } t j j | | d � } | j | � | j d � | j
� } | j d t � } | j
| � | j r�| j t � r�| j
t � n t | t t t j f � rqt | t j � r� t j d t d � | j � } t | t � r| j | � } | r)| j d d � } | j d t � } | j
| � | j r�| j t � r�| j
t � n6t | d � r�t | d � r�t j d t d � | j } d
} x� | j � } | j d � r�| d
d � d } n# | j d
� r | d
d � d } | sP| r9| j d � r9d | d d
� } | j d t � } | j
| � | } q�W| j r�| r�| j t � r�| j
t � n t d t | � � � d
S)z%Dump message contents to target file.r s
z8Use of StringIO input is deprecated, use BytesIO instead� s
From s
>From �read�bufferzDUse of text mode files is deprecated, use a binary mode file insteadNs
� s
� s From s >From � zInvalid message type: %s������) �
isinstancer, r r �io�BytesIO� generator�BytesGenerator�flatten�seekrM �replace�linesep�write�_append_newline�endswith�str�bytes�StringIO�warnings�warn�DeprecationWarning�getvaluerK r@ rN �readline�
startswith� TypeError�type) r r �targetZ mangle_from_rN �gen�dataZlastline�liner r r �
_dump_message� sZ
zMailbox._dump_message)$�__name__�
__module__� __qualname__�__doc__r r r r r" r# r& r$ r' r0 r. r* r1 r3 r5 r6 r7 r8 r9 r: r; r< r> r? rC rD rE rF rG rK r^ ro r r r r r sB c @ sH e Z d Z d Z d Z d d d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- Z d. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d S)6r z A qmail-style Maildir mailbox.�:NTc C s t j | | | | � d t j j | j d � d t j j | j d � d t j j | j d � i | _ t j j | j � s� | r� t j | j d � x9 | j j � D] } t j | d � q� Wn t
| j � � i | _ d d d d i | _ d | _
d | _ d S)z Initialize a Maildir instance.�tmp�new�curi� r g�������?N)r r r r �joinr �_paths�exists�mkdirr7 �NoSuchMailboxError�_toc� _toc_mtimes�
_last_read� _skewfactor)r �dirnamer r r r r r r s ! zMaildir.__init__c C s� | j � } y | j | | � Wn/ t k
rQ | j � t j | j � � Yn Xt | � t | t � r� | j
� } | j | j � } | | j k r� d } n d } d } t j
j | j � j | j � d } t j
j | j | | | � } t | t � r/t j | j t j
j | j � | j � f � yL t t d � rgt j | j | � t j | j � n t j | j | � Wn] t k
r�} z= t j | j � | j t j k r�t d | � � n � WYd d } ~ Xn X| S)z$Add message and return assigned key.� rv r �linkz$Name clash with existing message: %sN) � _create_tmpro �
BaseExceptionrG r r �name� _sync_closerT r �
get_subdir�colon�get_infor �basename�splitrx r �utime�getatime�get_dater@ r� �rename�OSError�errnoZEEXIST�ExternalClashError)r r Ztmp_file�subdir�suffix�uniq�dest�er r r r s@
% "
z Maildir.addc C s, t j t j j | j | j | � � � d S)z=Remove the keyed message; raise KeyError if it doesn't exist.N)r r r rx r �_lookup)r r r r r r I s zMaildir.removec C s0 y | j | � Wn t t f k
r+ Yn Xd S)z'If the keyed message exists, remove it.N)r r! �FileNotFoundError)r r r r r r" M s zMaildir.discardc C s! | j | � } | j | � } | j | � } t | t � rE | } n | } t j j | � } | j | k r� | j | j | j � d } n d } | j | � t j j
| j | � } t j j
| j | | | � }
t | t � r
t j | t j j
| � | j � f � t j | |
� d S)z>Replace the keyed message; raise KeyError if it doesn't exist.rP r� NrS )r� r rT r r r r� r� r� r" rx r r� r� r� r� ) r r r Z old_subpathZtemp_keyZ temp_subpathZdominant_subpathr� r� Ztmp_path�new_pathr r r r# U s"
zMaildir.__setitem__c C s� | j | � } t t j j | j | � d � �. } | j rN | j | � } n t | � } Wd QRXt j j | � \ } } | j | � | j
| k r� | j | j | j
� d � | j t j j
t j j | j | � � � | S)z4Return a Message representation or raise a KeyError.�rbNrP rS )r� �openr r rx r r r r� �
set_subdirr� �set_info�set_date�getmtime)r r Zsubpath�f�msgr� r� r r r r' p s $
+zMaildir.get_messagec
C sN t t j j | j | j | � � d � � } | j � j t d � SWd QRXd S)z2Return a bytes representation or raise a KeyError.r� s
N) r� r r rx r r� rM r[ r\ )r r r� r r r r. s -zMaildir.get_bytesc C s4 t t j j | j | j | � � d � } t | � S)z6Return a file-like representation or raise a KeyError.r� )r� r r rx r r� �
_ProxyFile)r r r� r r r r* � s *zMaildir.get_filec c sP | j � x? | j D]4 } y | j | � Wn t k
rB w Yn X| Vq Wd S)z Return an iterator over keys.N)�_refreshr} r� r! )r r r r r r1 � s
zMaildir.iterkeysc C s | j � | | j k S)z9Return True if the keyed message exists, False otherwise.)r� r} )r r r r r r: � s
zMaildir.__contains__c C s | j � t | j � S)z*Return a count of messages in the mailbox.)r� �lenr} )r r r r r; � s
zMaildir.__len__c C s d S)z"Write any pending changes to disk.Nr )r r r r rD � s z
Maildir.flushc C s d S)zLock the mailbox.Nr )r r r r rE � s z Maildir.lockc C s d S)z#Unlock the mailbox if it is locked.Nr )r r r r rF � s zMaildir.unlockc C s d S)z Flush and close the mailbox.Nr )r r r r rG � s z
Maildir.closec C s� g } xw t j | j � D]c } t | � d k r | d d k r t j j t j j | j | � � r | j | d d � � q W| S)z Return a list of folder names.rP r �.N)r �listdirr r� r �isdirrx �append)r r= �entryr r r � list_folders� s "$zMaildir.list_foldersc C s/ t t j j | j d | � d | j d d �S)z/Return a Maildir instance for the named folder.r� r r F)r r r rx r r )r �folderr r r �
get_folder� s zMaildir.get_folderc C s� t j j | j d | � } t | d | j �} t j j | d � } t j j | � s~ t j t j | t j t j
Bd � � | S)z>Create a folder and return a Maildir instance representing it.r� r Z
maildirfolderi� ) r r rx r r r rz rG r� �O_CREAT�O_WRONLY)r r� r r= Zmaildirfolder_pathr r r �
add_folder� s
zMaildir.add_folderc C s� t j j | j d | � } xt t j t j j | d � � t j t j j | d � � D]8 } t | � d k s | d d k rW t d | � � qW Wxr t j | � D]a } | d k r� | d k r� | d k r� t j j t j j | | � � r� t d | | f � � q� Wx� t j | d d
�D]i \ } } } x* | D]" } t j t j j | | � � q4Wx* | D]" } t j
t j j | | � � qaWq Wt j
| � d S) z-Delete the named folder, which must be empty.r� rv rw rP r z Folder contains message(s): %sru z%Folder contains subdirectory '%s': %s�topdownFN) r r rx r r� r� �
NotEmptyErrorr� �walkr �rmdir)r r� r r� �root�dirsZfilesr r r �
remove_folder� s #"$!%
$zMaildir.remove_folderc C s� t j � } xm t j t j j | j d � � D]J } t j j | j d | � } | t j j | � d k r. t j | � q. Wd S)zDelete old files in "tmp".ru i@� N)�timer r� r rx r r� r )r �nowr� r r r r �clean� s
( z
Maildir.cleanrP c C s t j � } t j � } d | k r6 | j d d � } d | k rT | j d d � } d t | � t | d d � t j � t j | f } t j j
| j d | � } y t j | � WnD t
k
r� t j d 7_ y t | � SWn t k
r� Yn XYn Xt d | � � d
S) z=Create a file in the tmp subdirectory and open and return it.�/z\057rt z\072z%s.M%sP%sQ%s.%srP g ��.Aru z&Name clash prevented file creation: %sN)r� �socket� gethostnamer[ �intr �getpidr �_countr rx r �statr� �_create_carefully�FileExistsErrorr� )r r� Zhostnamer� r r r r r� � s& &
zMaildir._create_tmpc C s6 t j � | j d | j k r� d } xP | j D]E } t j j | j | � } | | j | k rh d } | | j | <q0 W| s� d Si | _ x� | j D]� } | j | } xs t j | � D]b } t j j
| | � } t j j | � r� q� | j | j
� d } t j j
| | � | j | <q� Wq� Wt j � | _ d S)z!Update table of contents mapping.rO FTNr )r� r r� r~ r r r� ry r} r� rx r� r� r� )r Zrefreshr� �mtimer r� �pr� r r r r� � s&
$zMaildir._refreshc C s� y: t j j t j j | j | j | � � r9 | j | SWn t k
rN Yn X| j � y | j | SWn" t k
r� t d | � � Yn Xd S)z=Use TOC to return subpath for given key, or raise a KeyError.zNo message with key: %sN)r r rz rx r r} r! r� )r r r r r r� ! s +
zMaildir._lookupc C sk t | d � s | j � | _ xF y | t | j � SWq! t k
rN d SYq! t k
rb w! Yq! Xq! Wd S)z0Return the next message in a one-time iteration.�
_onetime_keysN)r@ r1 r� �next�
StopIterationr! )r r r r r� / s
z Maildir.next) rp rq rr rs r� r r r r" r# r' r. r* r1 r: r; rD rE rF rG r� r� r� r� r� r� r� r� r� r� r r r r r s6 )
$ c @ s� e Z d Z d Z d d d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d d! � Z d" d# � Z d S)$�_singlefileMailboxzA single-file mailbox.NTc C s t j | | | | � y t | j d � } Wn� t k
r� } zw | j t j k r� | rn t | j d � } q� t | j � � n3 | j t j t j f k r� t | j d � } n � WYd d } ~ Xn X| | _
d | _ d | _ d | _
d | _ d | _ d | _ d S)z!Initialize a single-file mailbox.zrb+zwb+r� Nr F)r r r� r r� r� �ENOENTr| �EACCES�EROFS�_filer} � _next_key�_pending�
_pending_sync�_locked� _file_length)r r r r r� r� r r r r ? s$ z_singlefileMailbox.__init__c C sF | j � | j | � | j | j <| j d 7_ d | _ | j d S)z$Add message and return assigned key.rP T)r� �_append_messager} r� r� )r r r r r r V s
z_singlefileMailbox.addc C s$ | j | � | j | =d | _ d S)z=Remove the keyed message; raise KeyError if it doesn't exist.TN)r� r} r� )r r r r r r ` s
z_singlefileMailbox.removec C s0 | j | � | j | � | j | <d | _ d S)z>Replace the keyed message; raise KeyError if it doesn't exist.TN)r� r� r} r� )r r r r r r r# f s
z _singlefileMailbox.__setitem__c c s | j � | j j � Ed Hd S)z Return an iterator over keys.N)r� r} r3 )r r r r r1 l s
z_singlefileMailbox.iterkeysc C s | j � | | j k S)z9Return True if the keyed message exists, False otherwise.)r� r} )r r r r r r: q s
z _singlefileMailbox.__contains__c C s | j � t | j � S)z*Return a count of messages in the mailbox.)r� r� r} )r r r r r; v s
z_singlefileMailbox.__len__c C s# | j s t | j � d | _ d S)zLock the mailbox.TN)r� �
_lock_filer� )r r r r rE { s
z_singlefileMailbox.lockc C s# | j r t | j � d | _ d S)z#Unlock the mailbox if it is locked.FN)r� � _unlock_filer� )r r r r rF � s
z_singlefileMailbox.unlockc
C s� | j s, | j r( t | j � d | _ d S| j d k sA t � | j j d d � | j j � } | | j k r� t d | j | f � � t
| j � } y� i } | j | � x� t
| j j � � D]� } | j | \ } } | j j | � | j | � | j � } x@ | j j t d | | j j � � � } | s:P| j | � q W| | j � f | | <| j | � q� W| j � | _ Wn% | j � t j | j � � Yn Xt | � | j j � t j | j � j } t j | j | � y t j | j | j � Wn8 t k
r=t j | j � t j | j | j � Yn Xt | j d � | _ | | _ d | _ d | _ | j r�t | j d d �d S) z"Write any pending changes to disk.FNr rO z4Size of mailbox file changed (expected %i, found %i)i zrb+�dotlock)!r� r� � _sync_flushr� r} �AssertionErrorrZ �tellr� r� �_create_temporaryr �_pre_mailbox_hook�sortedr3 �_pre_message_hookrM �minr] �_post_message_hookrG r r r� r� r� �st_mode�chmodr� r� r� r� r� )
r Zcur_lenZnew_fileZnew_tocr �start�stopZ new_startrN �moder r r rD � s`
z_singlefileMailbox.flushc C s d S)z,Called before writing the mailbox to file f.Nr )r r� r r r r� � s z$_singlefileMailbox._pre_mailbox_hookc C s d S)z-Called before writing each message to file f.Nr )r r� r r r r� � s z$_singlefileMailbox._pre_message_hookc C s d S)z,Called after writing each message to file f.Nr )r r� r r r r� � s z%_singlefileMailbox._post_message_hookc C s> z | j � Wd z | j r' | j � Wd | j j � XXd S)z Flush and close the mailbox.N)rD r� rF r� rG )r r r r rG � s z_singlefileMailbox.closec C s] | j d k r | j � | d k rY y | j | SWn" t k
rX t d | � � Yn Xd S)z'Return (start, stop) or raise KeyError.NzNo message with key: %s)r} �
_generate_tocr! )r r r r r r� � s
z_singlefileMailbox._lookupc C s� | j j d d � | j j � } t | j � d k rQ | j rQ | j | j � y3 | j | j � | j | � } | j | j � Wn% t
k
r� | j j | � � Yn X| j j � | j j � | _
| S)z;Append message to mailbox and return (start, stop) offsets.r rO )r� rZ r� r� r} r� r� r� �_install_messager� r� �truncaterD r� )r r ZbeforeZoffsetsr r r r� � s
z"_singlefileMailbox._append_message)rp rq rr rs r r r r# r1 r: r; rE rF rD r� r� r� rG r� r� r r r r r� < s"
@
r� c @ sa e Z d Z d Z d Z d d � Z d d d � Z d d d � Z d d
d � Z d d
� Z d S)� _mboxMMDFzAn mbox or MMDF mailbox.Tc C s� | j | � \ } } | j j | � | j j � j t d � } | j j | | j j � � } | j | j t d � � } | j | d d � j
d � � | S)z4Return a Message representation or raise a KeyError.� s
rQ Nr ) r� r� rZ rg r[ r\ rM r� �_message_factory�set_from�decode)r r r� r� � from_line�stringr� r r r r' s z_mboxMMDF.get_messageFc C s" t j | j | � � j d | � S)z3Return a string representation or raise a KeyError.�unixfrom)r, r- r. r/ )r r �from_r r r r0 s z_mboxMMDF.get_stringc C sg | j | � \ } } | j j | � | s8 | j j � | j j | | j j � � } | j t d � S)z3Return a string representation or raise a KeyError.s
)r� r� rZ rg rM r� r[ r\ )r r r� r� r� r� r r r r. s
z_mboxMMDF.get_bytesc C sT | j | � \ } } | j j | � | s8 | j j � t | j | j j � | � S)z6Return a file-like representation or raise a KeyError.)r� r� rZ rg � _PartialFiler� )r r r� r� r� r r r r* s
z_mboxMMDF.get_filec C s� d } t | t � r$ | j | � } t | t � r� | j d � r� | j d � } | d k r� | d | � } | | d d � } q | } d } nm t | t � r� | j � j d � } d | } n< t | t j
j � r | j � } | d k r | j d � } | d k r+d t
j t
j � � j � } | j j � } | j j | t � | j | | j | j � | j j � } | | f S) z1Format a message and blindly write to self._file.Ns From s
rP r� r s From MAILER-DAEMON rS )rT r` rK ra rh �find�_mboxMMDFMessage�get_fromrH r, r r � get_unixfromr� �asctime�gmtimer� r� r] r\ ro �
_mangle_from_)r r r� �newlineZauthorr� r� r r r r� $ s0
z_mboxMMDF._install_messageN)
rp rq rr rs r� r' r0 r. r* r� r r r r r� � s
r� c @ sL e Z d Z d Z d Z d Z d d d d � Z d d � Z d d � Z d S)
r zA classic mbox mailbox.TNc C s# t | _ t j | | | | � d S)zInitialize an mbox mailbox.N)r
r� r� r )r r r r r r r r J s z
mbox.__init__c C s | j t � d S)z,Called after writing each message to file f.N)r] r\ )r r� r r r r� O s zmbox._post_message_hookc C sJ g g } } d } | j j d � x� | j j � } | j j � } | j d � r� t | � t | � k r� | r� | j | t t � � n
| j | � | j | � d } q& | s� | r� | j | t t � � n
| j | � Pq& | t k r� d } q& d } q& Wt t t
| | � � � | _ t | j � | _ | j j � | _
d S)z0Generate key-to-(start, stop) table of contents.Fr s From TN)r� rZ r� rg rh r� r� r\ �dict� enumerate�zipr} r� r� )r �starts�stopsZlast_was_empty�line_posrn r r r r� S s0
zmbox._generate_toc) rp rq rr rs r� r^ r r� r� r r r r r A s c @ sL e Z d Z d Z d d d d � Z d d � Z d d � Z d
d � Z d S) r zAn MMDF mailbox.NTc C s# t | _ t j | | | | � d S)zInitialize an MMDF mailbox.N)r
r� r� r )r r r r r r r r x s z
MMDF.__init__c C s | j d t � d S)z-Called before writing each message to file f.s N)r] r\ )r r� r r r r� } s zMMDF._pre_message_hookc C s | j t d t � d S)z,Called after writing each message to file f.s N)r] r\ )r r� r r r r� � s zMMDF._post_message_hookc C s; g g } } | j j d � d } x� | } | j j � } | j j � } | j d t � r� | j | � xq | } | j j � } | j j � } | d t k r� | j | t t � � Pqm | sm | j | � Pqm Wq& | s& Pq& Wt t t
| | � � � | _ t | j � | _ | j j d d � | j j � | _
d S)z0Generate key-to-(start, stop) table of contents.r s rO N)r� rZ rg r� rh r\ r� r� r� r� r� r} r� r� )r r� r� �next_posr rn r r r r� � s2
zMMDF._generate_toc)rp rq rr rs r r� r� r� r r r r r u s
c @ s$ e Z d Z d Z d d d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d. d/ � Z d S)0r zAn MH mailbox.NTc C s� t j | | | | � t j j | j � s� | r� t j | j d � t j t j t j j | j d � t j
t j Bt j Bd � � n t
| j � � d | _ d S)zInitialize an MH instance.i� z
.mh_sequencesi� FN)r r r r rz r r{ rG r� rx r� �O_EXCLr� r| r� )r r r r r r r r � s !!z MH.__init__c C s, | j � } t | � d k r' d } n t | � d } t j j | j t | � � } t | � } d } z� | j r} t
| � z y | j | | � WnE t k
r� | j r� t
| � t | � d } t j | � � Yn Xt | t � r� | j | | � Wd | j rt
| � XWd | s't | � X| S)z$Add message and return assigned key.r rP FTN)r3 r� �maxr r rx r r` r� r� r� ro r� r� r� r rT r �_dump_sequences)r r r3 Znew_keyr� r� �closedr r r r � s6
zMH.addc C s� t j j | j t | � � } y t | d � } WnM t k
r� } z- | j t j k rk t d | � � n � WYd d } ~ Xn X| j
� t j | � d S)z=Remove the keyed message; raise KeyError if it doesn't exist.zrb+zNo message with key: %sN) r r rx r r` r� r� r� r� r! rG r )r r r r� r� r r r r � s
z MH.removec C s t j j | j t | � � } y t | d � } WnM t k
r� } z- | j t j k rk t d | � � n � WYd d } ~ Xn Xz� | j
r� t | � zV t j t j | t j
t j B� � | j | | � t | t � r� | j | | � Wd | j
rt | � XWd t | � Xd S)z>Replace the keyed message; raise KeyError if it doesn't exist.zrb+zNo message with key: %sN)r r rx r r` r� r� r� r� r! r� r� rG r� �O_TRUNCro rT r r r� r� )r r r r r� r� r r r r# � s$
# zMH.__setitem__c C s6 y^ | j r6 t t j j | j t | � � d � } n' t t j j | j t | � � d � } WnM t k
r� } z- | j t j k r� t
d | � � n � WYd d } ~ Xn X| �? | j r� t | � z t | � } Wd | j r� t
| � XWd QRXx9 | j � j � D]% \ } } | | k r | j | � q W| S)z4Return a Message representation or raise a KeyError.zrb+r� zNo message with key: %sN)r� r� r r rx r r` r� r� r� r! r� r r� �
get_sequencesr9 � add_sequence)r r r� r� r� r� �key_listr r r r' � s& *+
zMH.get_messagec C s y^ | j r6 t t j j | j t | � � d � } n' t t j j | j t | � � d � } WnM t k
r� } z- | j t j k r� t
d | � � n � WYd d } ~ Xn X| �I | j r� t | � z | j � j
t d � SWd | j r� t | � XWd QRXd S)z2Return a bytes representation or raise a KeyError.zrb+r� zNo message with key: %sNs
)r� r� r r rx r r` r� r� r� r! r� rM r[ r\ r� )r r r� r� r r r r. s *+
z MH.get_bytesc C s� y+ t t j j | j t | � � d � } WnM t k
rz } z- | j t j k re t d | � � n � WYd d } ~ Xn Xt
| � S)z6Return a file-like representation or raise a KeyError.r� zNo message with key: %sN) r� r r rx r r` r� r� r� r! r� )r r r� r� r r r r* ' s +z MH.get_filec C s) t t d d � t j | j � D� � � S)z Return an iterator over keys.c s s' | ] } | j � r t | � Vq d S)N)�isdigitr� )�.0r� r r r � <genexpr>4 s z MH.iterkeys.<locals>.<genexpr>)�iterr� r r� r )r r r r r1 2 s z MH.iterkeysc C s( t j j t j j | j t | � � � S)z9Return True if the keyed message exists, False otherwise.)r r rz rx r r` )r r r r r r: 7 s zMH.__contains__c C s t t | j � � � S)z*Return a count of messages in the mailbox.)r� r2 r1 )r r r r r; ; s z
MH.__len__c C sG | j sC t t j j | j d � d � | _ t | j � d | _ d S)zLock the mailbox.z
.mh_sequenceszrb+TN)r� r� r r rx r r� r� )r r r r rE ? s $
zMH.lockc C s6 | j r2 t | j � t | j � | ` d | _ d S)z#Unlock the mailbox if it is locked.FN)r� r� r� r� )r r r r rF F s
z MH.unlockc C s d S)z&Write any pending changes to the disk.Nr )r r r r rD N s zMH.flushc C s | j r | j � d S)z Flush and close the mailbox.N)r� rF )r r r r rG R s zMH.closec C sX g } xK t j | j � D]7 } t j j t j j | j | � � r | j | � q W| S)z Return a list of folder names.)r r� r r r� rx r� )r r= r� r r r r� W s
$zMH.list_foldersc C s+ t t j j | j | � d | j d d �S)z+Return an MH instance for the named folder.r r F)r r r rx r r )r r� r r r r� _ s z
MH.get_folderc C s% t t j j | j | � d | j �S)z:Create a folder and return an MH instance representing it.r )r r r rx r r )r r� r r r r� d s z
MH.add_folderc C s� t j j | j | � } t j | � } | d g k rU t j t j j | d � � n" | g k rd n t d | j � � t j | � d S)z-Delete the named folder, which must be empty.z
.mh_sequenceszFolder not empty: %sN)r r rx r r� r r� r� )r r� r �entriesr r r r� i s zMH.remove_folderc
s` i } t t j j | j d � d d d ��,} t | j � � � x| D]} y� | j d � \ } } t � } xr | j � D]d } | j � r� | j t
| � � q} d d � | j d � D� \ } } | j t | | d � � q} W� f d
d � t
| � D� | | <t | | � d k r%| | =WqI t k
rPt d
| j � � � YqI XqI WWd QRX| S)z=Return a name-to-key-list dictionary to define each sequence.z
.mh_sequences�r�encoding�ASCIIrt c s s | ] } t | � Vq d S)N)r� )r �xr r r r � s z#MH.get_sequences.<locals>.<genexpr>�-rP c s" g | ] } | � k r | � q Sr r )r r )�all_keysr r �
<listcomp>� s z$MH.get_sequences.<locals>.<listcomp>r z"Invalid sequence specification: %sN)r� r r rx r �setr3 r� r
r r� rC �ranger� r� rJ � FormatError�rstrip)
r Zresultsr� rn r� �contentsr3 �specr� r� r )r r r u s& *
" #
zMH.get_sequencesc C se t t j j | j d � d d d �} z,t j t j | j t j t j B� � x� | j � D]� \ } } t
| � d k r~ q] | j | d � d } d } x� t t
| � � D]o } | d | k r� | sd
} | j d � n7 | rd } | j d | | f � n | j d
| � | } q� W| rA| j t | � d � q] | j d � q] WWd t | � Xd S)z:Set sequences using the given name-to-key-list dictionary.z
.mh_sequenceszr+r r r rt NFrP Tr z%s %sz %s�
)r� r r rx r rG r� r� r r9 r� r] r� r r` r� )r � sequencesr� r� r3 �prevZ
completingr r r r �
set_sequences� s. '&
zMH.set_sequencesc C s� | j � } d } g } x| j � D]� } | d | k r| j | | d f � t t d � r� t j t j j | j t | � � t j j | j t | d � � � t j
t j j | j t | � � � nD t j t j j | j t | � � t j j | j t | d � � � | d 7} q% W| d | _ t
| � d k rBd SxP | j � D]B \ } } x3 | D]+ \ } } | | k rb| | | j | � <qbWqOW| j | � d S)z?Re-name messages to eliminate numbering gaps. Invalidates keys.r rP r� N)r r1 r� r@ r r� r rx r r` �unlinkr� r� r� r9 �indexr ) r r r Zchangesr r� r �oldrv r r r �pack� s* !#(!#
zMH.packc C s� | j � } | j � } xR | j � D]D \ } } | | k rM | j | � q% | | k r% | | j | � =q% Wx' | D] } | | k rt | g | | <qt W| j | � d S)z;Inspect a new MHMessage and update sequences appropriately.N)r r9 r� r! r )r r r Zpending_sequencesZ
all_sequencesr� r �sequencer r r r � s
zMH._dump_sequences)rp rq rr rs r r r r# r' r. r* r1 r: r; rE rF rD rG r� r� r� r� r r r# r r r r r r � s. " c @ s� e Z d Z d Z e d d d d d d d h � Z d d
d d � Z d
d � Z d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d S)%r z An Rmail-style Babyl mailbox.�unseen�deletedZfiled�answered� forwardedZedited�resentNTc C s# t j | | | | � i | _ d S)zInitialize a Babyl mailbox.N)r� r �_labels)r r r r r r r r � s zBabyl.__init__c C s8 t j | | � } t | t � r4 | j � | j | <| S)z$Add message and return assigned key.)r� r rT r �
get_labelsr* )r r r r r r r � s z Babyl.addc C s- t j | | � | | j k r) | j | =d S)z=Remove the keyed message; raise KeyError if it doesn't exist.N)r� r r* )r r r r r r � s z Babyl.removec C s9 t j | | | � t | t � r5 | j � | j | <d S)z>Replace the keyed message; raise KeyError if it doesn't exist.N)r� r# rT r r+ r* )r r r r r r r# � s zBabyl.__setitem__c
C sm | j | � \ } } | j j | � | j j � t j � } xD | j j � } | d t k sg | rh P| j | j t d � � qA Wt j � } x@ | j j � } | t k s� | r� P| j | j t d � � q� W| | j j � } | d k s� t
� | j j | � } | j t d � } t | j
� | � } | j | j
� � | | j k ri| j | j | � | S)z4Return a Message representation or raise a KeyError.s *** EOOH ***s
r )r� r� rZ rg rU rV r\ r] r[ r� r� rM r rf � set_visibler* �
set_labels)
r r r� r� �original_headersrn Zvisible_headers�nZbodyr� r r r r' � s0
zBabyl.get_messagec C s | j | � \ } } | j j | � | j j � t j � } xD | j j � } | d t k sg | rh P| j | j t d � � qA Wx' | j j � } | t k s� | r� Pq� W| j � } | | j j
� } | d k s� t � | j j | � } | j t d � } | | S)z3Return a string representation or raise a KeyError.s *** EOOH ***s
r )
r� r� rZ rg rU rV r\ r] r[ rf r� r� rM ) r r r� r� r. rn Zheadersr/ rm r r r r. s&
zBabyl.get_bytesc C s" t j | j | � j d t � � S)z6Return a file-like representation or raise a KeyError.s
)rU rV r. r[ r\ )r r r r r r* " s zBabyl.get_filec C sT | j � t � } x$ | j j � D] } | j | � q# W| j | j � t | � S)z4Return a list of user-defined labels in the mailbox.)r� r r* r7 rC �difference_update�_special_labelsr2 )r �labelsZ
label_listr r r r+ & s
zBabyl.get_labelsc C s� g g } } | j j d � d } g } x | } | j j � } | j j � } | d t k r� t | � t | � k r� | j | t t � � | j | � d d � | j j � d d � j d � D� } | j | � q, | d k s� | d t k r)t | � t | � k rG| j | t t � � q, | s, | j | t t � � Pq, Wt t t
| | � � � | _ t t | � � | _ t | j � | _
| j j d d � | j j � | _ d S)
z0Generate key-to-(start, stop) table of contents.r s c S s( g | ] } | j � r | j � � q Sr )�strip)r �labelr r r r = s z'Babyl._generate_toc.<locals>.<listcomp>rP N� ,s rO )r� rZ rg r� r\ r� r� r� r� r� r� r} r* r� r� )r r� r� r Z label_listsr rn r2 r r r r� / s4
& zBabyl._generate_tocc C sm d t } | d t 7} | j � } d d � | D� } | d d j | � t 7} | d 7} | j | � d S) z,Called before writing the mailbox to file f.s BABYL OPTIONS:s
Version: 5c s s | ] } | j � Vq d S)N)rH )r r4 r r r r R s z*Babyl._pre_mailbox_hook.<locals>.<genexpr>s Labels:r5 s N)r\ r+ rx r] )r r� Zbabylr2 r r r r� M s
zBabyl._pre_mailbox_hookc C s | j d t � d S)z-Called before writing each message to file f.s N)r] r\ )r r� r r r r� W s zBabyl._pre_message_hookc C s | j t d � d S)z,Called after writing each message to file f.s N)r] r\ )r r� r r r r� [ s zBabyl._post_message_hookc C s� | j j � } t | t � r� g } g } x@ | j � D]2 } | | j k r\ | j | � q7 | j | � q7 W| j j d � x( | D] } | j j d | j � � q� W| j j d � x, | D]$ } | j j d | j � d � q� W| j j t � n | j j d t � t | t
j j � r�t
j � } t
j j | d d � } | j | � | j d � x@ | j � } | j j | j d t � � | d k s�| rdPqdW| j j d
t � t | t � rDt
j � } t
j j | d d � }
|
j | j � � x� | j � } | j j | j d t � � | d k s<| rPqWnP | j d � x@ | j � } | j j | j d t � � | d k s�| rTPqTWx@| j d � } | s�P| j j | j d t � � q�Wnt | t t t
j f � r t | t
j � r t j d t d
� | j � } t | t � r:| j | � } | j d � d } | d d k r�| j j | d | � j d t � � | j j d
t � | j j | d | � j d t � � | j j | | d � j d t � � q�| j j d
t t � | j j | j d t � � n�t | d � r�t | d � rWt j d t d
� | j! } | j � }
d } x� | j � } | j" d � r�| d d � d } n# | j" d � r�| d d � d } | j j | j d t � � | d k s�| rl| r d } | j j d
t � | j |
� qlPqlWx� | j � } | s;P| j" d � ra| d d � t } nI | j" d � r�| d d � t } n# | j" d � r�| d d � t } | j j | � q(Wn t# d t$ | � � � | j j � } | | f S) z0Write message contents and return (start, stop).� 1s , s ,,� r5 s 1,,Fr s
s *** EOOH ***i z8Use of StringIO input is deprecated, use BytesIO insteadrL s
rO rP Nrg rN zDUse of text mode files is deprecated, use a binary mode file insteadTs
s
zInvalid message type: %srS rR rS rR rS rS )%r� r� rT r r+ r1 r� r] rH r\ r, r r rU rV rW rX rY rZ rg r[ � get_visiblerM ra r` rb rc rd re rf rK r� r@ rN r_ ri rj )r r r� Zspecial_labelsr2 r4 Z orig_bufferZorig_generatorrn Z
vis_bufferZ
vis_generatorrN Z
body_startZ original_posZ
first_passr� r r r r� _ s�
"
#
&&)
zBabyl._install_message)rp rq rr rs � frozensetr1 r r r r# r' r. r* r+ r� r� r� r� r� r r r r r � s
c @ s= e Z d Z d Z d d d � Z d d � Z d d � Z d S) r z0Message with mailbox-format-specific properties.Nc C s) t | t j j � rJ | j t j | � � t | t � r%| j | � n� t | t � rr | j t j | � � n� t | t
� r� | j t j | � � n� t | t j
� r� | j t j | � � n` t | d � r� | j t j | � � n8 | d k rt j j j | � n t d t | � � � d S)z Initialize a Message instance.rM NzInvalid message type: %s)rT r, r r �_become_message�copy�deepcopy� _explain_tora r- r` Zmessage_from_stringrU �
TextIOWrapperZmessage_from_filer@ Zmessage_from_binary_filer ri rj )r r r r r r � s zMessage.__init__c C sJ t | d g � } x1 | j D]& } | | k r | j | | j | <q Wd S)z0Assume the non-format-specific state of message.�_type_specific_attributesN)�getattr�__dict__)r r Z
type_specificr� r r r r: � s zMessage._become_messagec C s# t | t � r d St d � � d S)z:Copy format-specific state to message insofar as possible.Nz Cannot convert to specified type)rT r ri )r r r r r r= � s zMessage._explain_to)rp rq rr rs r r: r= r r r r r � s c @ s� e Z d Z d Z d d d g Z d d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d S) r z)Message with Maildir-specific properties.�_subdir�_info�_dateNc C s5 d | _ d | _ t j � | _ t j | | � d S)z%Initialize a MaildirMessage instance.rv r� N)rB rC r� rD r r )r r r r r r � s zMaildirMessage.__init__c C s | j S)zReturn 'new' or 'cur'.)rB )r r r r r� � s zMaildirMessage.get_subdirc C s8 | d k s | d k r$ | | _ n t d | � � d S)z Set subdir to 'new' or 'cur'.rv rw z!subdir must be 'new' or 'cur': %sN)rB rJ )r r� r r r r� s zMaildirMessage.set_subdirc C s+ | j j d � r# | j d d � Sd Sd S)z*Return as a string the flags that are set.z2,rO Nr� )rC rh )r r r r � get_flags s zMaildirMessage.get_flagsc C s d d j t | � � | _ d S)z)Set the given flags and unset all others.z2,r� N)rx r� rC )r �flagsr r r � set_flags s zMaildirMessage.set_flagsc C s0 | j d j t | j � � t | � B� � d S)z.Set the given flag(s) without changing others.r� N)rG rx r rE )r �flagr r r �add_flag s zMaildirMessage.add_flagc C s< | j � r8 | j d j t | j � � t | � � � d S)z7Unset the given string flag(s) without changing others.r� N)rE rG rx r )r rH r r r � remove_flag s zMaildirMessage.remove_flagc C s | j S)z<Return delivery date of message, in seconds since the epoch.)rD )r r r r r� s zMaildirMessage.get_datec C s< y t | � | _ Wn" t k
r7 t d | � � Yn Xd S)z9Set delivery date of message, in seconds since the epoch.zcan't convert to float: %sN)�floatrD rJ ri )r Zdater r r r� s
zMaildirMessage.set_datec C s | j S)z%Get the message's "info" as a string.)rC )r r r r r� & s zMaildirMessage.get_infoc C s5 t | t � r | | _ n t d t | � � � d S)z Set the message's "info" string.zinfo must be a string: %sN)rT r` rC ri rj )r �infor r r r� * s zMaildirMessage.set_infoc C s4 t | t � rK | j | j � � | j | j � � | j | j � � n�t | t � rt | j � � } d | k r� | j
d � | j � d k r� | j
d � d | k r� | j
d � d | k r� | j
d � d | k r� | j
d � | j d t j
| j � � � n t | t � r�t | j � � } d | k rK| j d
� d | k rd| j d � d | k r0| j d � n� t | t � rt | j � � } d | k r�| j d
� d | k r�| j d
� d | k r�| j d � d | k r0| j d � n( t | t � rn t d t | � � � d S)z;Copy Maildir-specific state to message insofar as possible.�S�Rrw �O�T�D�F�Az
MAILER-DAEMONr% �replied�flaggedr& r' �Pr( z$Cannot convert to specified type: %sN)rT r rG rE r� r� r� r� r� r rI r� r� r� r r r � add_labelr ri rj )r r rF r r r r= 1 sN
"
zMaildirMessage._explain_to)rp rq rr rs r? r r� r� rE rG rI rJ r� r� r� r� r= r r r r r � s c @ s� e Z d Z d Z d g Z d d d � Z d d � Z d d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d S)r� z/Message with mbox- or MMDF-specific properties.�_fromNc C sw | j d d � t | t j j � rc | j � } | d k rc | j d � rc | j | d d � � t j | | � d S)z'Initialize an mboxMMDFMessage instance.z
MAILER-DAEMONTNzFrom rQ )r� rT r, r r r� rh r )r r r� r r r r b s z_mboxMMDFMessage.__init__c C s | j S)z Return contents of "From " line.)rX )r r r r r� k s z_mboxMMDFMessage.get_fromc C sH | d k r; | d k r$ t j � } | d t j | � 7} | | _ d S)z>Set "From " line, formatting and appending time_ if specified.NT� )r� r� r� rX )r r� Ztime_r r r r� o s
z_mboxMMDFMessage.set_fromc C s | j d d � | j d d � S)z*Return as a string the flags that are set.�Statusr� zX-Status)r& )r r r r rE w s z_mboxMMDFMessage.get_flagsc C s t | � } d
\ } } x1 d D]) } | | k r | | 7} | j | � q Wx1 d D]) } | | k rS | | 7} | j | � qS W| d j t | � � 7} y | j d | � Wn" t k
r� | j d | � Yn Xy | j d | � Wn" t k
r
| j d | � Yn Xd S)
z)Set the given flags and unset all others.r� rN rO rQ rR rS rZ zX-StatusN)r� r� )rN rO )rQ rR rS )r r rx r� �replace_headerr! Z
add_header)r rF Z status_flagsZ
xstatus_flagsrH r r r rG { s&
z_mboxMMDFMessage.set_flagsc C s0 | j d j t | j � � t | � B� � d S)z.Set the given flag(s) without changing others.r� N)rG rx r rE )r rH r r r rI � s z_mboxMMDFMessage.add_flagc C sH d | k s d | k rD | j d j t | j � � t | � � � d S)z7Unset the given string flag(s) without changing others.rZ zX-Statusr� N)rG rx r rE )r rH r r r rJ � s z _mboxMMDFMessage.remove_flagc C ss t | t � rt | j � � } d | k r: | j d � d | k rS | j d � d | k rl | j d � d | k r� | j d � d | k r� | j d � | d =| d
=d j | j � j � d d
� � } y& | j t
j t j
| d � � � Wqot t f k
rYqoXnZt | t � rM| j | j � � | j | j � � n"t | t � r�t | j � � } d | k r�| j d � d | k r�| j d � d | k r�| j d � | d =| d
=n� t | t � rGt | j � � } d | k r| j d � d | k r | j d � d | k r6| j d � | d =| d
=n( t | t � rYn t d t | � � � d
S)zACopy mbox- or MMDF-specific state to message insofar as possible.rO rw rR rS rN rM rQ rP Zstatuszx-statusrY rQ Nz%a %b %d %H:%M:%S %Yr% rT rU r& r' z$Cannot convert to specified type: %s���)rT r r rE r� rI rx r� r� r� �calendarZtimegmr� ZstrptimerJ �
OverflowErrorr� rG r� r r r rW r ri rj )r r rF Z
maybe_dater r r r= � s^
%
z _mboxMMDFMessage._explain_to)
rp rq rr rs r? r r� r� rE rG rI rJ r= r r r r r� ] s r� c @ s e Z d Z d Z d S)r
z&Message with mbox-specific properties.N)rp rq rr rs r r r r r
� s c @ sj e Z d Z d Z d g Z d d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d S)r z$Message with MH-specific properties.�
_sequencesNc C s g | _ t j | | � d S)z!Initialize an MHMessage instance.N)r_ r r )r r r r r r � s zMHMessage.__init__c C s | j d d � S)z4Return a list of sequences that include the message.N)r_ )r r r r r � s zMHMessage.get_sequencesc C s t | � | _ d S)z3Set the list of sequences that include the message.N)r2 r_ )r r r r r r � s zMHMessage.set_sequencesc C sK t | t � r1 | | j k rG | j j | � n t d t | � � � d S)z8Add sequence to list of sequences including the message.z sequence type must be str: %sN)rT r` r_ r� ri rj )r r$ r r r r � s zMHMessage.add_sequencec C s- y | j j | � Wn t k
r( Yn Xd S)zARemove sequence from the list of sequences including the message.N)r_ r rJ )r r$ r r r �remove_sequence� s
zMHMessage.remove_sequencec C s� t | t � r� t | j � � } d | k r= | j d � n | j d � | j d � d | k rp | j d � d | k r�| j d � n3t | t � r t | j � � } d | k r� | j d � n
| j d � d | k r� | j d � d | k r�| j d
� n� t | t � rAx� | j � D] } | j | � q'Wn~ t | t � r�t | j � � } d | k r{| j
d � d | k r�| j
d � n( t | t � r�n t d t
| � � � d
S)z6Copy MH-specific state to message insofar as possible.r% rw rM rU rR rT rN �ROrO rS r' z$Cannot convert to specified type: %sN)rT r r r r� rI r� r r r rW r ri rj )r r r r$ r r r r= � s@
zMHMessage._explain_to) rp rq rr rs r? r r r r r` r= r r r r r � s c @ s� e Z d Z d Z d d g Z d d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d S)r z'Message with Babyl-specific properties.r* �_visibleNc C s) g | _ t � | _ t j | | � d S)z#Initialize a BabylMessage instance.N)r* r rb r )r r r r r r s zBabylMessage.__init__c C s | j d d � S)z'Return a list of labels on the message.N)r* )r r r r r+ % s zBabylMessage.get_labelsc C s t | � | _ d S)z&Set the list of labels on the message.N)r2 r* )r r2 r r r r- ) s zBabylMessage.set_labelsc C sK t | t � r1 | | j k rG | j j | � n t d t | � � � d S)z+Add label to list of labels on the message.zlabel must be a string: %sN)rT r` r* r� ri rj )r r4 r r r rW - s zBabylMessage.add_labelc C s- y | j j | � Wn t k
r( Yn Xd S)z4Remove label from the list of labels on the message.N)r* r rJ )r r4 r r r � remove_label5 s
zBabylMessage.remove_labelc C s
t | j � S)z3Return a Message representation of visible headers.)r rb )r r r r r8 < s zBabylMessage.get_visiblec C s t | � | _ d S)z2Set the Message representation of visible headers.N)r rb )r Zvisibler r r r, @ s zBabylMessage.set_visiblec C s� xG | j j � D]6 } | | k r< | j j | | | � q | j | =q Wx: d D]2 } | | k rQ | | j k rQ | | | j | <qQ Wd S) z9Update and/or sensibly generate a set of visible headers.�Date�From�Reply-To�To�CC�SubjectN)rd re rf rg rh ri )rb r3 r[ )r �headerr r r �update_visibleD s
zBabylMessage.update_visiblec C s� t | t � r� t | j � � } d | k r= | j d � n | j d � | j d � d | k so d | k r| | j d � d | k r� | j d � d | k r�| j d
� nFt | t � r0t | j � � } d | k r� | j d � n
| j d � d | k r| j d
� d | k r�| j d � n� t | t � r�t | j � � } d | k rj| j d � d | k r�| j d � nq t | t � r�| j
| j � � xL | j � D] } | j | � q�Wn( t | t
� r�n t d t | � � � d S)z9Copy Babyl-specific state to message insofar as possible.r% rw rM r( r) rV r' rN r& rP ra rO rQ rS rT z$Cannot convert to specified type: %sN)rT r r r+ r� rI r� r r r r, r8 rW r ri rj )r r r2 r4 r r r r= O sF
zBabylMessage._explain_to)rp rq rr rs r? r r+ r- rW rc r8 r, rk r= r r r r r s c @ s e Z d Z d Z d S)r
z&Message with MMDF-specific properties.N)rp rq rr rs r r r r r
y s c @ s� e Z d Z d Z d d d � Z d d d � Z d d d � Z d d d
� Z d d d � Z d
d � Z d d � Z
d d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z e d$ d% � � Z d S)&r� z A read-only wrapper of a file.Nc C s4 | | _ | d k r' | j � | _ n | | _ d S)zInitialize a _ProxyFile.N)r� r� �_pos)r r� �posr r r r � s z_ProxyFile.__init__c C s | j | | j j � S)z Read bytes.)�_readr� rM )r �sizer r r rM � s z_ProxyFile.readc C s | j | | j j � S)z Read bytes.)rn r� �read1)r ro r r r rp � s z_ProxyFile.read1c C s | j | | j j � S)z Read a line.)rn r� rg )r ro r r r rg � s z_ProxyFile.readlinec C sQ g } xD | D]< } | j | � | d k r
| t | � 8} | d k r
Pq
W| S)zRead multiple lines.Nr )r� r� )r �sizehintr= rn r r r � readlines� s
z_ProxyFile.readlinesc c s& x | j � } | s d S| Vq Wd S)zIterate over lines.N)rg )r rn r r r r6 � s
z_ProxyFile.__iter__c C s | j S)zReturn the position.)rl )r r r r r� � s z_ProxyFile.tellr c C sH | d k r | j j | j � | j j | | � | j j � | _ d S)zChange position.rP N)r� rZ rl r� )r �offset�whencer r r rZ � s z_ProxyFile.seekc
C s@ t | d � r< z# t | j d � r1 | j j � Wd | ` Xd S)zClose the file.r� rG N)r@ r� rG )r r r r rG � s
z_ProxyFile.closec C sG | d k r d } | j j | j � | | � } | j j � | _ | S)z"Read size bytes using read_method.NrP rS )r� rZ rl r� )r ro � read_methodr= r r r rn � s z_ProxyFile._readc C s | S)z$Context management protocol support.r )r r r r � __enter__� s z_ProxyFile.__enter__c G s | j � d S)N)rG )r �excr r r �__exit__� s z_ProxyFile.__exit__c C s
| j j � S)N)r� �readable)r r r r ry � s z_ProxyFile.readablec C s
| j j � S)N)r� �writable)r r r r rz � s z_ProxyFile.writablec C s
| j j � S)N)r� �seekable)r r r r r{ � s z_ProxyFile.seekablec C s
| j j � S)N)r� rD )r r r r rD � s z_ProxyFile.flushc C s3 t | d � s d St | j d � s) d S| j j S)Nr� Tr F)r@ r� r )r r r r r � s
z_ProxyFile.closed)rp rq rr rs r rM rp rg rr r6 r� rZ rG rn rv rx ry rz r{ rD �propertyr r r r r r� } s$ r� c @ s[ e Z d Z d Z d d d d � Z d d � Z d d d � Z d
d � Z d d
� Z d S)r� z&A read-only wrapper of part of a file.Nc C s) t j | | | � | | _ | | _ d S)zInitialize a _PartialFile.N)r� r �_start�_stop)r r� r� r� r r r r � s z_PartialFile.__init__c C s t j | � | j S)z*Return the position with respect to start.)r� r� r} )r r r r r� � s z_PartialFile.tellr c C sV | d k r! | j | _ d } n | d k r? | j | _ d } t j | | | � d S)z8Change position, possibly with respect to start or stop.r rP rO N)r} rl r~ r� rZ )r rs rt r r r rZ � s z_PartialFile.seekc C s] | j | j } | d k r d S| d k sD | d k sD | | k rJ | } t j | | | � S)z;Read size bytes using read_method, honoring start and stop.r r� N)r~ rl r� rn )r ro ru Z remainingr r r rn � s $z_PartialFile._readc C s t | d � r | ` d S)Nr� )r@ r� )r r r r rG s z_PartialFile.close) rp rq rr rs r r� rZ rn rG r r r r r� � s
r� Tc - C s� d } y�t r� y t j | t j t j B� Wn_ t k
r� } z? | j t j t j t j f k ry t d | j
� � n � WYd d } ~ Xn X| r�y! t | j
d � } | j � WnG t k
r� } z' | j t j t j f k r� d S� WYd d } ~ Xn Xyf t
t d � rEt j | j
| j
d � d } t j | j
� n t j | j
| j
d � d } Wn5 t k
r�t j | j
� t d | j
� � Yn XWn> t r�t j | t j � | r�t j | j
d � � Yn Xd S)z(Lock file f using lockf and dot locking.Fzlockf: lock unavailable: %sNz.lockr� Tzdot lock unavailable: %s)�fcntl�lockfZLOCK_EXZLOCK_NBr� r� ZEAGAINr� r� r� r� r� rG r@ r r� r r� r� r �LOCK_UN)r� r� Z dotlock_doner� Zpre_lockr r r r� sF !
r� c C sJ t r t j | t j � t j j | j d � rF t j | j d � d S)z*Unlock file f using lockf and dot locking.z.lockN)r r� r� r r rz r� r )r� r r r r� 0 s r� c C sL t j | t j t j Bt j Bd � } z t | d � SWd t j | � Xd S)zCCreate a file if it doesn't exist and open for reading and writing.i� zrb+N)r r� r� r �O_RDWRrG )r �fdr r r r� 7 s &r� c C s2 t d | t t j � � t j � t j � f � S)zBCreate a temp file based on path and open for reading and writing.z %s.%s.%s.%s)r� r� r� r� r� r r� )r r r r r� ? s r� c C s0 | j � t t d � r, t j | j � � d S)z0Ensure changes to file f are physically on disk.�fsyncN)rD r@ r r� �fileno)r� r r r r� E s
r� c C s t | � | j � d S)z:Close file f, ensuring all changes are physically on disk.N)r� rG )r� r r r r� K s
r� c @ s e Z d Z d Z d S)�Errorz"Raised for module-specific errors.N)rp rq rr rs r r r r r� Q s r� c @ s e Z d Z d Z d S)r| z:The specified mailbox does not exist and won't be created.N)rp rq rr rs r r r r r| T s r| c @ s e Z d Z d Z d S)r� z>The specified mailbox is not empty and deletion was requested.N)rp rq rr rs r r r r r� W s r� c @ s e Z d Z d Z d S)r� z)Another process caused an action to fail.N)rp rq rr rs r r r r r� Z s r� c @ s e Z d Z d Z d S)r z)A file appears to have an invalid format.N)rp rq rr rs r r r r r ] s r )0rs r r� r] r� r� r; rc r, Z
email.messageZemail.generatorrU r( r � ImportError�__all__r\ rH r r r� r� r r r r r r r r� r
r r r
r� r� r� r� r� r� r� r� � Exceptionr� r| r� r� r r r r r �<module> sd
�� 6�B4-� 1� %mqH_c')