acazares
962f43fe62
- Updated SealService to support tenant and company filtering with pagination and enhanced CRUD methods. - Refactored TrailerService to include tenant and company support, added filtering capabilities, and improved CRUD methods. - Introduced TenantCRUDRoutes for Trailer and Transporter routes to streamline API endpoint creation and management. - Enhanced TransporterService with tenant and company filtering, pagination, and improved CRUD operations. - Added Customs Broker module with DTOs, models, services, and routes for managing customs broker data. - Implemented CRUD operations for Customs Broker, including personnel and VU management. - Improved data validation and descriptions in DTOs for better API documentation.
114 lines
3.4 KiB
Python
114 lines
3.4 KiB
Python
from sqlalchemy.orm import Session
|
|
|
|
from . import dto, models
|
|
|
|
|
|
class CustomsBrokerService:
|
|
@staticmethod
|
|
def get_by_broker_key(db: Session, broker_key: str):
|
|
return (
|
|
db.query(models.CustomsBroker)
|
|
.filter(models.CustomsBroker.broker_key == broker_key)
|
|
.first()
|
|
)
|
|
|
|
@staticmethod
|
|
def create_customs_broker(db: Session, broker_data: dto.CustomsBrokerDTO):
|
|
new_broker = models.CustomsBroker(**broker_data.dict())
|
|
db.add(new_broker)
|
|
db.commit()
|
|
db.refresh(new_broker)
|
|
return new_broker
|
|
|
|
@staticmethod
|
|
def delete_customs_broker(db: Session, broker_key: str):
|
|
broker = CustomsBrokerService.get_by_broker_key(db, broker_key)
|
|
if broker:
|
|
db.delete(broker)
|
|
db.commit()
|
|
return broker
|
|
|
|
|
|
class CustomsBrokerVUService:
|
|
@staticmethod
|
|
def get_by_broker_key(db: Session, broker_key: str):
|
|
return (
|
|
db.query(models.CustomsBrokerVU)
|
|
.filter(models.CustomsBrokerVU.broker_key == broker_key)
|
|
.first()
|
|
)
|
|
|
|
@staticmethod
|
|
def create_vu(db: Session, vu_data: dto.CustomsBrokerVUCreateDTO):
|
|
new_vu = models.CustomsBrokerVU(**vu_data.dict())
|
|
db.add(new_vu)
|
|
db.commit()
|
|
db.refresh(new_vu)
|
|
return new_vu
|
|
|
|
@staticmethod
|
|
def update_vu(db: Session, broker_key: str, vu_data: dto.CustomsBrokerVUCreateDTO):
|
|
vu = CustomsBrokerVUService.get_by_broker_key(db, broker_key)
|
|
if vu:
|
|
for key, value in vu_data.dict(exclude_unset=True).items():
|
|
setattr(vu, key, value)
|
|
db.commit()
|
|
db.refresh(vu)
|
|
return vu
|
|
|
|
@staticmethod
|
|
def delete_vu(db: Session, broker_key: str):
|
|
vu = CustomsBrokerVUService.get_by_broker_key(db, broker_key)
|
|
if vu:
|
|
db.delete(vu)
|
|
db.commit()
|
|
return vu
|
|
|
|
|
|
class CustomsBrokerPersonnelService:
|
|
@staticmethod
|
|
def get_by_broker_key_and_line(db: Session, broker_key: str, line: int):
|
|
return (
|
|
db.query(models.CustomsBrokerPersonnel)
|
|
.filter(
|
|
models.CustomsBrokerPersonnel.broker_key == broker_key,
|
|
models.CustomsBrokerPersonnel.line == line,
|
|
)
|
|
.first()
|
|
)
|
|
|
|
@staticmethod
|
|
def create_personnel(db: Session, personnel_data: dto.CustomsBrokerPersonnelDTO):
|
|
new_personnel = models.CustomsBrokerPersonnel(**personnel_data.dict())
|
|
db.add(new_personnel)
|
|
db.commit()
|
|
db.refresh(new_personnel)
|
|
return new_personnel
|
|
|
|
@staticmethod
|
|
def update_personnel(
|
|
db: Session,
|
|
broker_key: str,
|
|
line: int,
|
|
personnel_data: dto.CustomsBrokerPersonnelDTO,
|
|
):
|
|
personnel = CustomsBrokerPersonnelService.get_by_broker_key_and_line(
|
|
db, broker_key, line
|
|
)
|
|
if personnel:
|
|
for key, value in personnel_data.dict(exclude_unset=True).items():
|
|
setattr(personnel, key, value)
|
|
db.commit()
|
|
db.refresh(personnel)
|
|
return personnel
|
|
|
|
@staticmethod
|
|
def delete_personnel(db: Session, broker_key: str, line: int):
|
|
personnel = CustomsBrokerPersonnelService.get_by_broker_key_and_line(
|
|
db, broker_key, line
|
|
)
|
|
if personnel:
|
|
db.delete(personnel)
|
|
db.commit()
|
|
return personnel
|