B d!@s@ddlmZmZmZddlZddlZddlZddlmZdZ d,ddZ e e d<d-d d Z e e d <GdddeZ e ZGdddeZGdddeZGdddeZGdddeZGdddeZd.ddZddZGd d!d!eZGd"d#d#eZejd$rddlZejd%kr>> def f(optional=libtbx.Auto) ... if optional is libtbx.Auto: ... optional = 5 ... return optional ... >>> print(f()) 5 >>> print(f(optional=10)) 10 NcCsdS)NAutor)selfrrr__str__8zAutoType.__str__cCs t||jS)N) isinstance __class__)r+otherrrr__eq__9szAutoType.__eq__cCs || S)N)r1)r+r0rrr__ne__;szAutoType.__ne__cCsttS)zQAutoType behaves as a singleton, so return the same hash value for all instances.)hashr))r+rrr__hash__=szAutoType.__hash__cs"|jdkrtt|||_|jS)N) singletonsuperr)__new__)cls)r/rrr7@s zAutoType.__new__) __name__ __module__ __qualname____doc__r5r,r1r2r4r7 __classcell__rr)r/rr)$sr)c@s$eZdZdZgZddZddZdS)slots_getstate_setstateaq Implements getstate and setstate for classes with __slots__ defined. Allows an object to easily pickle only certain attributes. Examples -------- >>> class sym_pair(libtbx.slots_getstate_setstate): ... __slots__ = ["i_seq", "j_seq"] ... def __init__(self, i_seq, j_seq): ... self.i_seq = i_seq ... self.j_seq = j_seq ... csrddl}|jdd}|j}z|dddlmWd||_||_XfddjD}tfdd|DS)a The name of some attributes may start with a double underscore such as cif_types.comp_comp_id.__rotamer_info. Python name mangling will rename such an attribute to _comp_comp_id_rotamer_info. Our __getstate__ function would then complain that the __slots__ list contains the non-existent attribute __rotamer_info. To fix this we manually mangle attributes with the compiler.misc.mangle function which does the right name mangling. rNignore)manglecsg|]}|jjqSr)r/r9).0name)r@r+rr nsz8slots_getstate_setstate.__getstate__..csg|]}|t|fqSr)getattr)rArB)r+rrrCps)warningsfilters showwarning simplefilterZ libtbx.utilsr@ __slots__dict)r+rEZwarning_filtersZ show_warningZmnamesr)r@r+r __getstate__Xs  z$slots_getstate_setstate.__getstate__cCs:x4|D](\}}t|tr&|d}t|||q WdS)Nutf8)itemsr.bytesdecodesetattr)r+staterBvaluerrr __setstate__rs  z$slots_getstate_setstate.__setstate__N)r9r:r;r<rIrKrSrrrrr>Gs r>c@seZdZdgZddZdS)mutablerRcCs ||_dS)N)rR)OrRrrr__init__{szmutable.__init__N)r9r:r;rIrVrrrrrTxsrTc@seZdZdZddZdS)+slots_getstate_setstate_default_initializera2 Merges together functionality from slots_getstate_setstate with adopt_optional_init_args. Examples -------- >>> class sym_pair(libtbx.slots_getstate_setstate_default_initializer): ... __slots__ = ["i_seq", "j_seq"] ... >>> svm_pair(i_seq=1, j_seq=2) >>> print(svm_pair.i_seq) 1 cKs.t|}x |D]}t||||dqWdS)N)rJrPget)r+kwdskeyrrrrVs z4slots_getstate_setstate_default_initializer.__init__N)r9r:r;r<rVrrrrrW~s rWc@s0eZdZdZddZddZddZdd Zd S) unpicklablez[ An inheritable class that raises a runtime exception that an object is unpicklable. cCstd|jjdS)Nz#pickling of %s objects is disabled.) RuntimeErrorr/r9)rUrrr _raise_errorszunpicklable._raise_errorcCs |dS)N)r])rUrrr__getinitargs__r-zunpicklable.__getinitargs__cCs |dS)N)r])rUrrrrKr-zunpicklable.__getstate__cCs |dS)N)r])rUrQrrrrSr-zunpicklable.__setstate__N)r9r:r;r<r]r^rKrSrrrrr[s r[c@seZdZddZdS)dict_with_default_0cCsdS)Nrr)r+rZrrr __missing__szdict_with_default_0.__missing__N)r9r:r;r`rrrrr_sr_rFcCsd|kr|d=n|d=x|D] }||=qW|dkrxx6|D]*}|drV|drVq>> class foo(object): ... def __init__(self, x, y=1, z=None): ... adopt_init_args(self, locals()) ... >>> a = foo('a', z=10) >>> assert a.x == 'a' >>> assert a.y == 1 >>> assert a.z == 10 r+rUFZ___N)keys startswithendswithhasattr__dict__AssertionErrorupdate)objargsZexcludeZhideZparamrZZ_keyrrradopt_init_argss  rkcCsDx>|D]2\}}t|j|s0td||jft|||q WdS)a Easy management of long list of arguments with default value passed to __init__. Parameters ---------- obj : object kwds : dict Examples -------- >>> class foo(object): ... z = 1 ... def __init__(self, **kwds): ... libtbx.adopt_optional_init_args(self, kwds) ... >>> a = foo() >>> assert a.z == 1 >>> a = foo(z=10) >>> assert a.z == 10 zh%s must be a class attribute of %s to be adopted as optional init argument by an instance of that class.N)rMrer/r\rP)rirYkr&rrradopt_optional_init_argss  rmc@s eZdZdZddZddZdS)ddaFcCs*|jrt||stdt|||dS)NzDynamic attributes disabled.)dynamic_attributes_disabledre TypeErrorr __setattr__)r+rlr&rrrrqszdda.__setattr__cCs d|_dS)NT)ro)r+rrrstop_dynamic_attributesszdda.stop_dynamic_attributesN)r9r:r;rorqrrrrrrrnsrnc@sTeZdZdZddZddZdddZd d Zd d Zd dZ dddZ ddZ dS) group_argsaT Class to build an arbitrary object from a list of keyword arguments. Examples -------- >>> from libtbx import group_args >>> obj = group_args(a=1, b=2, c=3) >>> print(obj.a, obj.b, obj.c) 1 2 3 Once stop_dynamic_attributes is called, adding new attributes won't be possible, that is this: obj.tmp=10 will fail. cKs|j|dS)N)rfrh)r+Zkeyword_argumentsrrrrVszgroup_args.__init__cCs|jS)N)rf)r+rrr__call__ szgroup_args.__call__NcCs|j||S)N)rfrX)r+kwZ default_valuerrrrX szgroup_args.getcCs |jS)N)rfrb)r+rrrrbszgroup_args.keyscCsbd}xXt|jD]F}t||}t|ddkrJ|d||f7}q|d||f7}qW|S)Nrsz ext.atom z %-30s : %s)sortedrfrbrDstrfindquote)r+ZoutlattrZtmprrr__repr__s zgroup_args.__repr__cCs|j|jdS)zG To merge other group_args into self. Overwrites matching fields!!!N)rfrh)r+r0rrrmergeszgroup_args.mergecCs||j|<dS)N)rf)r+rZrRrrradd!szgroup_args.addcCstf|S)z< produce shallow copy of self by converting to dict and back)rscopy)r+rrrr$szgroup_args.copy)N)NN) r9r:r;r<rVrtrXrbr|r}r~rrrrrrss   rsZLIBTBX_PRINT_TRACEcygwinopenrrvcCsNy t|||Stk r"Yntk r4YnXtj|}t|||S)N) builtin_openKeyboardInterrupt Exceptionospathrealpath)rBmode bufferingrrr open_realpath.s  rfile)r)rrrN)rF)rrv) __future__rrrZlibtbx.forward_compatibilitylibtbxrrrZmanual_date_stampr __builtins__r(r)r*r>rTrWr[rJr_rkrmrnrsenvironrXZlibtbx.start_print_traceplatformrrrrrrs<   !1 ( 5