Source code for registry.services.datastore.models

"""SQLAlchemy models for database integration."""

from datetime import datetime

from sqlalchemy import BigInteger, Column, DateTime, Enum, \
    ForeignKey, Index, Integer, SmallInteger, String, Text, text
from sqlalchemy.orm import relationship
from sqlalchemy.ext.declarative import declarative_base

from arxiv import taxonomy
from ... import domain

Base = declarative_base()


class DBClient(Base):
    """Persistence for :class:`domain.Client`."""

    __tablename__ = 'client'

    client_id = Column(Integer, primary_key=True, autoincrement=True)
    owner_id = Column(Integer)
    created = Column(DateTime, default=datetime.now)

    name = Column(String(255))
    url = Column(String(255))
    description = Column(Text)
    redirect_uri = Column(String(255), nullable=True)

    authorizations = relationship('DBClientAuthorization',
                                  back_populates='client', lazy='joined')
    credential = relationship('DBClientCredential', uselist=False,
                              back_populates='client', lazy='joined')
    grant_types = relationship('DBClientGrantType', back_populates='client',
                               lazy='joined')
    authorization_codes = relationship('DBAuthorizationCode',
                                       back_populates='client',
                                       lazy='joined')


class DBClientCredential(Base):
    """Persistence for :class:`domain.ClientCredential`."""

    __tablename__ = 'client_credential'

    credential_id = Column(Integer, primary_key=True, autoincrement=True)
    client_id = Column(ForeignKey('client.client_id'))
    client_secret = Column(String(255))
    created = Column(DateTime, default=datetime.now)

    client = relationship('DBClient', back_populates='credential',
                          uselist=False)


class DBClientAuthorization(Base):
    """Persistence for :class:`domain.ClientAuthorization`."""

    __tablename__ = 'client_authorization'

    authorization_id = Column(Integer, primary_key=True, autoincrement=True)
    client_id = Column(ForeignKey('client.client_id'))
    requested = Column(DateTime, default=datetime.now)
    authorized = Column(DateTime, nullable=True)
    scope = Column(String(2056))
    client = relationship('DBClient', back_populates='authorizations')


class DBClientGrantType(Base):
    """Persistence for :class:`domain.ClientGrantType`."""

    __tablename__ = 'client_grant_type'

    grant_type_id = Column(Integer, primary_key=True, autoincrement=True)
    client_id = Column(ForeignKey('client.client_id'))
    requested = Column(DateTime, default=datetime.now)
    authorized = Column(DateTime, nullable=True)
    grant_type = Column(Enum(*domain.ClientGrantType.GRANT_TYPES))
    client = relationship('DBClient', back_populates='grant_types')


class DBAuthorizationCode(Base):
    """Persistence for :class:`domain.AuthorizationCode`."""

    __tablename__ = 'authorization_code'

    user_id = Column(String(255))
    """The unique identifier of the arXiv user granting the authorization."""

    username = Column(String(255))
    """The username of the arXiv user granting the authorization."""

    user_email = Column(String(255))
    """The email address of the arXiv user granting the authorization."""

    client_id = Column(ForeignKey('client.client_id'), primary_key=True)
    """The unique identifier of the API client."""

    redirect_uri = Column(String(2056))
    """The URI to which the user should be redirected."""

    scope = Column(String(2056))
    """The scope authorized by the user."""

    code = Column(String(48), primary_key=True)
    """The authorization code itself."""

    created = Column(DateTime, default=datetime.now)
    """The time when the auth code was generated."""

    expires = Column(DateTime, default=datetime.now)
    """The time when the auth code expires."""

    client = relationship('DBClient', back_populates='authorization_codes')