OwlCyberSecurity - MANAGER
Edit File: augment.pyc
� �t�ac @ s[ d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m Z d d l m Z d d l m Z 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 e � Z e d" � Z d e f d � � YZ i d d 6d d 6d d 6d d 6d d 6d d 6Z d e f d � � YZ d e f d! � � YZ d S(# i����N( t inlineCallbackst returnValuet succeed( t Logger( t fullServerPatht config( t xmlaugmentsparser( t XMLAugmentsParser( t newElementTreeWithRoott addSubElementt writeXMLt readXML( t normalizeUUIDt defaultt nones accept-alwayss decline-alwayss accept-if-frees decline-if-busyt automatict AugmentRecordc B s/ e Z d Z e d e e d d e e d � Z RS( s0 Augmented directory record information t R c C sp | | _ | | _ | | _ | | _ | | _ | | _ | | _ | t k rQ | n d | _ | | _ t | _ d S( NR ( t uidt enabledt serverIDt enabledForCalendaringt enabledForAddressBookst enabledForLogint autoSchedulet allowedAutoScheduleModest autoScheduleModet autoAcceptGroupt Falset clonedFromDefault( t selfR R R R R R R R R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyt __init__5 s N( t __name__t __module__t __doc__R t Nonet TrueR ( ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR 0 s t Usert userst Groupt groupst Locationt locationst Resourcet resourcest Addresst addressest Wikit wikist AugmentDBc B s� e Z d Z d � Z e d � � Z e d � � Z e d � � Z d � Z e d � � Z d � Z d � Z d � Z d � Z RS( s= Abstract base class for an augment record database. c C s i | _ d S( N( t cachedRecords( R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR [ s c c s� g } g } xx | j � VD]i } t | � } | | k r | j | � V} t j | � } | j � | _ | j | � | j | � q q Wy | j | � V| j | � VWn t k r� n Xd S( s� Normalize (uppercase) all augment UIDs which are parseable as UUIDs. @return: a L{Deferred} that fires when all records have been normalized. N( t getAllUIDsR t _lookupAugmentRecordt copyt deepcopyt upperR t appendt removeAugmentRecordst addAugmentRecordst IOError( R t removet addR t nuidt oldt new( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyt normalizeUUIDs_ s c c sX | t k r t d � n t | } | j | � V} | d k rL t | � n x� d | | d d !f d | | d f d | d d !f d | d f d | f d f D]P } | j | � V} | d k r� t j | � } | | _ t | _ t | � q� q� Wt d d t d t d t d t �} | | j d <t j | � } | | _ t | _ t | � d S( s� Get an AugmentRecord for the specified UID or the default. @param uid: directory UID to lookup @type uid: C{str} @return: L{Deferred} s %s-%s*i i s %s*s %s-Defaultt DefaultR R R R N( t recordTypesMapR R# R4 t _cachedAugmentRecordR5 R6 R R$ R R R2 ( R R t recordTypet resultt lookup( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyt getAugmentRecordx s< c C s t d � � d S( sK Get all AugmentRecord UIDs. @return: L{Deferred} s Child class must define this.N( t NotImplementedError( R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR3 � s c C s t d � � d S( s� Get an AugmentRecord for the specified UID. @param uid: directory UID to lookup @type uid: C{str} @return: L{Deferred} s Child class must define this.N( RI ( R R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR4 � s c c sD | | j k r/ | j | � V} | | j | <n t | j | � d S( s� Get an AugmentRecord for the specified UID from the cache. @param uid: directory UID to lookup @type uid: C{str} @return: L{Deferred} N( R2 R4 R ( R R RF ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyRD � s c C s t d � � d S( s� Add an AugmentRecord to the DB. @param record: augment records to add @type record: C{list} of L{AugmentRecord} @return: L{Deferred} s Child class must define this.N( RI ( R t records( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR: � s c C s t d � � d S( s� Remove AugmentRecords with the specified UIDs. @param uid: directory UIDs to remove @type uid: C{list} of C{str} @return: L{Deferred} s Child class must define this.N( RI ( R t uids( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR9 � s c C s | j j � d S( sH Refresh any cached data. @return: L{Deferred} N( R2 t clearR# ( R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyt refresh� s c C s t d � � d S( sC Remove all records. @return: L{Deferred} s Child class must define this.N( RI ( R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyt clean� s ( R R! R" R R RA RH R3 R4 RD R: R9 RM RN ( ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR1 V s 1 t AugmentXMLDBc B s� e Z d Z d d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( s8 XMLFile based augment database implementation. i c C s� t t | � j � g | D] } t t j | � ^ q | _ i | _ x | j D] } d | j | <qN W| | _ d | _ i | _ y | j � | _ Wn! t k r� t j d � � n Xt j � | _ | j � d S( Ni s: Failed to parse XML augments file - fatal error on startup( i i ( t superRO R R R t DataRoott xmlFilest xmlFileStatst statSecondst lastCachedt dbt _parseXMLt RuntimeErrort logt errort timeRA ( R RR RT t path( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR s ( c C s t | j j � � S( sK Get all AugmentRecord UIDs. @return: L{Deferred} ( R RV t keys( R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR3 s c C s? t j � | j | j k r) | j � n t | j j | � � S( s� Get an AugmentRecord for the specified UID. @param uid: directory UID to lookup @type uid: C{str} @return: L{Deferred} ( R[ RU RT RM R RV t get( R R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR4 s c C s� t � } t � } x3 | D]+ } | j | j k r7 | n | j | � q W| ru x$ | j D] } | j | | � qX Wn | r� | j | j d | � n d | _ t d � S( s Add an AugmentRecord to the DB. @param records: augment records to add @type records: C{list} of L{AugmentRecord} @param update: C{True} if changing an existing record @type update: C{bool} @return: L{Deferred} i N( t listR RV R8 RR t _doModifyInFilet _doAddToFileRU R R# ( R RJ t new_recordst existing_recordst recordt xmlFile( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR: - s ) c C s} t j j | � s9t t j � \ } } x' | j j � D] } | j | | � q7 Wt | | � d } t j r� y t j t j � j } Wq� t k r� t j d d t j �q� Xn d } t j ry t j t j � j } Wqt k rt j d d t j �qXn | d k r9| d k r9t j | | | � q9n t | � \ } } x | D] } | j | | � qRWt | | � d S( Ni����s User not found: {user}t users Group not found: {grp}t grp( t osR\ t existsR R t ELEMENT_AUGMENTSRV t itervaluest _addRecordToXMLDBR R t UserNamet pwdt getpwnamt pw_uidt KeyErrorRY RZ t GroupNameRg t getgrnamt gr_gidt chownR ( R t xmlfileRJ t _ignore_etreet augments_nodeRd R t gid( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyRa M s. c C s� t j j | � s d St | � \ } } t g | D] } | j | f ^ q2 � } t } xj | D]b } | j t j k r{ q] n | j t j � j } | | k r] | | } | j | | � t } q] q] W| r� t | | � n d S( N( Rh R\ Ri R t dictR R t tagR t ELEMENT_RECORDt findt ELEMENT_UIDt textt _updateRecordInXMLDBR$ R ( R Rv RJ Rw Rx Rd t recordMapt changedt record_nodeR ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR` s s ( c C sq t � } x7 | D]/ } | | j k r | j | =| j | � q q Wx! | j D] } | j | | � qM Wt d � S( s� Remove AugmentRecords with the specified UIDs. @param uids: list of uids to remove @type uids: C{list} of C{str} @return: L{Deferred} N( t setRV R= RR t _doRemoveFromFileR R# ( R RK t removedR Re ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR9 � s c C s� y t | � \ } } Wn t k r* d SXt } xW | D]O } | j t j k rV q8 n | j t j � j | k r8 | j | � t } q8 q8 W| r� t | | � n d S( N( R R; R R{ R R| R} R~ R R< R$ R ( R Rv RK Rw Rx R� t child( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR� � s c C s& t | t j � } | j | | � d S( N( R R R| R� ( R Rd t parentNodeR� ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyRl � s c C s� | 2t | t j | j � | j r< t | t j | j � n t | t j | j rW d n d � t | t j | j ry d n d � t | t j | j r� d n d � | j r� t | t j | j � n | j r� t | t j | j � n d S( Nt truet false( R R R~ R R t ELEMENT_SERVERIDt ELEMENT_ENABLECALENDARR t ELEMENT_ENABLEADDRESSBOOKR t ELEMENT_ENABLELOGINR R t ELEMENT_AUTOSCHEDULE_MODER t ELEMENT_AUTOACCEPTGROUP( R Rd t recordNode( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyR� � s """ c C si t t | � j � y" | j � } | r4 | | _ n Wn t k rU t j d � n Xt j � | _ d S( s* Refresh any cached data. sA Failed to parse XML augments file during cache refresh - ignoringN( RP RO RM RW RV RX RY RZ R[ RU R# ( R t results( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyRM � s c C s | j | j j � � S( s% Remove all records. ( R9 RV R] ( R ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyRN � s c C s� x~ | D]v } t j j | � r | j j | d � \ } } t j j | � } t j j | � } | | k sv | | k r} t Sq q Wt S( ss Check to see whether any of the given files have been modified since we last parsed them. i ( i i ( Rh R\ Ri RS R^ t getmtimet getsizeR$ R ( R RR Re t oldModTimet oldSizet newModTimet newSize( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyt _shouldReparse� s c C s� i } xO | j D] } t j j | � r Pq q Wt d d t d t d t d t �| d <| j | j � r� xm | j D]_ } t j j | � rt t | | � t j j | � } t j j | � } | | f | j | <qt qt Wn | S( s{ Parse self.xmlFiles into AugmentRecords. If none of the xmlFiles exist, create a default record. RB R R R R ( RR Rh R\ Ri R R$ R� R R� R� RS ( R R� Re R� R� ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyRW � s$ ( R R! R" R R3 R4 R: Ra R` R9 R� Rl R� RM RN R� RW ( ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyRO � s & ( R R s accept-alwayss decline-alwayss accept-if-frees decline-if-busyR (! R5 Rg Rh Rn R[ t twisted.internet.deferR R R t twext.python.logR t twistedcaldav.configR R t twistedcaldav.directoryR t) twistedcaldav.directory.xmlaugmentsparserR t twistedcaldav.xmlutilR R R R t twistedcaldav.directory.utilR RY t frozensetR t objectR RC R1 RO ( ( ( s2 /opt/cpanel-ccs/twistedcaldav/directory/augment.pyt <module> s< " �