o i]@sJdZddlZddlZddlmZddlmZmZmZddl m Z ddl m Z ddl mZmZdd Zd d d ZGd d d Zee GdddeZee jGdddZee GdddeZGdddeZee jGdddZGdddejZee GdddeZee jGdddZee GdddeZdS)!zD Support for aliases(5) configuration files. @author: Jp Calderone N) implementer)deferprotocolreactor)smtp)IAlias)failurelogc Cspdd|ddD}t|dkr d}||f}t||dS|\}}||gttj|ddS) a Parse a line from an aliases file. @type result: L{dict} mapping L{bytes} to L{list} of L{bytes} @param result: A dictionary mapping username to aliases to which the results of parsing the line are added. @type line: L{bytes} @param line: A line from an aliases file. @type filename: L{bytes} @param filename: The full or relative path to the aliases file. @type lineNo: L{int} @param lineNo: The position of the line within the aliases file. cSg|]}|qSstrip).0pr r XC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/mail/alias.py )zhandle..:z+Invalid format on line %d of alias file %s.,N) splitlenr err setdefaultr extendmapstr) resultlinefilenameZlineNopartsfmtarguseraliasr r rhandles (r&c Csi}d}|durt|}d}nt|dd}d}d}z7|D],}|d7}|}|d r.q|d s8|d r=||}q|rFt|||||}qW|rP|n|rX|ww|rbt|||||D] \}} t| ||||<qf|S) a Load a file containing email aliases. Lines in the file should be formatted like so:: username: alias1, alias2, ..., aliasN Aliases beginning with a C{|} will be treated as programs, will be run, and the message will be written to their stdin. Aliases beginning with a C{:} will be treated as a file containing additional aliases for the username. Aliases beginning with a C{/} will be treated as the full pathname to a file to which the message will be appended. Aliases without a host part will be assumed to be addresses on localhost. If a username is specified multiple times, the aliases for each are joined together as if they had all been on one line. Lines beginning with a space or a tab are continuations of the previous line. Lines beginning with a C{#} are comments. @type domains: L{dict} mapping L{bytes} to L{IDomain} provider @param domains: A mapping of domain name to domain object. @type filename: L{bytes} or L{None} @param filename: The full or relative path to a file from which to load aliases. If omitted, the C{fp} parameter must be specified. @type fp: file-like object or L{None} @param fp: The file from which to load aliases. If specified, the C{filename} parameter is ignored. @rtype: L{dict} mapping L{bytes} to L{AliasGroup} @return: A mapping from username to group of aliases. FNTnamez rr#  ) opengetattrrstriplstrip startswithr&closeitems AliasGroup) domainsr fprr1iprevruar r r loadAliasFile3s>)    r:c@s*eZdZdZddZddZd ddZdS) AliasBasez The default base class for aliases. @ivar domains: See L{__init__}. @type original: L{Address} @ivar original: The original address being aliased. cCs||_t||_dS)z @type domains: L{dict} mapping L{bytes} to L{IDomain} provider @param domains: A mapping of domain name to domain object. @type original: L{bytes} @param original: The original address being aliased. N)r4rAddressoriginal)selfr4r=r r r__init__szAliasBase.__init__cCs|j|jjS)z Return the domain associated with original address. @rtype: L{IDomain} provider @return: The domain for the original address. N)r4r=domainr>r r rr@zAliasBase.domainNcCs0|duri}t||vrdSd|t|<|SaY Map this alias to its ultimate destination. @type aliasmap: L{dict} mapping L{bytes} to L{AliasBase} @param aliasmap: A mapping of username to alias or group of aliases. @type memo: L{None} or L{dict} of L{AliasBase} @param memo: A record of the aliases already considered in the resolution process. If provided, C{memo} is modified to include this alias. @rtype: L{IMessage } or L{None} @return: A message receiver for the ultimate destination or None for an invalid destination. N)rcreateMessageReceiverr>aliasmapmemor r rresolves   zAliasBase.resolveN)__name__ __module__ __qualname____doc__r?r@rHr r r rr;{s    r;c@s8eZdZdZddZdefddZddZd d d Zd S) AddressAliasz An alias which translates one email address into another. @type alias : L{Address} @ivar alias: The destination address. cGs"tj|g|Rt||_dS)aQ @type alias: L{Address}, L{User}, L{bytes} or object which can be converted into L{bytes} @param alias: The destination address. @type args: 2-L{tuple} of (0) L{dict} mapping L{bytes} to L{IDomain} provider, (1) L{bytes} @param args: Arguments for L{AliasBase.__init__}. N)r;r?rr<r%)r>r%argsr r rr?s zAddressAlias.__init__returncCd|jdS)z Build a string representation of this L{AddressAlias} instance. @rtype: L{bytes} @return: A string containing the destination address. z
N)r%rAr r r__str__rBzAddressAlias.__str__cCs|t|jS)z Create a message receiver which delivers a message to the destination address. @rtype: L{IMessage } provider @return: A message receiver. N)r@existsrr%rAr r rrDsz"AddressAlias.createMessageReceiverNcCs|duri}t||vrdSd|t|<z|t|jddd|WStjy/Ynw|jj|vr@||jj||SdSrC) rr@rTrZUserr%Z SMTPBadRcptlocalrHrEr r rrHs  " zAddressAlias.resolverI) rJrKrLrMr?rrSrDrHr r r rrNs    rNc@>eZdZdZddZddZddZdd Zd efd d Z d S) FileWrappera A message receiver which delivers a message to a file. @type fp: file-like object @ivar fp: A file used for temporary storage of the message. @type finalname: L{bytes} @ivar finalname: The name of the file in which the message should be stored. cCst|_||_dS)z @type filename: L{bytes} @param filename: The name of the file in which the message should be stored. N)tempfile TemporaryFiler5 finalname)r>r r r rr?s  zFileWrapper.__init__cCs|j|ddS)z Write a received line to the temporary file. @type line: L{bytes} @param line: A received line of the message.  N)r5writer>rr r r lineReceivedzFileWrapper.lineReceivedcCs|jddzt|jd}WntyttYSw|| |j |j Wdn1s:wYt |jS)ac Handle end of message by writing the message to the file. @rtype: L{Deferred } which successfully results in L{bytes} @return: A deferred which succeeds with the name of the file to which the message has been stored or fails if the message cannot be saved to the file. rr9N) r5seekr,rZ BaseExceptionrZfailrFailurer\readr1Zsucceed)r>fr r r eomReceiveds    zFileWrapper.eomReceivedcCs|jd|_dS)zG Close the temporary file when the connection is lost. N)r5r1rAr r rconnectionLost-s  zFileWrapper.connectionLostrPcCrQ)z Build a string representation of this L{FileWrapper} instance. @rtype: L{bytes} @return: A string containing the file name of the message. z r rOr r rr?Fs zFileAlias.__init__rPcCrQ)z Build a string representation of this L{FileAlias} instance. @rtype: L{bytes} @return: A string containing the name of the file. zs  rhc@seZdZdZdS)ProcessAliasTimeoutzb An error indicating that a timeout occurred while waiting for a process to complete. N)rJrKrLrMr r r rrkesrkc@s`eZdZdZdZdZdZeZdddZddZ d d Z d d Z d dZ ddZ defddZdS)MessageWrapperaF A message receiver which delivers a message to a child process. @type completionTimeout: L{int} or L{float} @ivar completionTimeout: The number of seconds to wait for the child process to exit before reporting the delivery as a failure. @type _timeoutCallID: L{None} or L{IDelayedCall } provider @ivar _timeoutCallID: The call used to time out delivery, started when the connection to the child process is closed. @type done: L{bool} @ivar done: A flag indicating whether the child process has exited (C{True}) or not (C{False}). @type reactor: L{IReactorTime } provider @ivar reactor: A reactor which will be used to schedule timeouts. @ivar protocol: See L{__init__}. @type processName: L{bytes} or L{None} @ivar processName: The process name. @type completion: L{Deferred } @ivar completion: The deferred which will be triggered by the protocol when the child process exits. F<NcCsD||_||_t|_|j|j_|j|j|dur ||_dSdS)a @type protocol: L{ProcessAliasProtocol} @param protocol: The protocol associated with the child process. @type process: L{bytes} or L{None} @param process: The process name. @type reactor: L{None} or L{IReactorTime } provider @param reactor: A reactor which will be used to schedule timeouts. N) processNamerrZDeferred completiononEndZaddBoth _processEndedr)r>rprocessrr r rr?s    zMessageWrapper.__init__cCs(d|_|jdur|jd|_dS|S)a Record process termination and cancel the timeout call if it is active. @type result: L{Failure } @param result: The reason the child process terminated. @rtype: L{None} or L{Failure } @return: None, if the process end is expected, or the reason the child process terminated, if the process end is unexpected. TN)done_timeoutCallIDcancel)r>rr r rrqs    zMessageWrapper._processEndedcCs |jrdS|jj|ddS)z Write a received line to the child process. @type line: L{bytes} @param line: A received line of the message. Nr[)rsr transportr\r]r r rr^szMessageWrapper.lineReceivedcCs,|js|jj|j|j|j|_|j S)z Disconnect from the child process and set up a timeout to wait for it to exit. @rtype: L{Deferred } @return: A deferred which will be called back when the child process exits. N) rsrrvZloseConnectionrZ callLatercompletionTimeout_completionCancelrtrorAr r rres  zMessageWrapper.eomReceivedcCsDd|_|jjdtd|jd}d|j_|jt |dS)z Handle the expiration of the timeout for the child process to exit by terminating the child process forcefully and issuing a failure to the L{completion} deferred. NZKILLzNo answer after z seconds) rtrrvZ signalProcessrkrwrproerrbackrrb)r>excr r rrxs z MessageWrapper._completionCancelcCsdS)z9 Ignore notification of lost connection. Nr rAr r rrfszMessageWrapper.connectionLostrPcCrQ)z Build a string representation of this L{MessageWrapper} instance. @rtype: L{bytes} @return: A string containing the name of the process. z} @ivar onEnd: If set, a deferred on which to errback when the process ends. NcCs|jdur |j|dSdS)z Call an errback. @type reason: L{Failure } @param reason: The reason the child process terminated. N)rpry)r>reasonr r r processEndeds z!ProcessAliasProtocol.processEnded)rJrKrLrMrpr~r r r rr|s r|c@s:eZdZdZeZddZdefddZddZd d Z d S) ProcessAliasa6 An alias which is handled by the execution of a program. @type path: L{list} of L{bytes} @ivar path: The arguments to pass to the process. The first string is the executable's name. @type program: L{bytes} @ivar program: The path of the program to be executed. @type reactor: L{IReactorTime } and L{IReactorProcess } provider @ivar reactor: A reactor which will be used to create and timeout the child process. cGs,tj|g|R||_|jd|_dS)aX @type path: L{bytes} @param path: The command to invoke the program consisting of the path to the executable followed by any arguments. @type args: 2-L{tuple} of (0) L{dict} mapping L{bytes} to L{IDomain} provider, (1) L{bytes} @param args: Arguments for L{AliasBase.__init__}. rN)r;r?rpathprogram)r>rrOr r rr?s zProcessAlias.__init__rPcCrQ)z Build a string representation of this L{ProcessAlias} instance. @rtype: L{bytes} @return: A string containing the command used to invoke the process. z } method on L{reactor} so that it can be customized for test purposes. @type proto: L{IProcessProtocol } provider @param proto: An object which will be notified of all events related to the created process. @type program: L{bytes} @param program: The full path name of the file to execute. @type path: L{list} of L{bytes} @param path: The arguments to pass to the process. The first string should be the executable's name. @rtype: L{IProcessTransport } provider @return: A process transport. N)r spawnProcess)r>protorrr r rr7szProcessAlias.spawnProcesscCs,t}t||j|j}|||j|j|S)z Launch a process and create a message receiver to pass a message to the process. @rtype: L{MessageWrapper} @return: A message receiver which delivers a message to the process. N)r|rlrrrr)r>rmr r rrDQsz"ProcessAlias.createMessageReceiverN) rJrKrLrMrr?rrSrrDr r r rr s rc@rV) MultiWrapperz A message receiver which delivers a single message to multiple other message receivers. @ivar objs: See L{__init__}. cCs ||_dS)z @type objs: L{list} of L{IMessage } provider @param objs: Message receivers to which the incoming message should be directed. N)objs)r>rr r rr?hs zMultiWrapper.__init__cCs|jD]}||qdS)z Pass a received line to the message receivers. @type line: L{bytes} @param line: A line of the message. N)rr^)r>ror r rr^ps  zMultiWrapper.lineReceivedcCstdd|jDS)aG Pass the end of message along to the message receivers. @rtype: L{DeferredList } whose successful results are L{bytes} or L{None} @return: A deferred list which triggers when all of the message receivers have finished handling their end of message. cSr r )re)rrr r rrrz,MultiWrapper.eomReceived..N)rZ DeferredListrrAr r rrezs zMultiWrapper.eomReceivedcCs|jD]}|qdS)zQ Inform the message receivers that the connection has been lost. N)rrf)r>rr r rrfs  zMultiWrapper.connectionLostrPcCsdtt|jdS)z Build a string representation of this L{MultiWrapper} instance. @rtype: L{bytes} @return: A string containing a list of the message receivers. z.r|/z Directory delivery not supported)r;r?aliasespopr r0r,rar rjoinrrappendprocessAliasFactoryosrisdirrhrN)r>r2rOaddrrdr r rr?s,    $    zAliasGroup.__init__cCri)z Return the number of aliases in the group. @rtype: L{int} @return: The number of aliases in the group. N)rrrAr r r__len__rjzAliasGroup.__len__rPcCsddtt|jS)z Build a string representation of this L{AliasGroup} instance. @rtype: L{bytes} @return: A string containing the aliases in the group. zz, N)rrrrrAr r rrSszAliasGroup.__str__cCstdd|jDS)a, Create a message receiver for each alias and return a message receiver which will pass on a message to each of those. @rtype: L{MultiWrapper} @return: A message receiver which passes a message on to message receivers for each alias in the group. cSr r )rD)rr9r r rrrz4AliasGroup.createMessageReceiver..N)rrrAr r rrDs z AliasGroup.createMessageReceiverNcCs<|duri}g}|jD] }||||q ttd|S)a Map each of the aliases in the group to its ultimate destination. @type aliasmap: L{dict} mapping L{bytes} to L{AliasBase} @param aliasmap: A mapping of username to alias or group of aliases. @type memo: L{None} or L{dict} of L{AliasBase} @param memo: A record of the aliases already considered in the resolution process. If provided, C{memo} is modified to include this alias. @rtype: L{MultiWrapper} @return: A message receiver which passes the message on to message receivers for the ultimate destination of each alias in the group. N)rrrHrfilter)r>rFrGrr9r r rrHs  zAliasGroup.resolverI) rJrKrLrMrrr?rrrSrDrHr r r rr3s &   r3r{)rMrrXZzope.interfacerZtwisted.internetrrrZ twisted.mailrZtwisted.mail.interfacesrZtwisted.pythonrr r&r:r;rNZIMessagerWrh ExceptionrkrlZProcessProtocolr|rrr3r r r rs8    H6FE&T6