ProPeler
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
opt
/
cpanel-ccs
/
txdav
/
who
/
Filename :
augment.pyc
back
Copy
� �t�ac @ sR d Z d g Z d d l Z d d l m Z d d l m Z m Z m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z m Z m Z m Z d d l m Z d d l m Z d d l m Z m Z d d l m Z m Z m Z e � Z! d � Z" e e e � d e e f d � � Y� Z# d e e f d � � YZ$ d S( s Augmenting Directory Service t AugmentedDirectoryServicei����N( t implementer( t inlineCallbackst returnValuet succeed( t AugmentRecord( t Logger( t DirectoryRecord( t DirectoryService( t IDirectoryServicet RecordTypet FieldNamet NotAllowedError( t ConstantsContainer( t IStoreDirectoryService( t CalendarDirectoryRecordMixint CalendarDirectoryServiceMixin( t AutoScheduleModeR R c s d � � � � f d � } | S( sc A decorator which keeps track of the wrapped function's call count and total duration c S s t j | t j � | � | S( s� Figures out how much time to add to the total time spent within the method identified by key and stores that in the timings dict. @param result: the result of the wrapped method @param timings: the dictionary to store timings in @type timings: C{dict} @param key: the method name @type key: C{str} @param startTime: the start time of the call in seconds @type startTime: C{float} ( R t _addTimingt time( t resultt keyt startTime( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyt recordTiming9 s c s8 t j � } � | | | � } | j � � j | � | S( sJ Records the start time of the call and the method's name ( R t addBotht func_name( t selft argst kwdsR t d( t fR ( s$ /opt/cpanel-ccs/txdav/who/augment.pyt timingWrapperI s ( ( R R ( ( R R s$ /opt/cpanel-ccs/txdav/who/augment.pyt timed3 s c B s� e Z d Z e e e f � Z i Z d � Z e d � � Z d � Z d � Z e d � � Z d � Z e d d d d � � Z e d d d � � Z e e d d � � � Z e e d d � � � Z e e d d d � � � Z e e d d � � � Z e e d d d � � � Z e d � � Z e d � � Z e d � � Z e e e d � � � Z d � Z e d � � Z d � Z! e d � � Z" e d � � Z# RS( s� Augmented directory service. This is a directory service that wraps an L{IDirectoryService} and augments directory records with additional or modified fields. c C s; t j | | j � | | _ | | _ | | _ d | _ d S( N( t BaseDirectoryServicet __init__t realmNamet _directoryt _storet _augmentDBt Nonet _ldapDS( R t directoryt storet augmentDB( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR" g s c C s] | | j k r d | j | <n | j | \ } } | d 7} | | 7} | | f | j | <d S( Ni g i ( i g ( t _timings( t clsR t durationt countt timeSpent( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR q s c C s | j j � S( N( R$ t flush( R ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR1 z s c C sE i } | j | j � | j d k r; | j | j j � n t | � S( N( t updateR, R( R' t poolStatsR ( R t results( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyt stats} s c C s | j j S( N( R$ t recordType( R ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR6 � s c C s | j j � S( N( R$ t recordTypes( R ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR7 � s c c sg | j j | d | d | d | �V} g } x+ | D]# } | j | � V} | j | � q2 Wt | � d S( NR7 t limitResultst timeoutSeconds( R$ t recordsFromExpressiont _augmentt appendR ( R t expressionR7 R8 R9 t recordst augmentedt record( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR: � s c c sd | j j | | d | d | �V} g } x+ | D]# } | j | � V} | j | � q/ Wt | � d S( NR8 R9 ( R$ t recordsWithFieldValueR; R< R ( R t fieldNamet valueR8 R9 R> R? R@ ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRA � s c c sX t | t � s! | j d � } n | j j | d | �V} | j | � V} t | � d S( Ns utf-8R9 ( t isinstancet unicodet decodeR$ t recordWithUIDR; R ( R t uidR9 R@ ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRG � s c c s7 | j j | d | �V} | j | � V} t | � d S( NR9 ( R$ t recordWithGUIDR; R ( R t guidR9 R@ ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRI � s c c sa | j j | d | d | �V} g } x+ | D]# } | j | � V} | j | � q, Wt | � d S( NR8 R9 ( R$ t recordsWithRecordTypeR; R< R ( R R6 R8 R9 R> R? R@ ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRK � s c c s[ t | t � s! | j d � } n | j j | | d | �V} | j | � V} t | � d S( Ns utf-8R9 ( RD RE RF R$ t recordWithShortNameR; R ( R R6 t shortNameR9 R@ ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRL � s c c s� t | t � s! | j d � } n | j j | d | d | �V} g } x+ | D]# } | j | � V} | j | � qM Wt | � d S( Ns utf-8R8 R9 ( RD RE RF R$ t recordsWithEmailAddressR; R< R ( R t emailAddressR8 R9 R> R? R@ ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRN � s c O s t j | | | � S( N( R t recordWithCalendarUserAddress( R R R ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRP � s c O s t j | | | � S( N( R t recordsMatchingTokens( R R R ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRQ � s c O s t j | | | � S( N( R t recordsMatchingFields( R R R ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRR s c c s g } g } x�| D]�} | j | � \ } } | r�| j t j k r�i d t j 6d t j 6d t j 6d t j 6d t j 6d t j 6j | j t j d � d � } i | j d 6| d 6} t j | k r� | t j | d <n t j | k r | t j | d <n t j | k r#| t j | d <n t j | k rF| t j | d <n t j | k ri| t j | d <n t | � } | j | � n | j t | j t | d � r�| j j n | � � q W| r�| j j | � Vn | ry | j j | d | �VWqt k r qXn d S( s� Pull out the augmented fields from each record, apply those to the augments database, then update the base records. t nones accept-alwayss decline-alwayss accept-if-frees decline-if-busyt automaticRH t autoScheduleModet enabledForCalendaringt enabledForAddressBookst enabledForLogint autoAcceptGroupt serverIDt _baseRecordt createN( t _splitFieldsR6 R t groupR RS t acceptt declinet acceptIfFreet declineIfBusyt acceptIfFreeDeclineIfBusyt getR RU R' RH t hasCalendarst hasContactst loginAllowedRY t serviceNodeUIDR R< R R$ t hasattrR[ t fieldsR& t addAugmentRecordst updateRecordsR ( R R> R\ t baseRecordst augmentRecordsR@ t baseFieldst augmentFieldsRU t kwargst augmentRecord( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRl sH 8 c C s� | d k r d Si } | j j � } xV t j t j t j t j t j t j f D]* } | | k rP | | | | <| | =qP qP W| | f S( s� Returns a tuple of two dictionaries; the first contains all the non augment fields, and the second contains all the augment fields. N( R' Rj t copyR Rg Re Rf RU RY Rh ( R R@ Rp Ro t field( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR] B s c c s& | j j | � V| j j | � Vd S( N( R& t removeAugmentRecordsR$ t removeRecords( R t uids( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyRv X s c C s/ | j j | � } | | k r+ | | | <n d S( sR Assign a value to a field only if not already present in fields. N( RB t lookupByName( R Rj t nameRC Rt ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyt _assignToField] s c c s/ | d k r t d � n | j j | j | j | j � � V} | d k rW t | � n | j j � } | r�| j t j k r� | j | d t � | j | d t � n, | j | d | j � | j | d | j � i t j d 6t j d 6t j d 6t j d 6t j d 6t j d 6j | j d � } | j t j t j f k rZ| d k rZt j } qZn | j | d | � | j d k r�| j | d | j j d � � n | j | d | j � | j | d | j j d � � n9 | j | d t � | j | d t � | j | d t � t | | | � } t | � d S( NRe Rf RS s accept-alwayss decline-alwayss accept-if-frees decline-if-busyRT RU RY s utf-8Rg Rh ( R' R R& t getAugmentRecordRH t recordTypeToOldNameR6 Rj Rs R R^ Rz t FalseRV RW R RS R_ R` Ra Rb Rc Rd RU t CalRecordTypet locationt resourceRY RF RX RZ t AugmentedDirectoryRecord( R R@ Rr Rj RU t augmentedRecord( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR; e sb c c s� | j j | j | j | j � � V} | d k r� i d t j 6d t j 6d t j 6d t j 6d t j 6d t j 6j | � } | | _ | j j | g � Vn d S( NRS s accept-alwayss decline-alwayss accept-if-frees decline-if-busyRT ( R& R{ RH R| R6 R' R RS R_ R` Ra Rb Rc Rd RU Rk ( R R@ RU Rr ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyt setAutoScheduleMode� s N($ t __name__t __module__t __doc__R t BaseFieldNameR RB R, R" t classmethodR R1 R5 t propertyR6 R7 R R' R: RA R RG RI RK RL RN RP RQ RR R} Rl R] Rv Rz R; R� ( ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR U sT : [R� c B s� e Z d Z d � Z e e d � � � Z d � Z d � Z d � Z e e d � � � Z e d � � Z e d � � Z e d � � Z RS( s% Augmented directory record. c C s- t j | | | � t j | � | | _ d S( N( R R" R R[ ( R t servicet baseRecordt augmentedFields( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR" � s c c sO g } | j j � V} x( | D] } | j | j j | � V� q Wt | � d S( N( R[ t membersR< R� R; R ( R R? R> R@ ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR� � s c C s | j j | � S( N( R[ t addMembers( R t memberRecords( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR� � s c C s | j j | � S( N( R[ t removeMembers( R R� ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR� � s c C s | j j | � S( N( R[ t setMembers( R R� ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR� � s c # s� g } � f d � } � j j j d | � V} xD | D]< } � j j | � V} | r5 | j � j j | � V� q5 q5 Wt | � d S( Nc s | j � j � S( N( t groupUIDsForRH ( t txn( R ( s$ /opt/cpanel-ccs/txdav/who/augment.pyt _groupUIDsFor� s s AugmentedDirectoryRecord.groups( R� R% t inTransactionRG R< R; R ( R R? R� t groupUIDst groupUIDt groupRecord( ( R s$ /opt/cpanel-ccs/txdav/who/augment.pyt groups� s !c C s | j j | � S( N( R[ t verifyPlaintextPassword( R t password( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR� s c G s | j j | � S( N( R[ t verifyHTTPDigest( R R ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR� s c C s | j j | � S( N( R[ t accessForRecord( R R@ ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR� s ( R� R� R� R" R R R� R� R� R� R� R� R� R� ( ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyR� � s (% R� t __all__R t zope.interfaceR t twisted.internet.deferR R R t twistedcaldav.directory.augmentR t twext.python.logR t twext.who.directoryR R R! t twext.who.idirectoryR R R R� R t twext.who.utilR t txdav.common.idirectoryserviceR t txdav.who.directoryR R t txdav.who.idirectoryR R~ t logR R R� ( ( ( s$ /opt/cpanel-ccs/txdav/who/augment.pyt <module> s& " "� ~