Id6<ddlZddlmZmZddlmZmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZmZmZmZddlmZmZmZmZmZmZiZGdd eZGd d e Zy) N)ABCMetaabstractmethod) ERR_NO_SUCH_OBJECT FLAG_MOD_ADDFLAG_MOD_REPLACELdbErrorMessageMessageElement SCOPE_BASE SCOPE_SUBTREE binary_encode)SDUtils) DeleteError DoesNotExistMultipleObjectsReturned ProtectErrorUnprotectError) DateTimeFieldDnFieldField GUIDField IntegerField StringFieldceZdZfdZxZS) ModelMetac t|||||fi|}|jdk7r5tt j |d|_||_|t|<|S)NModelc"t|tSN) isinstancer)fs B/usr/lib/python3/dist-packages/samba/netcmd/domain/models/model.pyz#ModelMeta.__new__...s 1e@T) super__new____name__dictinspect getmembersfieldsmetaMODELS)mclsnamebases namespacekwargscls __class__s r#r'zModelMeta.__new__*sWgodD%EfE <<7 "g006TUVCJCHF4L r%)r( __module__ __qualname__r' __classcell__)r5s@r#rr(s r%rceZdZedZedZedZeddZ e dZ edZ ed Z ed d Zed Ze ddZe ddZeddZeddZdZdZdZdZdZeedZedZeedZedZ dZ!d(dZ"d)dZ#edZ$ed Z%ed!Z&ed"Z'ed(d#Z(d$Z)d%Z*d&Z+d'Z,y)*rcndistinguishedNamedndsCorePropagationDataT)hidden instanceTyper0objectCategory objectClassc"|jSr )get_object_class)objs r#r$zModel.?s33G3G3Ir%)default objectGUID uSNChanged uSNCreated whenChanged whenCreatedc d|_|jjD]O\}}||vr||}n3t|jr|j |}n |j}t |||Qy)zCreate a new model instance and optionally populate fields. Does not save the object to the database, call .save() for that. :param kwargs: Optional input fields to populate object with N)_messager,itemscallablerEsetattr)selfr3 field_namefieldrEs r#__init__zModel.__init__Fsm !%!2!2!4 / JV# ,%--(----- D*g . /r%c<d|jjd|dS)z,Return object representation for this model.)r5r(rPs r#__repr__zModel.__repr__[s"4>>**+2dV155r%c,t|jS)z4Stringify model instance to implement in each model.)strr:rWs r#__str__z Model.__str___s477|r%c:|y|j|jk(S)zBasic object equality check only really checks if the dn matches. :param other: The other object to compare with Fr<)rPothers r#__eq__z Model.__eq__cs =77ehh& &r%cF|jt|jSy)zAutomatically called by custom JSONEncoder class. When turning an object into json any fields of type RelatedField will also end up calling this method. N)r<rZrWs r#__json__zModel.__json__ms! 77 tww<  r%cy)zReturn the base DN for the container of this model. :param ldb: Ldb connection :return: Dn to use for new objects N)ldbs r# get_base_dnzModel.get_base_dnvs r%c$|j|S)zReturn the DN used for querying. By default, this just calls get_base_dn, but it is possible to return a different Dn for querying. :param ldb: Ldb connection :return: Dn to use for searching )re)r4rds r# get_search_dnzModel.get_search_dnss##r%cy)z'Returns the objectClass for this model.Nrcrcr%r#rCzModel.get_object_classs r%c8|}|j|||S)zCreate a new model instance from the Ldb Message object. :param ldb: Ldb connection :param message: Ldb Message object to create instance from )_apply)r4rdmessagerDs r# from_messagezModel.from_messagese 3  r%c ||_|jjD]>\}}|j|vst |||j |||j@y)zInternal method to apply Ldb Message to current object. :param ldb: Ldb connection :param message: Ldb Message object to apply N)rLr,rMr0rO from_db_value)rPrdrkattrrRs r#rjz Model._applys[  ;;,,. SKD%zzW$dE$7$7WUZZ=P$QR Sr%NcT|r'|Dcgc]}|j|jc}nd} |j|jt|}|j||dycc}w#t $r4}|j dtk(rtd|jd}~wwxYw)zRefresh object from database. :param ldb: Ldb connection :param fields: Optional list of field names to refresh N)scopeattrsrzRefresh failed, object gone: ) r,r0searchr<r rargsrrrj)rPrdr,r"rrreses r#refreshz Model.refreshs :@f5Q$$5T **TWWJe*DC CQ 6  vvay.."%B477)#LMM  s A%"A** B'3/B""B'ci}|jjD]2\}}|jr|st||}|$|||j<4|S)zReturns a dict representation of the model. :param include_hidden: Include fields with hidden=True when set :returns: dict representation of model using Ldb field names as keys )r,rMr>getattrr0)rPinclude_hiddenobj_dictrorRvalues r#as_dictz Model.as_dictsY ;;,,. 1KD%<<>d+$+0HUZZ(  1 r%c Dt|}|j|d<t|}|dk(rdnd}|jD]P\}}|jj |}|st d|d|d|jdt|d z }R|dkDr|d z }|S) zyBuild LDAP search expression from kwargs. :kwargs: fields to use for expression using model field names object_classrz(&zUnknown field ''(=)) r)rClenrMr,get ValueErrorr0r )r4r3criteria num_fields expressionrQr|rRs r#build_expressionzModel.build_expressions<##7#9 ] %?R !)!1 C JJJNN:.E ?:,a!@AA Aejj\=+?*@B BJ  C > # Jr%c +"K|j|} |j|t|jdi|}|D]}|j||y#t$r*}|j dt k(rtd|d}~wwxYww)zReturns a search query for this model. :param ldb: Ldb connection :param kwargs: Search criteria as keyword args rqrrContainer does not exist: Nrc) rgrsr rrrtrrrl)r4rdr3base_dnresultrvrks r#queryz Model.querys##C( ZZ&3+?3+?+?+I&+I KF 1G""30 0 1 vvay.."%?y#IJJ  s'B(AB B "%BB  Bc |jd}|r |j|t}n:|j |} |j|t|jd i|}t|}|dkDrtd|d |dk(r|j||dSy#t$r!}|jdt k(rYd}~yd}~wwxYw#t$r*}|jdt k(rtd|d}~wwxYw) aGet one object, must always return one item. Either find object by dn=, or any combination of attributes via kwargs. If there are more than one result, MultipleObjectsReturned is raised. :param ldb: Ldb connection :param kwargs: Search criteria as keyword args :returns: User object or None if not found :raises: MultipleObjects returned if there are more than one results r<rqrNrrrz#More than one object returned (got z).rc) rrsr rrtrrgr rrrrrl)r4rdr3r<rurvrcounts r#rz Model.gets ZZ   jj:j6'',G jj'4,@C,@,@,J6,J!LC 19)5eWB?A A aZ##CQ0 01 66!9 22   66!9 22&)CG9'MNN s4B!(C! C *CCC  D%C<<Dc 8|di|}|j||S)zCreate object constructs object and calls save straight after. :param ldb: Ldb connection :param kwargs: Fields to populate object from :returns: object rc)save)r4rdr3rDs r#createz Model.create/smFm   r%c |j|fi|}|3t|}||j||j|fi|dfS|dfS)a!Retrieve object and if it doesn't exist create a new instance. :param ldb: Ldb connection :param defaults: Attributes only used for create but not search :param kwargs: Attributes used for searching existing object :returns: (object, bool created) TF)rr)updater)r4rddefaultsr3rDrrs r# get_or_createzModel.get_or_create;s^cggc$V$ ;LE# X&3::c+U+T1 1: r%c|j|j|}|jd|jxs |j||_t |j}|j jD]M\}}|dk7s t||}|j|t}|1t|s=|j|O|j||j|t}|j||dy|j!||j"} t |j}|j jD]q\}}|dk7s t||}t| |} || k7s*|j|t$}|dgfvrt'gt$|j}|j|st|r#|j)||j+|yy)aSave model to Ldb database. The save operation will save all fields excluding fields that return None when calling their `to_db_value` methods. The `to_db_value` method can either return a ldb Message object, or None if the field is to be excluded. For updates, the existing object is fetched and only fields that are changed are included in the update ldb Message. Also for updates, any fields that currently have a value, but are to be set to None will be seen as a delete operation. After the save operation the object is refreshed from the server, as often the server will populate some fields. :param ldb: Ldb connection NzCN=r]r<rr)r<re add_childr:r0r r,rMry to_db_valuerraddrsr rjrlrLrr modifyrw) rPrdr<rkrorRr|db_valueru existing_obj old_values r#rz Model.saveMs( 77?!!#&B LL3tww3$))45 6DG)G#{{002 . e4<#D$/E$00 EH +H  H- . GGG **Rz*2C KKSV $ ,,S$--@L)G#{{002 . e4<#D$/E ' d ;I )#(#4#4UrsY.'FFF"!!   CEiCEr%