90 lines
3.1 KiB
Python
90 lines
3.1 KiB
Python
from sqlalchemy.orm import Session
|
|
from . import models, dto
|
|
|
|
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 |