acazares
b68c4316ff
- Rearranged imports in multiple files for consistency and clarity. - Updated logging middleware to exclude specific paths from logging. - Enhanced security module by cleaning up token handling and improving tenant validation. - Added tenant and company scoped mixins for better database model management. - Implemented generic CRUD routes for tenant-scoped resources. - Improved error handling and response management in API routes. - Cleaned up login and logout processes to ensure proper session management. - Introduced mechanisms to clear local storage and cookies on tenant change. - Enhanced company store to detect tenant changes and clear data accordingly. - Added new DTO mixins for currency and value affect flags.
30 lines
811 B
Python
30 lines
811 B
Python
from sqlalchemy.orm import Session
|
|
|
|
from . import dto, models
|
|
|
|
|
|
class VehicleService:
|
|
@staticmethod
|
|
def get_vehicle_by_key(db: Session, vehicle_key: str):
|
|
return (
|
|
db.query(models.Vehicle)
|
|
.filter(models.Vehicle.vehicle_key == vehicle_key)
|
|
.first()
|
|
)
|
|
|
|
@staticmethod
|
|
def create_vehicle(db: Session, vehicle_data: dto.VehicleCreateDTO):
|
|
new_vehicle = models.Vehicle(**vehicle_data.dict())
|
|
db.add(new_vehicle)
|
|
db.commit()
|
|
db.refresh(new_vehicle)
|
|
return new_vehicle
|
|
|
|
@staticmethod
|
|
def delete_vehicle(db: Session, vehicle_key: str):
|
|
vehicle = VehicleService.get_vehicle_by_key(db, vehicle_key)
|
|
if vehicle:
|
|
db.delete(vehicle)
|
|
db.commit()
|
|
return vehicle
|