You've already forked RekomenciBackend
gitgernit
86 lines
3.1 KiB
Python
86 lines
3.1 KiB
Python
from sqlalchemy import (
|
|
Boolean,
|
|
Column,
|
|
DateTime,
|
|
Enum,
|
|
ForeignKey,
|
|
MetaData,
|
|
String,
|
|
Table,
|
|
UniqueConstraint,
|
|
)
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import registry
|
|
|
|
from template_project.application.access_token.entity import AccessToken
|
|
from template_project.application.auth_identity.entity import AuthIdentity, AuthMethod
|
|
from template_project.application.notification_device.entity import NotificationDevice
|
|
from template_project.application.user.entity import User
|
|
from template_project.application.user.profile.entity import Profile
|
|
|
|
meta_data = MetaData()
|
|
mapper_registry = registry()
|
|
|
|
user_table = Table(
|
|
"users",
|
|
meta_data,
|
|
Column("id", UUID, primary_key=True),
|
|
Column("deleted_at", DateTime(timezone=True)),
|
|
Column("created_at", DateTime(timezone=True), nullable=False),
|
|
)
|
|
|
|
access_token_table = Table(
|
|
"access_token",
|
|
meta_data,
|
|
Column("id", UUID, primary_key=True),
|
|
Column("user_id", UUID, ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
|
|
Column("revoked", Boolean, nullable=False),
|
|
Column("expires_in", DateTime(timezone=True), nullable=False),
|
|
Column("deleted_at", DateTime(timezone=True)),
|
|
Column("created_at", DateTime(timezone=True), nullable=False),
|
|
)
|
|
|
|
auth_identity_table = Table(
|
|
"auth_identities",
|
|
meta_data,
|
|
Column("id", UUID, primary_key=True),
|
|
Column("user_id", UUID, ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
|
|
Column("method", Enum(AuthMethod, name="auth_method"), nullable=False),
|
|
Column("identifier", String, nullable=False),
|
|
Column("secret_key", String, nullable=True),
|
|
Column("created_at", DateTime(timezone=True), nullable=False),
|
|
UniqueConstraint("method", "identifier", name="uq_auth_method_identifier"),
|
|
)
|
|
|
|
profile_table = Table(
|
|
"profiles",
|
|
meta_data,
|
|
Column("id", UUID, primary_key=True),
|
|
Column("user_id", UUID, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, unique=True),
|
|
Column("email", String, nullable=True),
|
|
Column("display_name", String, nullable=True),
|
|
Column("first_name", String, nullable=True),
|
|
Column("last_name", String, nullable=True),
|
|
Column("avatar_url", String, nullable=True),
|
|
Column("phone", String, nullable=True),
|
|
Column("deleted_at", DateTime(timezone=True)),
|
|
Column("created_at", DateTime(timezone=True), nullable=False),
|
|
)
|
|
|
|
notification_device_table = Table(
|
|
"notification_devices",
|
|
meta_data,
|
|
Column("id", UUID, primary_key=True),
|
|
Column("user_id", UUID, ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
|
|
Column("device_id", String, nullable=False),
|
|
Column("deleted_at", DateTime(timezone=True)),
|
|
Column("created_at", DateTime(timezone=True), nullable=False),
|
|
UniqueConstraint("user_id", "device_id", name="uq_user_device"),
|
|
)
|
|
|
|
mapper_registry.map_imperatively(User, user_table)
|
|
mapper_registry.map_imperatively(AccessToken, access_token_table)
|
|
mapper_registry.map_imperatively(AuthIdentity, auth_identity_table)
|
|
mapper_registry.map_imperatively(Profile, profile_table)
|
|
mapper_registry.map_imperatively(NotificationDevice, notification_device_table)
|