OwlCyberSecurity - MANAGER
Edit File: peruserdata.pyc
� �t�ac @ s[ d d l m Z d d l m Z m Z m Z m Z m Z d g Z d e f d � � YZ d S( i����( t CalendarFilter( t Componentt Propertyt PERUSER_COMPONENTt PERUSER_UIDt PERINSTANCE_COMPONENTt PerUserDataFilterc B s� e Z d Z d Z d Z e j g 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( s Filter per-user data t TRANSPt VALARMc C s | | _ d S( sz @param uid: unique identifier of the user for whom the data is being filtered @type uid: C{str} N( t uid( t selfR ( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyt __init__I s c C s� | j | � } d } x� t | j � � D]q } | j � t k r( | j t � | j k rh | j | � q� | d k r� | } | j | � q� t d � � q( q( W| r� | j | | � n | j | � | S( sH Filter the supplied iCalendar object using the request information. Assume that the object is a CalDAV calendar resource. @param ical: iCalendar object - this will be modified and returned @type ical: L{Component} or C{str} @return: L{Component} for the filtered calendar data sD Can't have two X-CALENDARSERVER-PERUSER components for the same userN( t validCalendart Nonet tuplet subcomponentst nameR t propertyValueR R t removeComponentt AssertionErrort _filterBackt _defaultFilter( R t icalt peruser_componentt component( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyt filterR s c C s� i } xR | j � D]D } | j � t k rG t d | j � f � � n | | | j � <q Wt | j � � } t | j � � } | j | � } | j | � } | j | � } | j � d k rxb | D]N } | j | � } | d k r� q� n | | } | j | | � | j | � q� Wn | r#n x7 | D]/ } | j | � } | | } | j | | � q*W| r�| j d � } | r�x0 | D]% } | j | � } | j | | � qWq�n d S( s� Merge the per-user data back into the main calendar data. @param ical: main calendar data to merge into @type ical: L{Component} @param peruser: the per-user data to merge in @type peruser: L{Component} s@ Wrong sub-component '%s' in a X-CALENDARSERVER-PERUSER componentN( R R R R t getRecurrenceIDUTCt sett getComponentInstancest keyst intersectiont differencet masterComponentR t deriveInstancet _filterBackComponentt addComponentt overriddenComponentt get( R R t perusert peruser_recurrence_mapt subcomponentt ical_recurrence_sett peruser_recurrence_sett union_sett ical_only_sett peruser_only_sett ridt ical_componentR t peruser_master( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyR v s< c C sd x! | j � D] } | j | � q Wx9 | j � D]+ } | j � d k rO q1 n | j | � q1 Wd S( s Copy all properties and sub-components from per-user data into the main component @param ical: main calendar component data to merge into @type ical: L{Component} @param peruser: the per-user data to merge in @type peruser: L{Component} s RECURRENCE-IDN( R R# t propertiesR t addProperty( R R R& R( t property( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyR" � s c C s` | j � } | j � d k r\ | j d � j � r\ | j d � r\ | j t d d � � n d S( s� There is no per-user component. Instead apply default properties to the data for this user. @param ical: the iCalendar object to process @type ical: L{Component} t VEVENTt DTSTARTR t TRANSPARENTN( t mainComponentR R t isDateOnlyt hasPropertyt addPropertyToAllComponentsR ( R R t comp( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyR � s 7c C s� | j | � } x; t | j � � D]' } | j � t k r" t d � � q" q" W| j | � | d k rj | S| j | � } | j | | � | S( sM Merge the new data with the old taking per-user information into account. @param icalnew: new calendar data @type icalnew: L{Component} or C{str} @param icalold: existing calendar data @type icalold: L{Component} or C{str} @return: L{Component} for the merged calendar data sI Cannot merge calendar data with X-CALENDARSERVER-PERUSER components in itN( R R R R R t ValueErrort _splitPerUserDataR t _mergeRepresentations( R t icalnewt icaloldR ( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyt merge� s c s� � � f d � } t � j � � } � j r6 | � n d } i } x�| D]�} | j � d k rg qI n | j d � } | d k r� | j � n d } � j r� t t � n d } t } � j | � x� t | j � � D]t } | j � t j k s| j � j d � r� | j � t j k r� � j r3| j | � n | j | � t } q� q� Wxs t | j � � D]_ } | j � t j k s�| j � j d � r`� j r�| j | � n | j | � t } q`q`W| rI | rI | | | <qI qI W� j r�t } | j d � } | r| j | � t } n x t | j � d d � �D]b \ } } | d k rTq6n | d k sl| | k r6| j t d | � � | j | � t } q6q6W| r�� j | � n � j � � n d S( sk Split the per-user data out of the "normal" iCalendar components into separate per-user components. Along the way keep the iCalendar representation in a "minimal" state by eliminating any components that are the same as the master derived component. @param ical: calendar data to process @type ical: L{Component} c sE t t � } | j t d � j � � � | j t t � j � � | S( Nt UID( R R R2 R t resourceUIDR R ( R& ( R R ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyt init_peruser_component� s t VTIMEZONEs RECURRENCE-IDs X-t keyc S s | d S( Ni ( ( t x( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyt <lambda>% t N( R R R R R R t duplicateR R t Falset _defaultMergeR1 R t PERUSER_PROPERTIESt startswitht IGNORE_X_PROPERTIESR2 t removePropertyt Truet PERUSER_SUBCOMPONENTSR# R R% t sortedt iteritemsR t _compactInstances( R R RD t componentsR t perinstance_componentsR R. t perinstance_componentt perinstance_id_differentR3 R( t peruser_component_differentt master_perinstancet perinstance( ( R R s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyR= � sT ? * ( c C s� | j � } | d k r d S| j � } x� t | j � � D]� } | j � d k s; | j � j d � rn q; n | j � } | d k r� q; n | j | d | �} | d k r; | | k r; | j | � q; q; Wd S( s� Remove recurrences instances that are the same as their master-derived counterparts. This gives the most compact representation of the calendar data. @param ical: calendar data to process @type ical: L{Component} NRE s X-t newcomp( R R t masterDerivedR R R RN R R! R ( R R t masterR^ R( R. t derived( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyRU 2 s ' c C sT | j � d k rP | j d � j � rP | j d � rP | j t d d � � n d S( s� Handle default property values during a merge. @param component: the iCalendar component to process @type component: L{Component} R4 R5 R t OPAQUEN( R R R8 R9 R2 R ( R R ( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyRL L s 7c C s? | j � r+ | j � r+ | j | | � n | j | | � d S( N( t isRecurringt _complexMerget _simpleMerge( R R? R@ ( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyR> Z s c C s | j � } | j � } | o- | j � d k } x� | j � D]� } | j � t k r= | j t � | j k r� | r� | j � } | r� | r� x? t | j � � D]( } | j � d k r� | j | � q� q� Wn t t | j � � � r� | j | � q� q� q= q= Wd S( N( Rb R R R R R R R R RJ R R R t lenR# ( R R? R@ t new_recurt old_recurt new_recur_has_no_masterR t newcomponentR( ( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyRd c s c C s� x� | j � D]� } | j � t k r | j t � | j k r� | j � } t g | j � D] } | j � | f ^ qY � } | j | j � � } x6 | j � D]( \ } } | | k r� | j | � q� q� Wt t | j � � � r� | j | � q� q� q q Wd S( N( R R R R R R RJ t dictR t validInstancesR RT R Re R R# ( R R? R@ R Ri R( t old_ridst valid_ridst old_rid( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyRc x s 1 ( R ( R ( t __name__t __module__t __doc__RM RR R t HIDDEN_INSTANCE_PROPERTYRO R R R R" R RA R= RU RL R> Rd Rc ( ( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyR : s $ 5 G N( t twistedcaldav.datafilters.filterR t twistedcaldav.icalR R R R R t __all__R ( ( ( s8 /opt/cpanel-ccs/twistedcaldav/datafilters/peruserdata.pyt <module> s ( $