feat: add Leave model with auto-migration
Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -53,6 +53,26 @@ async def lifespan(app: FastAPI):
|
|||||||
await conn.run_sync(lambda c: c.exec_driver_sql(
|
await conn.run_sync(lambda c: c.exec_driver_sql(
|
||||||
"CREATE TABLE IF NOT EXISTS system_config (key VARCHAR(64) PRIMARY KEY, value TEXT DEFAULT '')"
|
"CREATE TABLE IF NOT EXISTS system_config (key VARCHAR(64) PRIMARY KEY, value TEXT DEFAULT '')"
|
||||||
))
|
))
|
||||||
|
# leaves table for v0.7
|
||||||
|
await conn.run_sync(lambda c: c.exec_driver_sql(
|
||||||
|
"CREATE TABLE IF NOT EXISTS leaves ("
|
||||||
|
" id UUID PRIMARY KEY DEFAULT gen_random_uuid(),"
|
||||||
|
" manager_id UUID NOT NULL REFERENCES users(id),"
|
||||||
|
" leave_type VARCHAR(20) NOT NULL DEFAULT '事假',"
|
||||||
|
" start_date DATE NOT NULL,"
|
||||||
|
" end_date DATE NOT NULL,"
|
||||||
|
" reason VARCHAR(500) DEFAULT '',"
|
||||||
|
" submitted_by UUID NOT NULL REFERENCES users(id),"
|
||||||
|
" created_at TIMESTAMPTZ DEFAULT now(),"
|
||||||
|
" updated_at TIMESTAMPTZ DEFAULT now()"
|
||||||
|
")"
|
||||||
|
))
|
||||||
|
await conn.run_sync(lambda c: c.exec_driver_sql(
|
||||||
|
"CREATE INDEX IF NOT EXISTS idx_leaves_manager_id ON leaves(manager_id)"
|
||||||
|
))
|
||||||
|
await conn.run_sync(lambda c: c.exec_driver_sql(
|
||||||
|
"CREATE INDEX IF NOT EXISTS idx_leaves_dates ON leaves(start_date, end_date)"
|
||||||
|
))
|
||||||
|
|
||||||
# Read notification_time from DB (or use default 17:30)
|
# Read notification_time from DB (or use default 17:30)
|
||||||
notification_hour, notification_minute = 17, 30
|
notification_hour, notification_minute = 17, 30
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ from app.models.key_visit import KeyVisit
|
|||||||
from app.models.daily_note import DailyNote
|
from app.models.daily_note import DailyNote
|
||||||
from app.models.ai_summary import AISummary
|
from app.models.ai_summary import AISummary
|
||||||
from app.models.system_config import SystemConfig
|
from app.models.system_config import SystemConfig
|
||||||
|
from app.models.leave import Leave
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"User",
|
"User",
|
||||||
@@ -22,4 +23,5 @@ __all__ = [
|
|||||||
"DailyNote",
|
"DailyNote",
|
||||||
"AISummary",
|
"AISummary",
|
||||||
"SystemConfig",
|
"SystemConfig",
|
||||||
|
"Leave",
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -0,0 +1,30 @@
|
|||||||
|
import uuid
|
||||||
|
from datetime import date, datetime
|
||||||
|
from sqlalchemy import String, Date, DateTime, ForeignKey, func
|
||||||
|
from sqlalchemy.dialects.postgresql import UUID
|
||||||
|
from sqlalchemy.orm import Mapped, mapped_column
|
||||||
|
from app.database import Base
|
||||||
|
|
||||||
|
|
||||||
|
class Leave(Base):
|
||||||
|
__tablename__ = "leaves"
|
||||||
|
|
||||||
|
id: Mapped[uuid.UUID] = mapped_column(
|
||||||
|
UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
|
||||||
|
)
|
||||||
|
manager_id: Mapped[uuid.UUID] = mapped_column(
|
||||||
|
UUID(as_uuid=True), ForeignKey("users.id"), index=True
|
||||||
|
)
|
||||||
|
leave_type: Mapped[str] = mapped_column(String(20), default="事假")
|
||||||
|
start_date: Mapped[date] = mapped_column(Date)
|
||||||
|
end_date: Mapped[date] = mapped_column(Date)
|
||||||
|
reason: Mapped[str] = mapped_column(String(500), default="")
|
||||||
|
submitted_by: Mapped[uuid.UUID] = mapped_column(
|
||||||
|
UUID(as_uuid=True), ForeignKey("users.id")
|
||||||
|
)
|
||||||
|
created_at: Mapped[datetime] = mapped_column(
|
||||||
|
DateTime(timezone=True), server_default=func.now()
|
||||||
|
)
|
||||||
|
updated_at: Mapped[datetime] = mapped_column(
|
||||||
|
DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
|
||||||
|
)
|
||||||
Reference in New Issue
Block a user