75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import BigInteger, Boolean, DateTime, ForeignKey, Integer, String, Text, UniqueConstraint, func
|
|
from sqlalchemy.dialects.postgresql import JSONB
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.db.base import Base
|
|
|
|
|
|
class RawOpdCheckpoint(Base):
|
|
__tablename__ = "raw_opd_checkpoint"
|
|
__table_args__ = (
|
|
UniqueConstraint("hn", "vn", "location", name="uq_raw_opd_checkpoint_hn_vn_location"),
|
|
{"schema": "rawdata"},
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
|
hn: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
|
vn: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
|
location: Mapped[str] = mapped_column(Text, nullable=False)
|
|
type: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
timestamp_in: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
|
|
timestamp_out: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
waiting_time: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
bu: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
|
|
|
|
class ApiClient(Base):
|
|
__tablename__ = "api_client"
|
|
__table_args__ = {"schema": "fastapi"}
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
name: Mapped[str] = mapped_column(String(128), unique=True, nullable=False)
|
|
is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
|
|
|
api_keys: Mapped[list[ApiKey]] = relationship(
|
|
back_populates="client",
|
|
cascade="all, delete-orphan",
|
|
passive_deletes=True,
|
|
)
|
|
|
|
def __str__(self) -> str:
|
|
client_id = getattr(self, "id", None)
|
|
if client_id is None:
|
|
return self.name
|
|
return f"{self.name} ({client_id})"
|
|
|
|
def __repr__(self) -> str:
|
|
return str(self)
|
|
|
|
|
|
class ApiKey(Base):
|
|
__tablename__ = "api_key"
|
|
__table_args__ = {"schema": "fastapi"}
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
client_id: Mapped[int] = mapped_column(
|
|
ForeignKey("fastapi.api_client.id", ondelete="CASCADE"), nullable=False
|
|
)
|
|
name: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
|
|
key_prefix: Mapped[str] = mapped_column(String(12), nullable=False)
|
|
key_hash: Mapped[str] = mapped_column(Text, nullable=False)
|
|
|
|
permissions: Mapped[list[str]] = mapped_column(JSONB, nullable=False, default=list)
|
|
is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), nullable=False, server_default=func.now()
|
|
)
|
|
|
|
client: Mapped[ApiClient] = relationship(back_populates="api_keys")
|