coati-payroll 0.0.2__py3-none-any.whl

coati_payroll/nomina_engine/repositories/employee_repository.py

@@ -0,0 +1,34 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Repository for Employee operations."""
+
+from __future__ import annotations
+
+from typing import Optional
+
+from coati_payroll.model import Empleado
+from .base_repository import BaseRepository
+
+
+class EmployeeRepository(BaseRepository[Empleado]):
+    """Repository for Employee operations."""
+
+    def get_by_id(self, empleado_id: str) -> Optional[Empleado]:
+        """Get employee by ID."""
+        return self.session.get(Empleado, empleado_id)
+
+    def save(self, empleado: Empleado) -> Empleado:
+        """Save employee."""
+        self.session.add(empleado)
+        return empleado

coati_payroll/nomina_engine/repositories/exchange_rate_repository.py

@@ -0,0 +1,58 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Repository for Exchange Rate operations."""
+
+from __future__ import annotations
+
+from datetime import date
+from decimal import Decimal
+from typing import Optional
+
+from coati_payroll.model import TipoCambio
+from .base_repository import BaseRepository
+
+
+class ExchangeRateRepository(BaseRepository[TipoCambio]):
+    """Repository for TipoCambio operations."""
+
+    def get_by_id(self, tipo_cambio_id: str) -> Optional[TipoCambio]:
+        """Get exchange rate by ID."""
+        return self.session.get(TipoCambio, tipo_cambio_id)
+
+    def get_rate(self, moneda_origen_id: str, moneda_destino_id: str, fecha: date) -> Optional[Decimal]:
+        """Get exchange rate for currency pair on or before given date."""
+        from sqlalchemy import select
+
+        tipo_cambio = (
+            self.session.execute(
+                select(TipoCambio)
+                .filter(
+                    TipoCambio.moneda_origen_id == moneda_origen_id,
+                    TipoCambio.moneda_destino_id == moneda_destino_id,
+                    TipoCambio.fecha <= fecha,
+                )
+                .order_by(TipoCambio.fecha.desc())
+            )
+            .unique()
+            .scalar_one_or_none()
+        )
+
+        if tipo_cambio:
+            return Decimal(str(tipo_cambio.tasa))
+        return None
+
+    def save(self, tipo_cambio: TipoCambio) -> TipoCambio:
+        """Save exchange rate."""
+        self.session.add(tipo_cambio)
+        return tipo_cambio

coati_payroll/nomina_engine/repositories/novelty_repository.py

@@ -0,0 +1,54 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Repository for Novelty operations."""
+
+from __future__ import annotations
+
+from datetime import date
+from typing import Optional
+
+from coati_payroll.model import NominaNovedad
+from .base_repository import BaseRepository
+
+
+class NoveltyRepository(BaseRepository[NominaNovedad]):
+    """Repository for NominaNovedad operations."""
+
+    def get_by_id(self, novelty_id: str) -> Optional[NominaNovedad]:
+        """Get novelty by ID."""
+        return self.session.get(NominaNovedad, novelty_id)
+
+    def get_by_employee_and_period(
+        self, empleado_id: str, periodo_inicio: date, periodo_fin: date
+    ) -> list[NominaNovedad]:
+        """Get novelties for employee within period."""
+        from sqlalchemy import select
+
+        return list(
+            self.session.execute(
+                select(NominaNovedad).filter(
+                    NominaNovedad.empleado_id == empleado_id,
+                    NominaNovedad.fecha_novedad >= periodo_inicio,
+                    NominaNovedad.fecha_novedad <= periodo_fin,
+                )
+            )
+            .unique()
+            .scalars()
+            .all()
+        )
+
+    def save(self, novelty: NominaNovedad) -> NominaNovedad:
+        """Save novelty."""
+        self.session.add(novelty)
+        return novelty

coati_payroll/nomina_engine/repositories/planilla_repository.py

@@ -0,0 +1,52 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Repository for Planilla operations."""
+
+from __future__ import annotations
+
+from typing import Optional
+from sqlalchemy.orm import joinedload
+
+from coati_payroll.model import Planilla, PlanillaEmpleado, PlanillaIngreso, PlanillaDeduccion, PlanillaPrestacion
+from .base_repository import BaseRepository
+
+
+class PlanillaRepository(BaseRepository[Planilla]):
+    """Repository for Planilla operations."""
+
+    def get_by_id(self, planilla_id: str) -> Optional[Planilla]:
+        """Get planilla with all necessary relationships loaded."""
+        from sqlalchemy import select
+
+        return (
+            self.session.execute(
+                select(Planilla)
+                .options(
+                    joinedload(Planilla.tipo_planilla),
+                    joinedload(Planilla.moneda),
+                    joinedload(Planilla.planilla_empleados).joinedload(PlanillaEmpleado.empleado),
+                    joinedload(Planilla.planilla_percepciones).joinedload(PlanillaIngreso.percepcion),
+                    joinedload(Planilla.planilla_deducciones).joinedload(PlanillaDeduccion.deduccion),
+                    joinedload(Planilla.planilla_prestaciones).joinedload(PlanillaPrestacion.prestacion),
+                )
+                .filter(Planilla.id == planilla_id)
+            )
+            .unique()
+            .scalar_one_or_none()
+        )
+
+    def save(self, planilla: Planilla) -> Planilla:
+        """Save planilla."""
+        self.session.add(planilla)
+        return planilla

