Feature enhancement
This commit is contained in:
@@ -25,25 +25,25 @@ class CRUDDeparture:
|
||||
query = query.filter(Departure.status == status)
|
||||
|
||||
if date_from:
|
||||
query = query.filter(func.date(Departure.booked_out_dt) >= date_from)
|
||||
query = query.filter(func.date(Departure.created_dt) >= date_from)
|
||||
|
||||
if date_to:
|
||||
query = query.filter(func.date(Departure.booked_out_dt) <= date_to)
|
||||
query = query.filter(func.date(Departure.created_dt) <= date_to)
|
||||
|
||||
return query.order_by(desc(Departure.booked_out_dt)).offset(skip).limit(limit).all()
|
||||
return query.order_by(desc(Departure.created_dt)).offset(skip).limit(limit).all()
|
||||
|
||||
def get_departures_today(self, db: Session) -> List[Departure]:
|
||||
"""Get today's departures (booked out or departed)"""
|
||||
today = date.today()
|
||||
return db.query(Departure).filter(
|
||||
and_(
|
||||
func.date(Departure.booked_out_dt) == today,
|
||||
func.date(Departure.created_dt) == today,
|
||||
or_(
|
||||
Departure.status == DepartureStatus.BOOKED_OUT,
|
||||
Departure.status == DepartureStatus.DEPARTED
|
||||
)
|
||||
)
|
||||
).order_by(Departure.booked_out_dt).all()
|
||||
).order_by(Departure.created_dt).all()
|
||||
|
||||
def create(self, db: Session, obj_in: DepartureCreate, created_by: str) -> Departure:
|
||||
db_obj = Departure(
|
||||
@@ -82,7 +82,7 @@ class CRUDDeparture:
|
||||
db_obj.status = status
|
||||
|
||||
if status == DepartureStatus.DEPARTED and timestamp:
|
||||
db_obj.departure_dt = timestamp
|
||||
db_obj.departed_dt = timestamp
|
||||
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
|
||||
Reference in New Issue
Block a user