a Šxd Aã@s„dZddlZddlZddlmZddlmZddlmZdd„Z Gdd „d eƒZ Gd d „d e ƒZ Gd d „d e ƒZ Gdd„de ƒZ dS)a. This module contains C{L{OpenIDStore}} implementations that use various SQL databases to back them. Example of how to initialize a store database:: python -c 'from openid.store import sqlstore; import pysqlite2.dbapi2;' 'sqlstore.SQLiteStore(pysqlite2.dbapi2.connect("cstore.db")).createTables()' éN)Ú Association)Ú OpenIDStore)ÚnoncecsT‡fdd„}tˆdƒr>zˆjdd…|_Wnty<Yn0tˆdƒrPˆj|_|S)Ncs|jˆ|g|¢Ri|¤ŽS©N)Ú_callInTransaction)ÚselfÚargsÚkwargs©Úfunc©úM/var/www/html/Ranjet/env/lib/python3.9/site-packages/openid/store/sqlstore.pyÚwrappedsz_inTxn..wrappedÚ__name__éÚ__doc__)ÚhasattrrÚ TypeErrorr)r rr r r Ú_inTxns    rc@sÄeZdZdZdZdZd!dd„Zdd„Zd d „Zd d „Z d d„Z dd„Z dd„Z dd„Z ee ƒZdd„ZeeƒZd"dd„ZeeƒZdd„ZeeƒZdd„ZeeƒZdd„ZeeƒZdd „ZeeƒZdS)#ÚSQLStoreaŒ This is the parent class for the SQL stores, which contains the logic common to all of the SQL stores. The table names used are determined by the class variables C{L{associations_table}} and C{L{nonces_table}}. To change the name of the tables used, pass new table names into the constructor. To create the tables with the proper schema, see the C{L{createTables}} method. This class shouldn't be used directly. Use one of its subclasses instead, as those contain the code necessary to use a specific database. All methods other than C{L{__init__}} and C{L{createTables}} should be considered implementation details. @cvar associations_table: This is the default name of the table to keep associations in @cvar nonces_table: This is the default name of the table to keep nonces in. @sort: __init__, createTables Zoid_associationsZ oid_noncesNcCst||_d|_i|_|p|j|p"|jdœ|_d|_t|jdƒrPt|jdƒrP|j|_t|jdƒrht|jdƒspt dƒ‚dS)a¸ This creates a new SQLStore instance. It requires an established database connection be given to it, and it allows overriding the default table names. @param conn: This must be an established connection to a database of the correct type for the SQLStore subclass you're using. @type conn: A python database API compatible connection object. @param associations_table: This is an optional parameter to specify the name of the table used for storing associations. The default value is specified in C{L{SQLStore.associations_table}}. @type associations_table: C{str} @param nonces_table: This is an optional parameter to specify the name of the table used for storing nonces. The default value is specified in C{L{SQLStore.nonces_table}}. @type nonces_table: C{str} N)Ú associationsZnoncesi`TÚIntegrityErrorÚOperationalErrorzDError using database connection module (Maybe it can't be imported?)) ÚconnÚcurÚ_statement_cacheÚassociations_tableÚ nonces_tableÚ _table_namesZ max_nonce_agerÚ exceptionsÚ RuntimeError)rrrrr r r Ú__init__Dsþ  ÿ  ÿzSQLStore.__init__cCs|S)zSConvert a blob as returned by the SQL engine into a str object. str -> strr ©rÚblobr r r Ú blobDecodevszSQLStore.blobDecodecCs|S)z]Convert a str object into the necessary object for storing in the database as a blob.r ©rÚsr r r Ú blobEncode|szSQLStore.blobEncodecCsFz |j|WSty@t||ƒ}||j;}||j|<|YS0dSr)rÚKeyErrorÚgetattrr)rÚsql_nameÚsqlr r r Ú_getSQLs     zSQLStore._getSQLcGs2| |¡}dd„}tt||ƒƒ}|j ||¡dS)NcSst|tƒrt|ƒS|SdSr)Ú isinstanceÚstr)Úargr r r Úunicode_to_str‘s z)SQLStore._execSQL..unicode_to_str)r,ÚlistÚmaprÚexecute)rr*rr+r0Zstr_argsr r r Ú_execSQLŠs zSQLStore._execSQLcsP|dd…dkr>|dd…d‰‡‡fdd„}tˆ||ƒ|Std|fƒ‚dS)NéZdb_Z_sqlcsˆjˆg|¢RŽSr)r4)r©rr*r r r ¡sz"SQLStore.__getattr__..funczAttribute %r not found)ÚsetattrÚAttributeError)rÚattrr r r6r Ú __getattr__šs  zSQLStore.__getattr__cOsx|j ¡zF|j ¡|_z"||i|¤Ž}W|j ¡d|_n|j ¡d|_0Wn|j ¡‚Yn 0|j ¡|S)z­Execute the given function inside of a transaction, with an open cursor. If no exception is raised, the transaction is comitted, otherwise it is rolled back.N)rÚrollbackÚcursorrÚcloseÚcommit)rr rr Úretr r r r©s   ÿ    zSQLStore._callInTransactioncCs| ¡| ¡dS)z¡ This method creates the database tables necessary for this store to work. It should not be called if the tables already exist. N)Zdb_create_nonceZdb_create_assoc©rr r r Útxn_createTables¿szSQLStore.txn_createTablescCs,|}| ||j| |j¡|j|j|j¡dS)zQSet the association for the server URL. Association -> NoneType N)Ú db_set_assocÚhandler'ÚsecretÚissuedÚlifetimeÚ assoc_type)rÚ server_urlZ associationÚar r r Útxn_storeAssociationÊs  þzSQLStore.txn_storeAssociationcCs´|dur| ||¡n | |¡|j ¡}t|ƒdkr:dSg}|D]P}t|ƒ}| |d¡|d<t|Ž}|jdkr‚|  ||j ¡qB|  |j |f¡qB|r¬|  ¡|ddSdSdS)zŠGet the most recent association that has been set for this server URL and handle. str -> NoneType or Association Nrééÿÿÿÿ)Ú db_get_assocZ db_get_assocsrÚfetchallÚlenr1r$rZ expiresInÚtxn_removeAssociationrCÚappendrEÚsort)rrHrCÚrowsrÚvaluesZassocr r r Útxn_getAssociationÖs$     zSQLStore.txn_getAssociationcCs| ||¡|jjdkS)zšRemove the association for the given server URL and handle, returning whether the association existed at all. (str, str) -> bool r)Zdb_remove_assocrÚrowcount)rrHrCr r r rP÷s zSQLStore.txn_removeAssociationcCsNt|t ¡ƒtjkrdSz| |||¡Wn|jjyDYdS0dSdS)zmReturn whether this nonce is present, and if it is, then remove it from the set. str -> boolFTN)ÚabsÚtimerÚSKEWZ db_add_noncerr)rrHÚ timestampZsaltr r r Ú txn_useNonceszSQLStore.txn_useNoncecCs | tt ¡ƒtj¡|jjSr)Zdb_clean_nonceÚintrXrrYrrVr@r r r Útxn_cleanupNoncesszSQLStore.txn_cleanupNoncescCs| tt ¡ƒ¡|jjSr)Zdb_clean_assocr\rXrrVr@r r r Útxn_cleanupAssociationssz SQLStore.txn_cleanupAssociations)NN)N)rÚ __module__Ú __qualname__rrrr!r$r'r,r4r:rrArZ createTablesrJZstoreAssociationrUZgetAssociationrPZremoveAssociationr[ÚuseNoncer]Z cleanupNoncesr^ZcleanupAssociationsr r r r r"s0 2     rcsPeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d d „Z‡fdd„Z‡ZS)Ú SQLiteStorez÷ This is an SQLite-based specialization of C{L{SQLStore}}. To create an instance, see C{L{SQLStore.__init__}}. To create the tables it will use, see C{L{SQLStore.createTables}}. All other methods are implementation details. z¤ CREATE TABLE %(nonces)s ( server_url VARCHAR, timestamp INTEGER, salt CHAR(40), UNIQUE(server_url, timestamp, salt) ); a CREATE TABLE %(associations)s ( server_url VARCHAR(2047), handle VARCHAR(255), secret BLOB(128), issued INTEGER, lifetime INTEGER, assoc_type VARCHAR(64), PRIMARY KEY (server_url, handle) ); z}INSERT OR REPLACE INTO %(associations)s (server_url, handle, secret, issued, lifetime, assoc_type) VALUES (?, ?, ?, ?, ?, ?);z_SELECT handle, secret, issued, lifetime, assoc_type FROM %(associations)s WHERE server_url = ?;znSELECT handle, secret, issued, lifetime, assoc_type FROM %(associations)s WHERE server_url = ? AND handle = ?;zDSELECT server_url FROM %(associations)s WHERE issued + lifetime < ?;zADELETE FROM %(associations)s WHERE server_url = ? AND handle = ?;z9DELETE FROM %(associations)s WHERE issued + lifetime < ?;z(INSERT INTO %(nonces)s VALUES (?, ?, ?);z+DELETE FROM %(nonces)s WHERE timestamp < ?;cCst|ƒSr)Ú memoryviewr%r r r r'XszSQLiteStore.blobEncodec sfztt|ƒj|i|¤ŽWS|jjy`}z,t dt|ƒ¡rJWYd}~dS‚WYd}~n d}~00dS)Nz^columns .* are not unique$F)ÚsuperrbrarrÚreÚmatchr.)rrr Úwhy©Ú __class__r r ra[s zSQLiteStore.useNonce)rr_r`rÚcreate_nonce_sqlÚcreate_assoc_sqlÚ set_assoc_sqlÚget_assocs_sqlÚ get_assoc_sqlÚget_expired_sqlÚremove_assoc_sqlÚclean_assoc_sqlÚ add_nonce_sqlÚclean_nonce_sqlr'raÚ __classcell__r r rhr rb"s   ÿrbc@s\eZdZdZz ddlZWney.dZYn0dZdZdZ dZ dZ d Z d Z d Zd Zd ZdS)Ú MySQLStorea& This is a MySQL-based specialization of C{L{SQLStore}}. Uses InnoDB tables for transaction support. To create an instance, see C{L{SQLStore.__init__}}. To create the tables it will use, see C{L{SQLStore.createTables}}. All other methods are implementation details. rNzÙ CREATE TABLE %(nonces)s ( server_url BLOB NOT NULL, timestamp INTEGER NOT NULL, salt CHAR(40) NOT NULL, PRIMARY KEY (server_url(255), timestamp, salt) ) ENGINE=InnoDB; aG CREATE TABLE %(associations)s ( server_url BLOB NOT NULL, handle VARCHAR(255) NOT NULL, secret BLOB NOT NULL, issued INTEGER NOT NULL, lifetime INTEGER NOT NULL, assoc_type VARCHAR(64) NOT NULL, PRIMARY KEY (server_url(255), handle) ) ENGINE=InnoDB; zDREPLACE INTO %(associations)s VALUES (%%s, %%s, %%s, %%s, %%s, %%s);úaSELECT handle, secret, issued, lifetime, assoc_type FROM %(associations)s WHERE server_url = %%s;úFSELECT server_url FROM %(associations)s WHERE issued + lifetime < %%s;úrSELECT handle, secret, issued, lifetime, assoc_type FROM %(associations)s WHERE server_url = %%s AND handle = %%s;úEDELETE FROM %(associations)s WHERE server_url = %%s AND handle = %%s;ú;DELETE FROM %(associations)s WHERE issued + lifetime < %%s;ú.INSERT INTO %(nonces)s VALUES (%%s, %%s, %%s);ú-DELETE FROM %(nonces)s WHERE timestamp < %%s;)rr_r`rZMySQLdbrÚ ImportErrorrjrkrlrmrornrprqrrrsr r r r ruhs      ÿruc@sŒeZdZdZz ddlZWn&ey>ddlmZe ¡Yn0dZ dZ dZ dd„Z d Z d Zd Zd Zd ZdZdZdZdZdd„Zdd„ZdS)ÚPostgreSQLStorezú This is a PostgreSQL-based specialization of C{L{SQLStore}}. To create an instance, see C{L{SQLStore.__init__}}. To create the tables it will use, see C{L{SQLStore.createTables}}. All other methods are implementation details. rN)ÚcompatzË CREATE TABLE %(nonces)s ( server_url VARCHAR(2047) NOT NULL, timestamp INTEGER NOT NULL, salt CHAR(40) NOT NULL, PRIMARY KEY (server_url, timestamp, salt) ); a… CREATE TABLE %(associations)s ( server_url VARCHAR(2047) NOT NULL, handle VARCHAR(255) NOT NULL, secret BYTEA NOT NULL, issued INTEGER NOT NULL, lifetime INTEGER NOT NULL, assoc_type VARCHAR(64) NOT NULL, PRIMARY KEY (server_url, handle), CONSTRAINT secret_length_constraint CHECK (LENGTH(secret) <= 128) ); c CsJ| ||¡}|j ¡}t|ƒr2| ||||||¡S| ||||||¡SdS)z¥ Set an association. This is implemented as a method because REPLACE INTO is not supported by PostgreSQL (and is not standard SQL). N)rMrrNrOZdb_update_assocZ db_new_assoc) rrHrCrDrErFrGÚresultrSr r r rBÎs   ÿ ÿzPostgreSQLStore.db_set_assoczCINSERT INTO %(associations)s VALUES (%%s, %%s, %%s, %%s, %%s, %%s);zUPDATE %(associations)s SET secret = %%s, issued = %%s, lifetime = %%s, assoc_type = %%s WHERE server_url = %%s AND handle = %%s;rvrwrxryrzr{r|cCsddlm}||ƒS)Nr)ÚBinary)Úpsycopg2r)rr#rr r r r'øs zPostgreSQLStore.blobEncodecCs| ¡Sr)Útobytesr"r r r r$ýszPostgreSQLStore.blobDecode)rr_r`rr‚r}Z psycopg2cffirÚregisterrrjrkrBZ new_assoc_sqlZupdate_assoc_sqlrmrornrprqrrrsr'r$r r r r r~¥s,     ÿr~)rrerXZopenid.associationrZopenid.store.interfacerZ openid.storerrrrbrur~r r r r Ús    F=