schemas added & lang models added

Schemas/company/employee.py

@@ -0,0 +1,153 @@
+from sqlalchemy import (
+    String,
+    ForeignKey,
+    Index,
+    Numeric,
+)
+from sqlalchemy.orm import mapped_column, Mapped
+
+from databases.language_models.company.employee import (
+    StaffLanguageModel,
+    EmployeesLanguageModel,
+    EmployeeHistoryLanguageModel,
+    EmployeesSalariesLanguageModel,
+)
+from databases.sql_models.core_mixin import CrudCollection
+
+from api_validations.validations_request import InsertCompanyEmployees
+
+
+class Staff(CrudCollection):
+
+    __tablename__ = "staff"
+    __exclude__fields__ = []
+    __language_model__ = StaffLanguageModel
+
+    staff_description: Mapped[str] = mapped_column(
+        String, server_default="", comment="Staff Description"
+    )
+    staff_name: Mapped[str] = mapped_column(
+        String, nullable=False, comment="Staff Name"
+    )
+    staff_code: Mapped[str] = mapped_column(
+        String, nullable=False, comment="Staff Code"
+    )
+
+    duties_id: Mapped[int] = mapped_column(ForeignKey("duties.id"), nullable=False)
+    duties_uu_id: Mapped[str] = mapped_column(
+        String, nullable=False, comment="Duty UUID"
+    )
+
+    # people: Mapped["People"] = relationship(
+    #     "People", back_populates="employees", foreign_keys=[people_id], uselist=True
+    # )
+    # duty: Mapped["CompanyDuty"] = relationship(
+    #     "CompanyDuty", back_populates="employees", foreign_keys=[duty_id]
+    # )
+
+    @classmethod
+    def create_action(cls, data: InsertCompanyEmployees):
+        from databases import Duties
+
+        data_dict = data.model_dump()
+        if duty := Duties.find_one(uu_id=data.duty_uu_id):
+            data_dict["duty_id"] = duty.id
+        # if person := People.find_one(uu_id=data.person_uu_id):
+        #     data_dict["people_id"] = person.id
+        if data.start_date:
+            data_dict["expiry_starts"] = data.start_date
+        if data.stop_date:
+            data_dict["expiry_ends"] = data.stop_date
+        # del data_dict["duty_uu_id"], data_dict["person_uu_id"]
+        del data_dict["start_date"], data_dict["stop_date"], data_dict["duty_uu_id"]
+        return cls.find_or_create(**data_dict)
+
+    __table_args__ = ({"comment": "Staff Information"},)
+
+
+class Employees(CrudCollection):
+
+    __tablename__ = "employees"
+    __exclude__fields__ = []
+    __language_model__ = EmployeesLanguageModel
+
+    staff_id: Mapped[int] = mapped_column(ForeignKey("staff.id"))
+    staff_uu_id: Mapped[str] = mapped_column(
+        String, nullable=False, comment="Staff UUID"
+    )
+    people_id: Mapped[int] = mapped_column(ForeignKey("people.id"), nullable=True)
+    people_uu_id: Mapped[str] = mapped_column(
+        String, nullable=True, comment="People UUID"
+    )
+
+    __table_args__ = (
+        Index("employees_ndx_00", people_id, staff_id, unique=True),
+        {"comment": "Employee Person Information"},
+    )
+
+
+class EmployeeHistory(CrudCollection):
+
+    __tablename__ = "employee_history"
+    __exclude__fields__ = []
+    __language_model__ = EmployeeHistoryLanguageModel
+
+    staff_id: Mapped[int] = mapped_column(
+        ForeignKey("staff.id"), nullable=False, comment="Staff ID"
+    )
+    staff_uu_id: Mapped[str] = mapped_column(
+        String, nullable=False, comment="Staff UUID"
+    )
+    people_id: Mapped[int] = mapped_column(
+        ForeignKey("people.id"), nullable=False, comment="People ID"
+    )
+    people_uu_id: Mapped[str] = mapped_column(
+        String, nullable=False, comment="People UUID"
+    )
+
+    __table_args__ = (
+        Index("_employee_history_ndx_00", people_id, staff_id),
+        {"comment": "Employee History Information"},
+    )
+
+
+class EmployeesSalaries(CrudCollection):
+
+    __tablename__ = "employee_salaries"
+    __exclude__fields__ = []
+    __language_model__ = EmployeesSalariesLanguageModel
+
+    gross_salary: Mapped[float] = mapped_column(
+        Numeric(20, 6), nullable=False, comment="Gross Salary"
+    )
+    net_salary: Mapped[float] = mapped_column(
+        Numeric(20, 6), nullable=False, comment="Net Salary"
+    )
+
+    people_id: Mapped[int] = mapped_column(ForeignKey("people.id"), nullable=False)
+    people_uu_id: Mapped[str] = mapped_column(
+        String, nullable=False, comment="People UUID"
+    )
+
+    # people: Mapped["People"] = relationship(
+    #     "People", back_populates="employee_salaries", foreign_keys=[people_id]
+    # )
+
+    __table_args__ = (
+        Index("_employee_salaries_ndx_00", people_id, "expiry_starts"),
+        {"comment": "Employee Salaries Information"},
+    )
+
+
+# class Events2Employees(CrudCollection):
+#
+#     __tablename__ = "events2employees"
+#     __exclude__fields__ = []
+#
+#     event_id = mapped_column(ForeignKey("events.id"), nullable=False)
+#     employees_id = mapped_column(ForeignKey("employees.id"), nullable=False)
+#
+#     __table_args__ = (
+#         Index("_events2employees_ndx_00", event_id, employees_id),
+#         {"comment": "Events2Employees Information"},
+#     )