coati_payroll/nomina_engine/results/__init__.py

@@ -0,0 +1,24 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Result DTOs for payroll processing."""
+
+from .validation_result import ValidationResult
+from .error_result import ErrorResult
+from .payroll_result import PayrollResult
+
+__all__ = [
+    "ValidationResult",
+    "ErrorResult",
+    "PayrollResult",
+]

coati_payroll/nomina_engine/results/error_result.py

@@ -0,0 +1,28 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Error result DTO."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class ErrorResult:
+    """Represents an error that occurred during processing."""
+
+    message: str
+    code: Optional[str] = None
+    details: Optional[dict] = None

coati_payroll/nomina_engine/results/payroll_result.py

@@ -0,0 +1,53 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Payroll execution result DTO."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Optional, TYPE_CHECKING
+
+from coati_payroll.model import Nomina
+
+if TYPE_CHECKING:
+    from ..domain.employee_calculation import EmpleadoCalculo
+    from ..domain.payroll_context import PayrollContext
+
+
+@dataclass
+class PayrollResult:
+    """Result of a payroll execution."""
+
+    context: "PayrollContext"
+    nomina: Optional[Nomina] = None
+    employee_calculations: list["EmpleadoCalculo"] = field(default_factory=list)
+    errors: list[str] = field(default_factory=list)
+    warnings: list[str] = field(default_factory=list)
+
+    def add_error(self, error: str) -> None:
+        """Add an error message."""
+        self.errors.append(error)
+
+    def add_warning(self, warning: str) -> None:
+        """Add a warning message."""
+        self.warnings.append(warning)
+
+    def add_errors(self, errors: list[str]) -> None:
+        """Add multiple error messages."""
+        self.errors.extend(errors)
+
+    def set_success(self, nomina: Nomina, calculations: list["EmpleadoCalculo"]) -> None:
+        """Set the result as successful."""
+        self.nomina = nomina
+        self.employee_calculations = calculations

coati_payroll/nomina_engine/results/validation_result.py

@@ -0,0 +1,39 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Validation result DTO."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+
+
+@dataclass
+class ValidationResult:
+    """Result of a validation operation."""
+
+    errors: list[str] = field(default_factory=list)
+    warnings: list[str] = field(default_factory=list)
+
+    def add_error(self, error: str) -> None:
+        """Add an error message."""
+        self.errors.append(error)
+
+    def add_warning(self, warning: str) -> None:
+        """Add a warning message."""
+        self.warnings.append(warning)
+
+    @property
+    def is_valid(self) -> bool:
+        """Check if validation passed (no errors)."""
+        return len(self.errors) == 0

coati_payroll/nomina_engine/services/__init__.py

@@ -0,0 +1,22 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Business services for payroll processing."""
+
+from .employee_processing_service import EmployeeProcessingService
+from .payroll_execution_service import PayrollExecutionService
+
+__all__ = [
+    "EmployeeProcessingService",
+    "PayrollExecutionService",
+]