)Jf AdZddlZddlZddlmZddlmZddlmZdZ GddeZ Gd d e Z Gd d e Z Gd 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)noncecfd}tdr& jdd|_n#t$rYnwxYwtdr j|_|S)Nc(|j|g|Ri|SN)_callInTransaction)selfargskwargsfuncs X/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/openid/store/sqlstore.pywrappedz_inTxn..wrappeds(&t&tTCDCCCFCCC__name____doc__)hasattrr TypeErrorr)r rs` r_inTxnrsDDDDDtZ   #}QRR0G      D tY', Ns - ::c eZdZdZdZdZddZdZdZdZ d Z d Z d Z d Z ee Zd ZeeZddZeeZdZeeZdZeeZdZeeZdZeeZdS)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_noncesNcL||_d|_i|_|p|j|p|jd|_d|_t|jdr!t|jdr |j|_t|jdrt|jdstddS)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) associationsnoncesi`TIntegrityErrorOperationalErrorzDError using database connection module (Maybe it can't be imported?)) conncur_statement_cacheassociations_table nonces_table _table_names max_nonce_ager exceptions RuntimeError)r r r#r$s r__init__zSQLStore.__init__Ds:  ".I$2I"7d&7  ) DI/ 0 0 ( #566 ("iDO)9:: @);<< @ ?@@ @ @ @rc|S)zSConvert a blob as returned by the SQL engine into a str object. str -> strr blobs r blobDecodezSQLStore.blobDecodevs  rc|S)z]Convert a str object into the necessary object for storing in the database as a blob.r+r ss r blobEncodezSQLStore.blobEncode|s rc |j|S#t$r)t||}||jz}||j|<|cYSwxYwr)r"KeyErrorgetattrr%)r sql_namesqls r_getSQLzSQLStore._getSQLsb (2 2   $))C 4$ $C.1D !( +JJJ  s 0AAc||}d}tt||}|j||dS)NcNt|trt|S|Sr) isinstancestr)args runicode_to_strz)SQLStore._execSQL..unicode_to_strs$#s## 3xx r)r8listmapr!execute)r r6r r7r>str_argss r_execSQLzSQLStore._execSQLsYll8$$     ND1122 h'''''rc|dddkr&|dddzfd}t|||Std|d)Ndb__sqlc jg|RSr)rC)r r r6s rr z"SQLStore.__getattr__..funcs$t}X55555rz Attribute z not found)setattrAttributeError)r attrr r6s` @r __getattr__zSQLStore.__getattr__sy 8u  ABBx&(H 6 6 6 6 6 6 D$ % % %K .TTT!DEE Erc|j |j|_ ||i|}|jd|_n%#|jd|_wxYw |jn#|jxYw|S)zExecute 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 rollbackcursorr!closecommit)r r r r rets rr zSQLStore._callInTransactions  y''))DH dD+F++       I         I    s"B#A#!B##"BB##B?cV||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_noncedb_create_assocr s rtxn_createTableszSQLStore.txn_createTabless.  rc|}|||j||j|j|j|jdS)zQSet the association for the server URL. Association -> NoneType N) db_set_assochandler2secretissuedlifetime assoc_type)r server_url associationas rtxn_storeAssociationzSQLStore.txn_storeAssociationsO  *ah//!(33QXqz, ( ( ( ( (rc||||n|||j}t |dkrdSg}|D]{}t |}||d|d<t|}|jdkr| ||j _| |j |f||r"| |ddSdS)zGet 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!fetchalllenr?r.r expiresIntxn_removeAssociationrZappendr\sort)r r_rZrowsrvaluesassocs rtxn_getAssociationzSQLStore.txn_getAssociations     j& 1 1 1 1   z * * *x  "" t99>>4L ? ?f OOF1I66q #V,?a''..z5<HHHH ''u(=>>>> !!####B'**trcN||||jjdkS)zRemove the association for the given server URL and handle, returning whether the association existed at all. (str, str) -> bool r)db_remove_assocr!rowcount)r r_rZs rrkzSQLStore.txn_removeAssociations* Z000x 1$$rct|tjz tjkrdS ||||dS#|jj$rYdSwxYw)zmReturn whether this nonce is present, and if it is, then remove it from the set. str -> boolFT)abstimerSKEW db_add_noncer'r)r r_ timestampsalts r txn_useNoncezSQLStore.txn_useNoncesv y49;;& ' '%* 4 45    j)T : : : 4 -   55 sAA! A!c|ttjtjz |jjSr)db_clean_nonceintrwrrxr!rtrVs rtxn_cleanupNonceszSQLStore.txn_cleanupNoncess6 C ,,uz9:::x  rc|ttj|jjSr)db_clean_assocrrwr!rtrVs rtxn_cleanupAssociationsz SQLStore.txn_cleanupAssociationss/ C ,,---x  r)NNr)r __module__ __qualname__rr#r$r)r.r2r8rCrLr rWr createTablesrbstoreAssociationrqgetAssociationrkremoveAssociationr|useNoncer cleanupNoncesrcleanupAssociationsr+rrrr"s<,L0@0@0@0@d  ((( F F F,6*++L(((v233>V.//N%%%455"vl##H!!!F,--M!!!!&!899rrcPeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zfd ZxZS) 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 < ?;c t|Sr) memoryviewr0s rr2zSQLiteStore.blobEncodeXs!}}rc tt|j|i|S#|jj$r.}t jdt|rYd}~dSd}~wwxYw)Nz^columns .* are not unique$F)superrrr'rrematchr<)r r r why __class__s rrzSQLiteStore.useNonce[sy 45d++4dEfEE E/   x5s3xx@@ uuuuu  s#A "AAA )rrrrcreate_nonce_sqlcreate_assoc_sql set_assoc_sqlget_assocs_sql get_assoc_sqlget_expired_sqlremove_assoc_sqlclean_assoc_sql add_nonce_sqlclean_nonce_sqlr2r __classcell__)rs@rrr"s 2MDN ELO?RO>MCO         rrc`eZdZdZ ddlZn #e$rdZYnwxYwdZdZdZ 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;)rrrrMySQLdbr' ImportErrorrrrrrrrrrrr+rrrrhs  $$$$$   >MGNNO JCTODMEOOOs  rceZdZdZ ddlZn%#e$rddlmZeYnwxYwdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZdZdZdZdZdS)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|||}|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). )rfr!rhridb_update_assoc db_new_assoc) r r_rZr[r\r]r^resultrns rrYzPostgreSQLStore.db_set_assocs"":v66x  "" t99 ;''*(2F<< < $$Z%-z;; ;rzCINSERT 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;rrrrrrrc$ddlm}||S)Nr)Binary)psycopg2r)r r-rs rr2zPostgreSQLStore.blobEncodes"######vd||rc*|Sr)tobytesr,s rr.zPostgreSQLStore.blobDecodes||~~r)rrrrrr psycopg2cffirregisterr'rrrY new_assoc_sqlupdate_assoc_sqlrrrrrrrr2r.r+rrrrs ''''''J ;;;&>MCGNNO JCTODMEO s  //r)rrrwopenid.associationropenid.store.interfacer openid.storerrrrrrr+rrrs? ******......    }:}:}:}:}:{}:}:}:@CCCCC(CCCL:F:F:F:F:F:F:F:FzYYYYYhYYYYYr