Id;hddlmZddlZddlmZmZddlmZddlmZddl m Z m Z m Z m Z ddlmZddlmZmZGd d e ZGd d eZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddeZy) )IntEnumN)ABCMetaabstractmethod)datetime) ElementTree)DnMessageElementstring_to_time timestring)GUID)ndr_pack ndr_unpackc8eZdZdZddZedZedZy)FieldaMBase class for all fields. Each field will need to implement from_db_value and to_db_value. A field must correctly support converting both single valued fields, and list type fields. The only thing many=True does is say the field "prefers" to be a list, but really any field can be a list or single value. Nch||_||_||_|jr |g|_y||_y)aDCreates a new field, should be subclassed. :param name: Ldb field name. :param many: If true always convert field to a list when loaded. :param default: Default value or callback method (obj is first argument) :param hidden: If this is True, exclude the field when calling as_dict() N)namemanyhiddendefault)selfrrrrs C/usr/lib/python3/dist-packages/samba/netcmd/domain/models/fields.py__init__zField.__init__/s4    99DL"DLcy)zConverts value read from the database to Python value. :param ldb: Ldb connection :param value: MessageElement value from the database :returns: Parsed value as Python type N)rldbvalues r from_db_valuezField.from_db_valueBs rcy)a,Converts value to database value. This should return a MessageElement or None, where None means the field will be unset on the next save. :param value: Input value from Python field :param flags: MessageElement flags :returns: MessageElement or None Nr)rrflagss r to_db_valuezField.to_db_valueLs r)FNF)__name__ __module__ __qualname____doc__rrrr!rrrrr#s4 #&      rr) metaclassceZdZdZdZdZy) IntegerFieldz5A simple integer field, can be an int or list of int.c|yt|dkDs |jr|Dcgc] }t|c}St|dScc}w)z-Convert MessageElement to int or list of int.Nr)lenrintrrritems rrzIntegerField.from_db_value]sC =  Z!^tyy*/0$CI0 0uQx= 1Ac|yt|tr.t|Dcgc] }t|c}||jStt|||jScc}w)z-Convert int or list of int to MessageElement.N isinstancelistr strrrrr r.s rr!zIntegerField.to_db_valuef[ =  t $!',-tT-udiiA A"#e*eTYY? ?.A"Nr"r#r$r%rr!rrrr(r(Zs?!@rr(ceZdZdZdZdZy) BinaryFieldzSimilar to StringField but using bytes instead of str. This tends to be quite easy because a MessageElement already uses bytes. c|yt|dkDs |jr|Dcgc] }t|c}St|dScc}w)zConvert MessageElement to bytes or list of bytes. The values on the MessageElement should already be bytes so the cast to bytes() is likely not needed in from_db_value. Nr*r)r+rbytesr-s rrzBinaryField.from_db_valuewsE =  Z!^tyy,12DE$K2 2q? "3r/c|yt|tr.t|Dcgc] }t|c}||jStt|||jScc}w)z1Convert bytes or list of bytes to MessageElement.N)r2r3r r<rr5s rr!zBinaryField.to_db_values[ =  t $!)./t/ C C"%,tyyA A0r7Nr8rrrr:r:qs #Brr:ceZdZdZdZdZy) StringFieldz6A simple string field, may contain str or list of str.c|yt|dkDs |jr|Dcgc] }t|c}St|Scc}w)z-Convert MessageElement to str or list of str.Nr*r+rr4r-s rrzStringField.from_db_values? =  Z!^tyy*/0$CI0 0u: 1sAc|yt|tr.t|Dcgc] }t|c}||jStt|||jScc}w)z-Convert str or list of str to MessageElement.Nr1r5s rr!zStringField.to_db_valuer6r7Nr8rrrr?r?s@@rr?c6eZdZdZdfd ZdZdZdZxZS) EnumFieldz(A field based around Python's Enum type.c6||_t| |||y)z0Create a new EnumField for the given enum class.N)enumsuperr)rrrFrr __class__s rrzEnumField.__init__s  tW-rct|jtr#|jtt |S|jt |S)zvReturn Enum instance from value. Has a special case for IntEnum as the constructor only accepts int. ) issubclassrFrr,r4)rrs renum_from_valuezEnumField.enum_from_values< dii )99SU_- -99SZ( (rc|yt|dkDs |jr|Dcgc]}|j|c}S|j|Scc}w)z/Convert MessageElement to enum or list of enum.Nr*)r+rrKr-s rrzEnumField.from_db_valuesN =  Z!^tyy;@A4D((.A A''. .BsAc|yt|tr8t|Dcgc]}t|jc}||j Stt|j||j Scc}w)z/Convert enum or list of enum to MessageElement.N)r2r3r r4rrr5s rr!zEnumField.to_db_valuesd =  t $!-23TTZZ3UDIIG G"#ekk"2E499E E4A6FN) r"r#r$r%rrKrr! __classcell__rHs@rrDrDs2. )/FrrDceZdZdZdZdZy) DateTimeFieldz8A field for parsing ldb timestamps into Python datetime.c |yt|dkDs |jr5|Dcgc])}tjt t |+c}Stjt t |Scc}w)z7Convert MessageElement to datetime or list of datetime.Nr*)r+rr fromtimestampr r4r-s rrzDateTimeField.from_db_valuesk =  Z!^tyy %'**>#d)+DE' ')).U*DE E's.A:c >|yt|trJt|Dcgc])}tt t j |+c}||jSttt t j |||jScc}w)z7Convert datetime or list of datetime to MessageElement.N)r2r3r r r,r timestamprr5s rr!zDateTimeField.to_db_values =  t $!GLMtC 2 24 89:Mtyy" ""*S1C1CE1J-K"L"'4 4Ns.BNr8rrrrSrSsBF 4rrSc0eZdZdZdfd ZdZdZxZS) RelatedFieldzA field that automatically fetches the related objects. Use sparingly, can be a little slow. If in doubt just use DnField instead. c6||_t| |||y)z.Create a new RelatedField for the given model.N)modelrGr)rrr[rrrHs rrzRelatedField.__init__s  tW-rc "|yt|dkDs |jr>|Dcgc]2}|jj|t |t |4c}S|jj|t |t |Scc}w)zConvert Message element to related object or list of objects. Note that fetching related items is not using any sort of lazy loading so use this field sparingly. Nr*)dn)r+rr[getrr4r-s rrzRelatedField.from_db_valuesr =  Z!^tyyKPQ4DJJNN32c3t9+=N>Q Q::>>#"S#e**=>> >Rs7B c|yt|tr8t|Dcgc]}t|jc}||j Stt|j||j Scc}w)zConvert Dn object or list of Dn objects into a MessageElement.Nr1r5s rr!zDnField.to_db_valuer6r7Nr8rrrrara s? '@rraceZdZdZdZdZy) GUIDFieldz4A GUID field decodes fields containing binary GUIDs.c |yt|dkDs |jr'|Dcgc]}ttt|c}Sttt|dScc}w)z=Convert MessageElement with a GUID into a str or list of str.Nr*r)r+rr4rr r-s rrzGUIDField.from_db_value%sT =  Z!^tyy3KrreceZdZdZdZdZy) BooleanFieldz6A simple boolean field, can be a bool or list of bool.c|yt|dkDs |jr|Dcgc]}t|dk(c}St|dk(Scc}w)z3Convert MessageElement into a bool or list of bool.Nr*TRUErAr-s rrzBooleanField.from_db_value<sI =  Z!^tyy49:DCI': :u:' ';sAc *|yt|trEt|Dcgc]$}tt |j &c}||j Sttt |j ||j Scc}w)z3Convert bool or list of bool into a MessageElement.N)r2r3r r4boolupperrr5s rr!zBooleanField.to_db_valueEsv =  t $!5:;TT$Z&&(;UDIIO O"#d5k"2"8"8":E499M M}, {"ValueDisplayName: "Display name"}, {"ValueDescription: "Optional description or None for no description"}, {"Value": }, ] Note that the GUID needs to be created client-side when adding entries, leaving it as None then saving it doesn't generate the GUID. The field itself just converts the XML to list and vice versa, it doesn't automatically generate GUIDs for entries, this is entirely up to the caller. z http://www.w3.org/2001/XMLSchemaz)http://www.w3.org/2001/XMLSchema-instancezChttp://schemas.microsoft.com/2010/08/ActiveDirectory/PossibleValues)xsdxsic |tjt|}|jd|j}g}|j d|jD]}|j |jd|jj|jd|jj|jd|jj|jd|jjd|Sy) z/Parse MessageElement with XML to list of dicts.N StringListItem ValueGUIDValueDisplayNameValueDescriptionValue)rwrxryrz)r fromstringr4find NAMESPACEfindallappendtext)rrrroot string_listvaluesr.s rrz&PossibleClaimValuesField.from_db_valuejs  ))#e*5D))L$..AKF#++FDNNC  !%;!G!L!L(, 2D26..)BBF$(, 2D26..)BBF$!YYw?DD  M rc`|yt|tr|}n|g}t|dk(rytjd}|j j D]2\}}|dk(r|jd||jd||4tj|d}|D]}tj|d} tj| d } |d | _ tj| d } |d | _ tj| d } |d | _ tj| d } |d | _ tj}tj|j|d ddt|jjd ||j S)z6Convert list of dicts back to XML as a MessageElement.NrPossibleClaimValuesrsxmlnszxmlns:rurvrwrxryrzzutf-16TF)encodingxml_declarationshort_empty_elements)r2r3r+rElementr}itemsset SubElementrioBytesIOwriter getvaluedecoder)rrr possible_valuesrrurlr item_dictr. item_guid item_name item_desc item_valueouts rr!z$PossibleClaimValuesField.to_db_value}s =  eT "#O$gO  1 $ ""#89--/ /ID#rz#&6$#.  /",,T<@ ) 1I))+v>D#..t[AI&{3IN#..t5GHI&'9:IN#..t5GHI&'9:IN$//g>J'0JO 1jjl%++C5=<@AF , H clln33H=udiiPPrN)r"r#r$r%r}rr!rrrrprpPs$(2: QI &2Qrrp)rFrrabcrrr xml.etreerrrr r r samba.dcerpc.miscr samba.ndrr rrr(r:r?rDrSrYrarerirprrrrs. '!>>"*4 g4 n@5@.B%B<@%@.#F#FL4E44 C5 CF@e@2KK.N5N._Qu_Qr