a ù!f~ã@sNdZddlmZddlZddlmZGdd„dejjƒZ Gdd„dej ƒZ dS) a¥OAuth 2.0 utilities for SQLAlchemy. Utilities for using OAuth 2.0 in conjunction with a SQLAlchemy. Configuration ============= In order to use this storage, you'll need to create table with :class:`oauth2client.contrib.sqlalchemy.CredentialsType` column. It's recommended to either put this column on some sort of user info table or put the column in a table with a belongs-to relationship to a user info table. Here's an example of a simple table with a :class:`CredentialsType` column that's related to a user table by the `user_id` key. .. code-block:: python from sqlalchemy import Column, ForeignKey, Integer from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import relationship from oauth2client.contrib.sqlalchemy import CredentialsType Base = declarative_base() class Credentials(Base): __tablename__ = 'credentials' user_id = Column(Integer, ForeignKey('user.id')) credentials = Column(CredentialsType) class User(Base): id = Column(Integer, primary_key=True) # bunch of other columns credentials = relationship('Credentials') Usage ===== With tables ready, you are now able to store credentials in database. We will reuse tables defined above. .. code-block:: python from sqlalchemy.orm import Session from oauth2client.client import OAuth2Credentials from oauth2client.contrib.sql_alchemy import Storage session = Session() user = session.query(User).first() storage = Storage( session=session, model_class=Credentials, # This is the key column used to identify # the row that stores the credentials. key_name='user_id', key_value=user.id, property_name='credentials', ) # Store credentials = OAuth2Credentials(...) storage.put(credentials) # Retrieve credentials = storage.get() # Delete storage.delete() é)Úabsolute_importN)Úclientc@seZdZdZdS)ÚCredentialsTypezXType representing credentials. Alias for :class:`sqlalchemy.types.PickleType`. N)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r ú`/var/www/html/python-backend/venv/lib/python3.9/site-packages/oauth2client/contrib/sqlalchemy.pyrdsrcs8eZdZdZ‡fdd„Zdd„Zdd„Zdd „Z‡ZS) ÚStoragez²Store and retrieve a single credential to and from SQLAlchemy. This helper presumes the Credentials have been stored as a Credentials column on a db model class. cs0tt|ƒ ¡||_||_||_||_||_dS)a?Constructor for Storage. Args: session: An instance of :class:`sqlalchemy.orm.Session`. model_class: SQLAlchemy declarative mapping. key_name: string, key name for the entity that has the credentials key_value: key value for the entity that has the credentials property_name: A string indicating which property on the ``model_class`` to store the credentials. This property must be a :class:`CredentialsType` column. N)Úsuperr Ú__init__ÚsessionÚ model_classÚkey_nameÚ key_valueÚ property_name)Úselfrrrrr©Ú __class__r r r rs zStorage.__init__cCsb|j|ji}|j |j¡jfi|¤Ž}| ¡}|rZt||jƒ}|rVt |dƒrV|  |¡|SdSdS)zzRetrieve stored credential. Returns: A :class:`oauth2client.Credentials` instance or `None`. Ú set_storeN) rrrÚqueryrÚ filter_byÚfirstÚgetattrrÚhasattrr)rÚfiltersrÚentityZ credentialr r r Ú locked_getˆs   zStorage.locked_getcCs`|j|ji}|j |j¡jfi|¤Ž}| ¡}|sB|jfi|¤Ž}t||j|ƒ|j  |¡dS)zƒWrite a credentials to the SQLAlchemy datastore. Args: credentials: :class:`oauth2client.Credentials` N) rrrrrrrÚsetattrrÚadd)rÚ credentialsrrrr r r Ú locked_putšs zStorage.locked_putcCs.|j|ji}|j |j¡jfi|¤Ž ¡dS)z1Delete credentials from the SQLAlchemy datastore.N)rrrrrrÚdelete)rrr r r Ú locked_deleteªs zStorage.locked_delete) rrrrr rr"r$Ú __classcell__r r rr r ks  r ) rÚ __future__rZsqlalchemy.typesZ sqlalchemyZ oauth2clientrÚtypesZ PickleTyperr r r r r Ús N