70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
"""
|
|
Service layer for PedimentoPayments CRUD operations
|
|
"""
|
|
|
|
from typing import Optional
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
from ..dtos.pedimento_payments import PedimentoPaymentsCreate, PedimentoPaymentsUpdate
|
|
from ..models.pedimento_payments import PedimentoPayments
|
|
|
|
|
|
class PedimentoPaymentsService:
|
|
"""Service class for PedimentoPayments business logic"""
|
|
|
|
@staticmethod
|
|
def get_by_pedimento_id(
|
|
db: Session, pedimento_id: int, tenant_id: int, company_id: int = None
|
|
) -> Optional[PedimentoPayments]:
|
|
"""Get payments by pedimento ID"""
|
|
return (
|
|
db.query(PedimentoPayments)
|
|
.filter(
|
|
PedimentoPayments.pedimento_id == pedimento_id,
|
|
PedimentoPayments.tenant_id == tenant_id,
|
|
)
|
|
.first()
|
|
)
|
|
|
|
@staticmethod
|
|
def create(db: Session, data: PedimentoPaymentsCreate) -> PedimentoPayments:
|
|
"""Create new payments"""
|
|
payments = PedimentoPayments(**data.model_dump())
|
|
db.add(payments)
|
|
db.commit()
|
|
db.refresh(payments)
|
|
return payments
|
|
|
|
@staticmethod
|
|
def update(
|
|
db: Session, pedimento_id: int, tenant_id: int, data: PedimentoPaymentsUpdate
|
|
) -> Optional[PedimentoPayments]:
|
|
"""Update payments"""
|
|
payments = PedimentoPaymentsService.get_by_pedimento_id(
|
|
db, pedimento_id, tenant_id
|
|
)
|
|
if not payments:
|
|
return None
|
|
|
|
update_data = data.model_dump(exclude_unset=True)
|
|
for field, value in update_data.items():
|
|
setattr(payments, field, value)
|
|
|
|
db.commit()
|
|
db.refresh(payments)
|
|
return payments
|
|
|
|
@staticmethod
|
|
def delete(db: Session, pedimento_id: int, tenant_id: int) -> bool:
|
|
"""Delete payments"""
|
|
payments = PedimentoPaymentsService.get_by_pedimento_id(
|
|
db, pedimento_id, tenant_id
|
|
)
|
|
if not payments:
|
|
return False
|
|
|
|
db.delete(payments)
|
|
db.commit()
|
|
return True
|