o ãË·e 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ƒrz ˆjdd…|_Wn tyYnwtˆdƒr(ˆj|_|S)Ncs|jˆ|g|¢Ri|¤ŽS©N)Ú_callInTransaction)ÚselfÚargsÚkwargs©Úfunc©úL/var/www/ideatree/venv/lib/python3.10/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 Úoid_associationsÚ oid_noncesNcCst||_d|_i|_|p |j|p|jdœ|_d|_t|jdƒr(t|jdƒr(|j|_t|jdƒr4t|jdƒs8t 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)Ú associationsÚnoncesi`TÚIntegrityErrorÚOperationalErrorzDError using database connection module (Maybe it can't be imported?)) ÚconnÚcurÚ_statement_cacheÚassociations_tableÚ nonces_tableÚ _table_namesÚ max_nonce_agerÚ exceptionsÚ RuntimeError)rrrr r r r Ú__init__Ds þ  ÿ  ÿþzSQLStore.__init__cCó|S)zSConvert a blob as returned by the SQL engine into a str object. str -> strr ©rÚblobr r r Ú blobDecodevszSQLStore.blobDecodecCr&)z]Convert a str object into the necessary object for storing in the database as a blob.r ©rÚsr r r Ú blobEncode|szSQLStore.blobEncodecCsBz|j|WSty t||ƒ}||j;}||j|<|YSwr)rÚKeyErrorÚgetattrr!)rÚsql_nameÚsqlr r r Ú_getSQLs     üzSQLStore._getSQLcGs2| |¡}dd„}tt||ƒƒ}|j ||¡dS)NcSst|tƒr t|ƒS|Sr)Ú isinstanceÚstr)Úargr r r Úunicode_to_str‘s z)SQLStore._execSQL..unicode_to_str)r1ÚlistÚmaprÚexecute)rr/rr0r5Ústr_argsr r r Ú_execSQLŠs zSQLStore._execSQLcsL|dd…dkr|dd…d‰‡‡fdd„}tˆ||ƒ|Std|fƒ‚)NéÚdb_Ú_sqlcsˆjˆg|¢RŽSr)r:)r©rr/r r r ¡sz"SQLStore.__getattr__..funczAttribute %r not found)ÚsetattrÚAttributeError)rÚattrr r r>r Ú __getattr__šs  zSQLStore.__getattr__cOsr|j ¡z#|j ¡|_z||i|¤Ž}W|j ¡d|_n |j ¡d|_wWn |j ¡‚|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)Údb_create_nonceÚdb_create_assoc©rr r r Útxn_createTables¿s zSQLStore.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_urlÚ associationÚar r r Útxn_storeAssociationÊs  þzSQLStore.txn_storeAssociationcCs°|dur | ||¡n| |¡|j ¡}t|ƒdkrdSg}|D](}t|ƒ}| |d¡|d<t|Ž}|jdkrA|  ||j ¡q!|  |j |f¡q!|rV|  ¡|ddSdS)zŠGet the most recent association that has been set for this server URL and handle. str -> NoneType or Association Nrééÿÿÿÿ)Ú db_get_assocÚ db_get_assocsrÚfetchallÚlenr6r)rÚ expiresInÚtxn_removeAssociationrMÚappendrOÚsort)rrRrMÚrowsrÚvaluesÚassocr 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)Údb_remove_assocrÚrowcount)rrRrMr r r r]÷s  zSQLStore.txn_removeAssociationcCsHt|t ¡ƒtjkr dSz | |||¡WdS|jjy#YdSw)zmReturn whether this nonce is present, and if it is, then remove it from the set. str -> boolFT)ÚabsÚtimerÚSKEWÚ db_add_noncer#r)rrRÚ timestampÚsaltr r r Ú txn_useNoncesûþzSQLStore.txn_useNoncecCs | tt ¡ƒtj¡|jjSr)Údb_clean_nonceÚintrgrrhrrerJr r r Útxn_cleanupNoncesszSQLStore.txn_cleanupNoncescCs| tt ¡ƒ¡|jjSr)Údb_clean_assocrnrgrrerJr r r Útxn_cleanupAssociationssz SQLStore.txn_cleanupAssociations)NNr)rÚ __module__Ú __qualname__rrr r%r)r,r1r:rBrrKrÚ createTablesrUÚstoreAssociationrcÚgetAssociationr]ÚremoveAssociationrlÚuseNonceroÚ cleanupNoncesrqÚcleanupAssociationsr r r r r"s2 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,XózSQLiteStore.blobEncodec sVz tt|ƒj|i|¤ŽWS|jjy*}zt dt|ƒ¡r%WYd}~dS‚d}~ww)Nz^columns .* are not unique$F)Úsuperr{rxr#rÚreÚmatchr3)rrr Úwhy©Ú __class__r r rx[s€üzSQLiteStore.useNonce)rrrrsrÚ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,rxÚ __classcell__r r r‚r r{"s   ÿr{c@s\eZdZdZzddlZWn eydZYnwdZdZdZ 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;)rrrrsrÚMySQLdbr#Ú ImportErrorr„r…r†r‡r‰rˆrŠr‹rŒrr r r r rhs$  ÿ ÿrc@sŒeZdZdZzddlZWneyddlmZe ¡YnwdZ 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 CsF| ||¡}|j ¡}t|ƒr| ||||||¡S| ||||||¡S)z¥ Set an association. This is implemented as a method because REPLACE INTO is not supported by PostgreSQL (and is not standard SQL). )rXrrZr[Údb_update_assocÚ db_new_assoc) rrRrMrNrOrPrQÚresultr`r r r rLÎ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;rr‘r’r“r”r•r–cCsddlm}||ƒS)Nr)ÚBinary)Úpsycopg2rž)rr(ržr r r r,øs zPostgreSQLStore.blobEncodecCs| ¡Sr)Útobytesr'r r r r)ýr}zPostgreSQLStore.blobDecode)rrrrsrrŸr˜Ú psycopg2cffiršÚregisterr#r„r…rLÚ new_assoc_sqlÚupdate_assoc_sqlr‡r‰rˆrŠr‹rŒrr,r)r r r r r™¥s0    þ ÿ r™)rrrgÚopenid.associationrÚopenid.store.interfacerÚ openid.storerrrr{rr™r r r r Ús    F=