U Acj@sXdZddlmZmZddgZGdddZGdddZGdd d ZGd ddZd S) zM L{twisted.python.fakepwd} provides a fake implementation of the L{pwd} API. )ListOptional UserDatabaseShadowDatabasec @sBeZdZdZeeeeeeeddddZedddZd d ZdS) _UserRecordz L{_UserRecord} holds the user data for a single user in L{UserDatabase}. It corresponds to the C{passwd} structure from the L{pwd} module. See that module for attribute documentation. N)namepassworduidgidgecoshomeshellreturncCs.||_||_||_||_||_||_||_dSNpw_name pw_passwdpw_uidpw_gidpw_gecospw_dirpw_shell)selfrrr r r r r r:/tmp/pip-unpacked-wheel-nrvc_22y/twisted/python/fakepwd.py__init__s z_UserRecord.__init__rcCsdS)Nrrrrr__len__'sz_UserRecord.__len__cCs$|j|j|j|j|j|j|jf|Srrrindexrrr __getitem__*sz_UserRecord.__getitem__ __name__ __module__ __qualname____doc__strintrrr"rrrrrs rc @seZdZUdZeeed<dZeed<dZ eed<ddd d Z de e e ee ee e e ddddZ eedddZe edddZeedddZdS)rz L{UserDatabase} holds a traditional POSIX user data in memory and makes it available via the same API as L{pwd}. @ivar _users: A C{list} of L{_UserRecord} instances holding all user data added to this database. _usersiu'_lastUIDiN_lastGIDNrcCs g|_dSrr*rrrrrCszUserDatabase.__init__r/bin/sh)usernamerr r r r r rc Cs\|dkr|j}|jd7_|dkr8|j}|jd7_t|||||||}|j|dS)a Add a new user record to this database. @param username: The value for the C{pw_name} field of the user record to add. @param password: The value for the C{pw_passwd} field of the user record to add. @param uid: The value for the C{pw_uid} field of the user record to add. @param gid: The value for the C{pw_gid} field of the user record to add. @param gecos: The value for the C{pw_gecos} field of the user record to add. @param home: The value for the C{pw_dir} field of the user record to add. @param shell: The value for the C{pw_shell} field of the user record to add. N)r+r,rr*append) rr0rr r r r r ZnewUserrrraddUserFs"zUserDatabase.addUser)r rcCs(|jD]}|j|kr|SqtdS)zH Return the user record corresponding to the given uid. N)r*rKeyError)rr entryrrrgetpwuidqs   zUserDatabase.getpwuid)rrcCsDt|tstdt||jD]}|j|kr"|Sq"tdS)zM Return the user record corresponding to the given username. z%getpwuam() argument must be str, not N) isinstancer( TypeErrortyper*rr4)rrr5rrrgetpwnamzs     zUserDatabase.getpwnamcCs|jS)z4 Return a list of all user records. r-rrrrgetpwallszUserDatabase.getpwall)rNNr.r.r/)r$r%r&r'rr__annotations__r+r)r,rr(rr3r6r:r;rrrrr6s0     +  c @sFeZdZdZeeeeeeeeedd ddZedddZd d ZdS) _ShadowRecordz L{_ShadowRecord} holds the shadow user data for a single user in L{ShadowDatabase}. It corresponds to C{spwd.struct_spwd}. See that class for attribute documentation. N r0r lastChangeminmaxwarninactexpireflagrc Cs:||_||_||_||_||_||_||_||_| |_dSr sp_namZsp_pwdZ sp_lstchgZsp_minZsp_maxZsp_warnZsp_inactZ sp_expireZsp_flag rr0rr?r@rArBrCrDrErrrrs z_ShadowRecord.__init__rcCsdS)N rrrrrrsz_ShadowRecord.__len__c Cs,|j|j|j|j|j|j|j|j|jf |SrrFr rrrr"s z_ShadowRecord.__getitem__r#rrrrr=s r=c @sdeZdZUdZeeed<ddddZeee e e e e e e dd dd Z eed d d Z d dZ dS)rz L{ShadowDatabase} holds a shadow user database in memory and makes it available via the same API as C{spwd}. @ivar _users: A C{list} of L{_ShadowRecord} instances holding all user data added to this database. @since: 12.0 r*NrcCs g|_dSrr-rrrrrszShadowDatabase.__init__r>c Cs$|jt||||||||| dS)a Add a new user record to this database. @param username: The value for the C{sp_nam} field of the user record to add. @param password: The value for the C{sp_pwd} field of the user record to add. @param lastChange: The value for the C{sp_lstchg} field of the user record to add. @param min: The value for the C{sp_min} field of the user record to add. @param max: The value for the C{sp_max} field of the user record to add. @param warn: The value for the C{sp_warn} field of the user record to add. @param inact: The value for the C{sp_inact} field of the user record to add. @param expire: The value for the C{sp_expire} field of the user record to add. @param flag: The value for the C{sp_flag} field of the user record to add. N)r*r2r=rHrrrr3s(zShadowDatabase.addUser)r0rcCsFt|tstdt||jD]}|j|kr"|Sq"t|dS)zT Return the shadow user record corresponding to the given username. z%getspnam() argument must be str, not N)r7r(r8r9r*rGr4)rr0r5rrrgetspnams     zShadowDatabase.getspnamcCs|jS)z; Return a list of all shadow user records. r-rrrrgetspallszShadowDatabase.getspall) r$r%r&r'rr=r<rr(r)r3rJrKrrrrrs   . N) r'typingrr__all__rrr=rrrrrs (V.