feat: implemented deleting tracks, drivers, vehicles with proper data correction

This commit is contained in:
2025-01-16 01:30:32 +01:00
parent 1df2b155f3
commit a88c9e5ff7
9 changed files with 337 additions and 152 deletions

View File

@@ -1,7 +1,7 @@
from sqlalchemy.orm import Session
from errors.NotFoundException import NotFoundError
from modules.geoObjects import Driver
from modules.geoObjects import Driver, Track
class DriverHandler:
@@ -19,30 +19,32 @@ class DriverHandler:
raise ValueError("name is empty")
driver = Driver(name=name)
self.dbSession.add(driver)
self.dbSession.commit()
return driver
except Exception as e:
raise RuntimeError(f"failed to create driver with errors{e}")
# handles getting a driver by its id from the database
def getDriver(self, driverID: int) -> Driver:
try:
driver = self.dbSession.query(Driver).filter_by(id=driverID).first()
driver = self.dbSession.query(
Driver).filter_by(id=driverID).first()
if driver:
return driver
else:
raise NotFoundError(f"Driver with ID {driverID} not found", errors=[])
raise NotFoundError(
f"Driver with ID {driverID} not found", errors=[])
except Exception as e:
raise RuntimeError(f"failed to get driver with errors{e}")
# handles getting all drivers from the database
def getDrivers(self):
try:
drivers = self.dbSession.query(Driver).all()
driverList = [
{
"id": driver.id,
@@ -53,6 +55,19 @@ class DriverHandler:
]
return driverList
except Exception as e:
raise RuntimeError(f"failed to get drivers with errors{e}")
raise RuntimeError(f"failed to get drivers with errors{e}")
def deleteDriver(self, id):
try:
updated_rows = (
self.dbSession.query(Track).filter(
Track.driver_id == id).update({Track.driver_id: 1})
)
self.dbSession.query(Driver).filter(Driver.id == id).delete()
self.dbSession.commit()
except Exception as e:
raise RuntimeError(f"failed to delete driver due to error: {e}")