e4d32b86cb
- Add VocDataIn schema (date, topic, sub_topic, level, depart_id, dep_name) - Add RawVocData SQLAlchemy model (rawdata.raw_voc_data, BIGSERIAL PK) - Add POST /api/v1/voc-data endpoint with voc.data:write permission - Dual-write to local PostgreSQL + Supabase - Table auto-created on startup via Base.metadata.create_all()
128 lines
5.8 KiB
Python
128 lines
5.8 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import date, datetime
|
|
|
|
from sqlalchemy import BigInteger, Boolean, Date, 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 RawWaitingTime(Base):
|
|
__tablename__ = "raw_waiting_time"
|
|
__table_args__ = {"schema": "rawdata"}
|
|
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
|
vn: Mapped[int | None] = mapped_column(BigInteger, nullable=True)
|
|
txn: Mapped[int | None] = mapped_column(BigInteger, nullable=True)
|
|
hn: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
|
name: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
doctor_code: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
|
doctor_name: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
location_code: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
|
location_name: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
step_name: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
time: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, server_default=func.now())
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, server_default=func.now())
|
|
|
|
|
|
class PatientAppointment(Base):
|
|
__tablename__ = "patient_appointment"
|
|
__table_args__ = (
|
|
UniqueConstraint("hn", "date", "time", name="uq_patient_appointment_hn_date_time"),
|
|
{"schema": "rawdata"},
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True, autoincrement=True)
|
|
hn: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
txn: Mapped[int | None] = mapped_column(BigInteger, nullable=True)
|
|
date: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
|
|
time: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
|
|
doctor_code: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
|
period: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
|
appointment_type: Mapped[str | None] = mapped_column(String(100), nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, server_default=func.now())
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, server_default=func.now())
|
|
|
|
|
|
class RawVocData(Base):
|
|
__tablename__ = "raw_voc_data"
|
|
__table_args__ = {"schema": "rawdata"}
|
|
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True, autoincrement=True)
|
|
date: Mapped[date] = mapped_column(Date, nullable=False)
|
|
topic: Mapped[str] = mapped_column(String(200), nullable=False)
|
|
sub_topic: Mapped[str] = mapped_column(String(200), nullable=False)
|
|
level: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
depart_id: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
dep_name: Mapped[str | None] = mapped_column(String(200), nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
|
|
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(Text, nullable=False)
|
|
key_hash: Mapped[str] = mapped_column(Text, nullable=False)
|
|
encrypted_key: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
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")
|