Initial commit: KPI Management System Backend
Features: - FastAPI backend with JWT authentication - MySQL database with SQLAlchemy ORM - KPI workflow: draft → pending → approved → evaluation → completed - Ollama LLM API integration for AI features - Gitea API integration for version control - Complete API endpoints for KPI, dashboard, notifications Tables: KPI_D_* prefix naming convention 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
27
app/models/__init__.py
Normal file
27
app/models/__init__.py
Normal file
@@ -0,0 +1,27 @@
|
||||
"""
|
||||
SQLAlchemy Models
|
||||
"""
|
||||
from app.models.department import Department
|
||||
from app.models.employee import Employee
|
||||
from app.models.kpi_period import KPIPeriod
|
||||
from app.models.kpi_template import KPITemplate, KPIPreset, KPIPresetItem
|
||||
from app.models.kpi_sheet import KPISheet
|
||||
from app.models.kpi_item import KPIItem
|
||||
from app.models.kpi_review_log import KPIReviewLog
|
||||
from app.models.notification import Notification, NotificationPreference
|
||||
from app.models.dashboard_alert import DashboardAlert
|
||||
|
||||
__all__ = [
|
||||
"Department",
|
||||
"Employee",
|
||||
"KPIPeriod",
|
||||
"KPITemplate",
|
||||
"KPIPreset",
|
||||
"KPIPresetItem",
|
||||
"KPISheet",
|
||||
"KPIItem",
|
||||
"KPIReviewLog",
|
||||
"Notification",
|
||||
"NotificationPreference",
|
||||
"DashboardAlert",
|
||||
]
|
||||
41
app/models/dashboard_alert.py
Normal file
41
app/models/dashboard_alert.py
Normal file
@@ -0,0 +1,41 @@
|
||||
"""
|
||||
儀表板警示 Model
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, Integer, Boolean, DateTime, Text, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.employee import Employee
|
||||
from app.models.kpi_sheet import KPISheet
|
||||
|
||||
|
||||
class DashboardAlert(Base):
|
||||
"""儀表板警示"""
|
||||
|
||||
__tablename__ = "KPI_D_alerts"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
alert_type: Mapped[str] = mapped_column(String(50), nullable=False)
|
||||
# deadline_approaching, overdue, weight_invalid
|
||||
severity: Mapped[str] = mapped_column(String(20), default="warning") # info, warning, error
|
||||
title: Mapped[str] = mapped_column(String(200), nullable=False)
|
||||
description: Mapped[Optional[str]] = mapped_column(Text)
|
||||
related_sheet_id: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_sheets.id", ondelete="CASCADE"))
|
||||
related_employee_id: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_employees.id"))
|
||||
is_resolved: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||
resolved_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
|
||||
resolved_by: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_employees.id"))
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
related_sheet: Mapped[Optional["KPISheet"]] = relationship("KPISheet")
|
||||
related_employee: Mapped[Optional["Employee"]] = relationship("Employee", foreign_keys=[related_employee_id])
|
||||
resolver: Mapped[Optional["Employee"]] = relationship("Employee", foreign_keys=[resolved_by])
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<DashboardAlert id={self.id} type={self.alert_type} severity={self.severity}>"
|
||||
42
app/models/department.py
Normal file
42
app/models/department.py
Normal file
@@ -0,0 +1,42 @@
|
||||
"""
|
||||
部門 Model
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional, List, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, Integer, Boolean, DateTime, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.employee import Employee
|
||||
|
||||
|
||||
class Department(Base):
|
||||
"""部門"""
|
||||
|
||||
__tablename__ = "KPI_D_departments"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
code: Mapped[str] = mapped_column(String(20), unique=True, nullable=False)
|
||||
name: Mapped[str] = mapped_column(String(100), nullable=False)
|
||||
level: Mapped[str] = mapped_column(String(20), default="DEPT") # COMPANY, BU, DEPT, TEAM
|
||||
parent_id: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_departments.id"))
|
||||
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
parent: Mapped[Optional["Department"]] = relationship(
|
||||
"Department", remote_side=[id], back_populates="children"
|
||||
)
|
||||
children: Mapped[List["Department"]] = relationship(
|
||||
"Department", back_populates="parent"
|
||||
)
|
||||
employees: Mapped[List["Employee"]] = relationship(
|
||||
"Employee", back_populates="department"
|
||||
)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Department {self.code}: {self.name}>"
|
||||
68
app/models/employee.py
Normal file
68
app/models/employee.py
Normal file
@@ -0,0 +1,68 @@
|
||||
"""
|
||||
員工 Model
|
||||
"""
|
||||
from datetime import datetime, date
|
||||
from typing import Optional, List, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, Integer, Boolean, DateTime, Date, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.department import Department
|
||||
from app.models.kpi_sheet import KPISheet
|
||||
from app.models.notification import Notification
|
||||
|
||||
|
||||
class Employee(Base):
|
||||
"""員工"""
|
||||
|
||||
__tablename__ = "KPI_D_employees"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
employee_no: Mapped[str] = mapped_column(String(20), unique=True, nullable=False)
|
||||
name: Mapped[str] = mapped_column(String(100), nullable=False)
|
||||
email: Mapped[str] = mapped_column(String(200), unique=True, nullable=False)
|
||||
password_hash: Mapped[str] = mapped_column(String(200), nullable=False)
|
||||
department_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_departments.id"), nullable=False)
|
||||
manager_id: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_employees.id"))
|
||||
job_title: Mapped[Optional[str]] = mapped_column(String(100))
|
||||
role: Mapped[str] = mapped_column(String(20), default="employee") # employee, manager, admin, hr
|
||||
status: Mapped[str] = mapped_column(String(20), default="active") # active, inactive, resigned
|
||||
hire_date: Mapped[Optional[date]] = mapped_column(Date)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
department: Mapped["Department"] = relationship("Department", back_populates="employees")
|
||||
manager: Mapped[Optional["Employee"]] = relationship(
|
||||
"Employee", remote_side=[id], back_populates="subordinates"
|
||||
)
|
||||
subordinates: Mapped[List["Employee"]] = relationship(
|
||||
"Employee", back_populates="manager"
|
||||
)
|
||||
kpi_sheets: Mapped[List["KPISheet"]] = relationship(
|
||||
"KPISheet", back_populates="employee", foreign_keys="KPISheet.employee_id"
|
||||
)
|
||||
notifications: Mapped[List["Notification"]] = relationship(
|
||||
"Notification", back_populates="recipient"
|
||||
)
|
||||
|
||||
@property
|
||||
def is_manager(self) -> bool:
|
||||
"""是否為主管"""
|
||||
return self.role in ("manager", "admin")
|
||||
|
||||
@property
|
||||
def is_admin(self) -> bool:
|
||||
"""是否為管理員"""
|
||||
return self.role == "admin"
|
||||
|
||||
@property
|
||||
def is_hr(self) -> bool:
|
||||
"""是否為人資"""
|
||||
return self.role == "hr"
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Employee {self.employee_no}: {self.name}>"
|
||||
55
app/models/kpi_item.py
Normal file
55
app/models/kpi_item.py
Normal file
@@ -0,0 +1,55 @@
|
||||
"""
|
||||
KPI 項目 Model
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, Integer, DateTime, Text, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.kpi_sheet import KPISheet
|
||||
from app.models.kpi_template import KPITemplate
|
||||
|
||||
|
||||
class KPIItem(Base):
|
||||
"""KPI 項目"""
|
||||
|
||||
__tablename__ = "KPI_D_items"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
sheet_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_sheets.id", ondelete="CASCADE"), nullable=False)
|
||||
template_id: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_templates.id"))
|
||||
sort_order: Mapped[int] = mapped_column(Integer, default=0)
|
||||
|
||||
# 項目資訊
|
||||
name: Mapped[str] = mapped_column(String(200), nullable=False)
|
||||
category: Mapped[str] = mapped_column(String(50), nullable=False) # financial, customer, internal, learning
|
||||
weight: Mapped[int] = mapped_column(Integer, nullable=False) # 權重百分比 (1-100)
|
||||
|
||||
# 等級標準
|
||||
level0_criteria: Mapped[Optional[str]] = mapped_column(Text)
|
||||
level1_criteria: Mapped[Optional[str]] = mapped_column(Text)
|
||||
level2_criteria: Mapped[Optional[str]] = mapped_column(Text)
|
||||
level3_criteria: Mapped[Optional[str]] = mapped_column(Text)
|
||||
level4_criteria: Mapped[Optional[str]] = mapped_column(Text)
|
||||
|
||||
# 自評
|
||||
self_eval_level: Mapped[Optional[int]] = mapped_column(Integer) # 0-4
|
||||
self_eval_note: Mapped[Optional[str]] = mapped_column(Text)
|
||||
|
||||
# 主管評核
|
||||
final_level: Mapped[Optional[int]] = mapped_column(Integer) # 0-4
|
||||
final_note: Mapped[Optional[str]] = mapped_column(Text)
|
||||
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
sheet: Mapped["KPISheet"] = relationship("KPISheet", back_populates="items")
|
||||
template: Mapped[Optional["KPITemplate"]] = relationship("KPITemplate", back_populates="kpi_items")
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<KPIItem id={self.id} name={self.name} weight={self.weight}>"
|
||||
63
app/models/kpi_period.py
Normal file
63
app/models/kpi_period.py
Normal file
@@ -0,0 +1,63 @@
|
||||
"""
|
||||
KPI 期間 Model
|
||||
"""
|
||||
from datetime import datetime, date
|
||||
from typing import Optional, List, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, DateTime, Date
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.kpi_sheet import KPISheet
|
||||
|
||||
|
||||
class KPIPeriod(Base):
|
||||
"""KPI 期間"""
|
||||
|
||||
__tablename__ = "KPI_D_periods"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
code: Mapped[str] = mapped_column(String(20), unique=True, nullable=False) # 例: 2024H1
|
||||
name: Mapped[str] = mapped_column(String(100), nullable=False)
|
||||
start_date: Mapped[date] = mapped_column(Date, nullable=False)
|
||||
end_date: Mapped[date] = mapped_column(Date, nullable=False)
|
||||
setting_start: Mapped[date] = mapped_column(Date, nullable=False)
|
||||
setting_end: Mapped[date] = mapped_column(Date, nullable=False)
|
||||
self_eval_start: Mapped[Optional[date]] = mapped_column(Date)
|
||||
self_eval_end: Mapped[Optional[date]] = mapped_column(Date)
|
||||
manager_eval_start: Mapped[Optional[date]] = mapped_column(Date)
|
||||
manager_eval_end: Mapped[Optional[date]] = mapped_column(Date)
|
||||
status: Mapped[str] = mapped_column(String(20), default="draft")
|
||||
# draft, setting, approved, self_eval, manager_eval, completed
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
kpi_sheets: Mapped[List["KPISheet"]] = relationship("KPISheet", back_populates="period")
|
||||
|
||||
@property
|
||||
def is_setting_period(self) -> bool:
|
||||
"""是否在設定期間"""
|
||||
today = date.today()
|
||||
return self.setting_start <= today <= self.setting_end
|
||||
|
||||
@property
|
||||
def is_self_eval_period(self) -> bool:
|
||||
"""是否在自評期間"""
|
||||
if not self.self_eval_start or not self.self_eval_end:
|
||||
return False
|
||||
today = date.today()
|
||||
return self.self_eval_start <= today <= self.self_eval_end
|
||||
|
||||
@property
|
||||
def is_manager_eval_period(self) -> bool:
|
||||
"""是否在主管評核期間"""
|
||||
if not self.manager_eval_start or not self.manager_eval_end:
|
||||
return False
|
||||
today = date.today()
|
||||
return self.manager_eval_start <= today <= self.manager_eval_end
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<KPIPeriod {self.code}: {self.name}>"
|
||||
36
app/models/kpi_review_log.py
Normal file
36
app/models/kpi_review_log.py
Normal file
@@ -0,0 +1,36 @@
|
||||
"""
|
||||
KPI 審核紀錄 Model
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, Integer, DateTime, Text, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.kpi_sheet import KPISheet
|
||||
from app.models.employee import Employee
|
||||
|
||||
|
||||
class KPIReviewLog(Base):
|
||||
"""KPI 審核紀錄"""
|
||||
|
||||
__tablename__ = "KPI_D_review_logs"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
sheet_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_sheets.id", ondelete="CASCADE"), nullable=False)
|
||||
action: Mapped[str] = mapped_column(String(50), nullable=False) # submit, approve, reject, self_eval, manager_eval
|
||||
actor_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_employees.id"), nullable=False)
|
||||
from_status: Mapped[Optional[str]] = mapped_column(String(20))
|
||||
to_status: Mapped[Optional[str]] = mapped_column(String(20))
|
||||
comment: Mapped[Optional[str]] = mapped_column(Text)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
sheet: Mapped["KPISheet"] = relationship("KPISheet", back_populates="review_logs")
|
||||
actor: Mapped["Employee"] = relationship("Employee")
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<KPIReviewLog id={self.id} action={self.action} sheet={self.sheet_id}>"
|
||||
103
app/models/kpi_sheet.py
Normal file
103
app/models/kpi_sheet.py
Normal file
@@ -0,0 +1,103 @@
|
||||
"""
|
||||
KPI 表單 Model
|
||||
"""
|
||||
from datetime import datetime
|
||||
from decimal import Decimal
|
||||
from typing import Optional, List, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, Integer, DateTime, Text, Numeric, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.employee import Employee
|
||||
from app.models.kpi_period import KPIPeriod
|
||||
from app.models.department import Department
|
||||
from app.models.kpi_item import KPIItem
|
||||
from app.models.kpi_review_log import KPIReviewLog
|
||||
|
||||
|
||||
# KPI 表單狀態
|
||||
class KPISheetStatus:
|
||||
DRAFT = "draft" # 草稿
|
||||
PENDING = "pending" # 待審核
|
||||
APPROVED = "approved" # 已核准
|
||||
SELF_EVAL = "self_eval" # 自評中
|
||||
MANAGER_EVAL = "manager_eval" # 主管評核中
|
||||
COMPLETED = "completed" # 已完成
|
||||
SETTLED = "settled" # 已結算
|
||||
|
||||
|
||||
# 狀態轉換規則
|
||||
VALID_STATUS_TRANSITIONS = {
|
||||
KPISheetStatus.DRAFT: [KPISheetStatus.PENDING],
|
||||
KPISheetStatus.PENDING: [KPISheetStatus.APPROVED, KPISheetStatus.DRAFT],
|
||||
KPISheetStatus.APPROVED: [KPISheetStatus.SELF_EVAL],
|
||||
KPISheetStatus.SELF_EVAL: [KPISheetStatus.MANAGER_EVAL],
|
||||
KPISheetStatus.MANAGER_EVAL: [KPISheetStatus.COMPLETED],
|
||||
KPISheetStatus.COMPLETED: [KPISheetStatus.SETTLED],
|
||||
KPISheetStatus.SETTLED: [],
|
||||
}
|
||||
|
||||
|
||||
class KPISheet(Base):
|
||||
"""KPI 表單"""
|
||||
|
||||
__tablename__ = "KPI_D_sheets"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
employee_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_employees.id"), nullable=False)
|
||||
period_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_periods.id"), nullable=False)
|
||||
department_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_departments.id"), nullable=False)
|
||||
status: Mapped[str] = mapped_column(String(20), default=KPISheetStatus.DRAFT)
|
||||
|
||||
# 提交資訊
|
||||
submitted_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
|
||||
|
||||
# 審核資訊
|
||||
approved_by: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_employees.id"))
|
||||
approved_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
|
||||
approve_comment: Mapped[Optional[str]] = mapped_column(Text)
|
||||
|
||||
# 退回資訊
|
||||
rejected_by: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_employees.id"))
|
||||
rejected_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
|
||||
reject_reason: Mapped[Optional[str]] = mapped_column(Text)
|
||||
|
||||
# 自評資訊
|
||||
self_eval_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
|
||||
|
||||
# 主管評核資訊
|
||||
manager_eval_by: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_employees.id"))
|
||||
manager_eval_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
|
||||
manager_eval_comment: Mapped[Optional[str]] = mapped_column(Text)
|
||||
|
||||
# 分數
|
||||
total_score: Mapped[Optional[Decimal]] = mapped_column(Numeric(5, 4))
|
||||
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
employee: Mapped["Employee"] = relationship(
|
||||
"Employee", back_populates="kpi_sheets", foreign_keys=[employee_id]
|
||||
)
|
||||
period: Mapped["KPIPeriod"] = relationship("KPIPeriod", back_populates="kpi_sheets")
|
||||
department: Mapped["Department"] = relationship("Department")
|
||||
items: Mapped[List["KPIItem"]] = relationship(
|
||||
"KPIItem", back_populates="sheet", cascade="all, delete-orphan"
|
||||
)
|
||||
review_logs: Mapped[List["KPIReviewLog"]] = relationship(
|
||||
"KPIReviewLog", back_populates="sheet", cascade="all, delete-orphan"
|
||||
)
|
||||
approver: Mapped[Optional["Employee"]] = relationship("Employee", foreign_keys=[approved_by])
|
||||
rejecter: Mapped[Optional["Employee"]] = relationship("Employee", foreign_keys=[rejected_by])
|
||||
manager_evaluator: Mapped[Optional["Employee"]] = relationship("Employee", foreign_keys=[manager_eval_by])
|
||||
|
||||
def can_transition_to(self, new_status: str) -> bool:
|
||||
"""檢查是否可以轉換到指定狀態"""
|
||||
return new_status in VALID_STATUS_TRANSITIONS.get(self.status, [])
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<KPISheet id={self.id} employee={self.employee_id} period={self.period_id} status={self.status}>"
|
||||
88
app/models/kpi_template.py
Normal file
88
app/models/kpi_template.py
Normal file
@@ -0,0 +1,88 @@
|
||||
"""
|
||||
KPI 範本 Model
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional, List, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, Integer, Boolean, DateTime, Text, ForeignKey, JSON
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.kpi_item import KPIItem
|
||||
|
||||
|
||||
class KPITemplate(Base):
|
||||
"""KPI 範本"""
|
||||
|
||||
__tablename__ = "KPI_D_templates"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
code: Mapped[str] = mapped_column(String(50), unique=True, nullable=False)
|
||||
name: Mapped[str] = mapped_column(String(200), nullable=False)
|
||||
category: Mapped[str] = mapped_column(String(50), nullable=False) # financial, customer, internal, learning
|
||||
description: Mapped[Optional[str]] = mapped_column(Text)
|
||||
default_weight: Mapped[int] = mapped_column(Integer, default=20)
|
||||
level0_desc: Mapped[str] = mapped_column(Text, nullable=False)
|
||||
level1_desc: Mapped[str] = mapped_column(Text, nullable=False)
|
||||
level2_desc: Mapped[str] = mapped_column(Text, nullable=False)
|
||||
level3_desc: Mapped[str] = mapped_column(Text, nullable=False)
|
||||
level4_desc: Mapped[str] = mapped_column(Text, nullable=False)
|
||||
applicable_roles: Mapped[Optional[str]] = mapped_column(JSON) # JSON array for MySQL
|
||||
applicable_depts: Mapped[Optional[str]] = mapped_column(JSON) # JSON array for MySQL
|
||||
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
kpi_items: Mapped[List["KPIItem"]] = relationship("KPIItem", back_populates="template")
|
||||
preset_items: Mapped[List["KPIPresetItem"]] = relationship("KPIPresetItem", back_populates="template")
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<KPITemplate {self.code}: {self.name}>"
|
||||
|
||||
|
||||
class KPIPreset(Base):
|
||||
"""KPI 預設組合"""
|
||||
|
||||
__tablename__ = "KPI_D_presets"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
code: Mapped[str] = mapped_column(String(50), unique=True, nullable=False)
|
||||
name: Mapped[str] = mapped_column(String(200), nullable=False)
|
||||
description: Mapped[Optional[str]] = mapped_column(Text)
|
||||
applicable_roles: Mapped[Optional[str]] = mapped_column(JSON) # JSON array for MySQL
|
||||
applicable_depts: Mapped[Optional[str]] = mapped_column(JSON) # JSON array for MySQL
|
||||
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
items: Mapped[List["KPIPresetItem"]] = relationship(
|
||||
"KPIPresetItem", back_populates="preset", cascade="all, delete-orphan"
|
||||
)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<KPIPreset {self.code}: {self.name}>"
|
||||
|
||||
|
||||
class KPIPresetItem(Base):
|
||||
"""KPI 預設項目"""
|
||||
|
||||
__tablename__ = "KPI_D_preset_items"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
preset_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_presets.id", ondelete="CASCADE"), nullable=False)
|
||||
template_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_templates.id"), nullable=False)
|
||||
default_weight: Mapped[int] = mapped_column(Integer, nullable=False)
|
||||
is_mandatory: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||
sort_order: Mapped[int] = mapped_column(Integer, default=0)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
preset: Mapped["KPIPreset"] = relationship("KPIPreset", back_populates="items")
|
||||
template: Mapped["KPITemplate"] = relationship("KPITemplate", back_populates="preset_items")
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<KPIPresetItem preset={self.preset_id} template={self.template_id}>"
|
||||
58
app/models/notification.py
Normal file
58
app/models/notification.py
Normal file
@@ -0,0 +1,58 @@
|
||||
"""
|
||||
通知 Model
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional, TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import String, Integer, Boolean, DateTime, Text, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.core.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.employee import Employee
|
||||
from app.models.kpi_sheet import KPISheet
|
||||
|
||||
|
||||
class Notification(Base):
|
||||
"""通知"""
|
||||
|
||||
__tablename__ = "KPI_D_notifications"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
recipient_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_employees.id"), nullable=False)
|
||||
type: Mapped[str] = mapped_column(String(50), nullable=False)
|
||||
# kpi_submitted, kpi_approved, kpi_rejected, eval_reminder, etc.
|
||||
title: Mapped[str] = mapped_column(String(200), nullable=False)
|
||||
content: Mapped[Optional[str]] = mapped_column(Text)
|
||||
related_sheet_id: Mapped[Optional[int]] = mapped_column(ForeignKey("KPI_D_sheets.id", ondelete="SET NULL"))
|
||||
is_read: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||
read_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
recipient: Mapped["Employee"] = relationship("Employee", back_populates="notifications")
|
||||
related_sheet: Mapped[Optional["KPISheet"]] = relationship("KPISheet")
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Notification id={self.id} type={self.type} recipient={self.recipient_id}>"
|
||||
|
||||
|
||||
class NotificationPreference(Base):
|
||||
"""通知偏好"""
|
||||
|
||||
__tablename__ = "KPI_D_notification_preferences"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
employee_id: Mapped[int] = mapped_column(ForeignKey("KPI_D_employees.id"), unique=True, nullable=False)
|
||||
email_enabled: Mapped[bool] = mapped_column(Boolean, default=True)
|
||||
in_app_enabled: Mapped[bool] = mapped_column(Boolean, default=True)
|
||||
reminder_days_before: Mapped[int] = mapped_column(Integer, default=3)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||||
|
||||
# Relationships
|
||||
employee: Mapped["Employee"] = relationship("Employee")
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<NotificationPreference employee={self.employee_id}>"
|
||||
Reference in New Issue
Block a user