Merge branch 'restaurants-route' into 'collab_front'

Restaurants route

See merge request !30

backend/db/crud.py

@@ -275,3 +275,54 @@ def delete_opening_hours(id: int, db: Session):
     else:
         db.query(models.OpeningHours).filter(models.OpeningHours.id == id).delete()
     db.commit()
+
+
+# Restaurants information
+
+def get_restaurants(db: Session):
+    current_date = datetime.now(tz=pytz.timezone("Europe/Paris"))
+    weekday, current_time = current_date.weekday(), current_date.time()
+    restaurant_names = [r.place for r in db.query(models.OpeningHours.place.distinct).distinct()]
+    restaurants = []
+
+    for name in restaurant_names:
+        restaurant = {}
+        restaurant["name"] = name
+        first_timeslot = get_timeslot(name, weekday, True, db)
+        second_timeslot = get_timeslot(name, weekday, False, db)
+
+        if (first_timeslot and first_timeslot[0] <= current_time < first_timeslot[1]) or (
+                second_timeslot and second_timeslot[0] <= current_time < second_timeslot[1]):
+            restaurant["status"] = True
+        else:
+            restaurant["status"] = False
+
+        if first_timeslot and second_timeslot:
+            restaurant["timetable"] = (
+                f"{first_timeslot[0].hour:{'0'}{2}}h{first_timeslot[0].minute:{'0'}{2}}-"
+                f"{first_timeslot[1].hour:{'0'}{2}}h{first_timeslot[1].minute:{'0'}{2}} / "
+                f"{second_timeslot[0].hour:{'0'}{2}}h{second_timeslot[0].minute:{'0'}{2}}-"
+                f"{second_timeslot[1].hour:{'0'}{2}}h{second_timeslot[1].minute:{'0'}{2}}")
+        elif first_timeslot:
+            restaurant["timetable"] = (
+                f"{first_timeslot[0].hour:{'0'}{2}}h{first_timeslot[0].minute:{'0'}{2}}-"
+                f"{first_timeslot[1].hour:{'0'}{2}}h{first_timeslot[1].minute:{'0'}{2}}")
+        else:
+            restaurant["timeslot"] = "-"
+
+        if restaurant["status"]:
+            waiting_time = db.query(
+                models.Records.waiting_time
+            ).filter(
+                models.Records.place == name
+            ).order_by(
+                models.Records.date.desc()
+            ).first()
+            waiting_time_minutes = round(waiting_time[0].total_seconds() / 60)
+            restaurant["waiting_time"] = waiting_time_minutes
+        else:
+            restaurant["waiting_time"] = None
+
+        restaurants.append(restaurant)
+
+    return restaurants

backend/db/models.py

@@ -11,7 +11,7 @@ class Records(Base):
     __tablename__ = "records"
 
     id = Column(Integer, primary_key=True, index=True)
-    place = Column(String(10))
+    place = Column(String(30))
     date = Column(DateTime)
     density = Column(Float)
     waiting_time = Column(Interval)
@@ -24,7 +24,7 @@ class Comments(Base):
     id = Column(Integer, primary_key=True, index=True)
     content = Column(Text)
     published_at = Column(DateTime)
-    place = Column(String(10))
+    place = Column(String(30))
 
 
 class News(Base):
@@ -36,7 +36,7 @@ class News(Base):
     content = Column(Text)
     published_at = Column(DateTime)
     end_date = Column(DateTime)
-    place = Column(String(10))
+    place = Column(String(30))
 
 
 class OpeningHours(Base):
@@ -44,7 +44,7 @@ class OpeningHours(Base):
     __tablename__ = "opening_hours"
 
     id = Column(Integer, primary_key=True, index=True)
-    place = Column(String(10))
+    place = Column(String(30))
     day = Column(Integer)
     timeslot = Column(Boolean)
     open_time = Column(Time)

backend/routers/opening_hours.py

+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+from typing import List
+
+from db import schemas, crud
+from db.database import get_db
+
+
+router = APIRouter(prefix="/api", tags=["opening_hours"])
+
+
+@router.get('/{place}/opening_hours', response_model=List[schemas.OpeningHours])
+async def get_opening_hours(place: str, db: Session = Depends(get_db)):
+    return crud.get_opening_hours(place, db)
+
+
+@router.get('/{place}/opening_hours/{day}/{timeslot}', response_model=List[schemas.OpeningHours])
+async def get_timeslot(place: str, day: int, timeslot: bool, db: Session = Depends(get_db)):
+    return crud.get_timeslot(place, day, timeslot, db)
+
+
+@router.post('/opening_hours', response_model=schemas.OpeningHours)
+async def create_opening_hours(opening_hours: schemas.OpeningHoursBase, db: Session = Depends(get_db)):
+    return crud.create_opening_hours(opening_hours, db)
+
+
+@router.delete('/opening_hours/{id}', response_model=None)
+async def delete_opening_hours(id: int, db: Session = Depends(get_db)):
+    return crud.delete_opening_hours(id, db)

backend/routers/stats.py

 from fastapi import APIRouter, Depends
 from sqlalchemy.orm import Session
+from typing import List
 
 from db import schemas, crud
 from db.database import get_db
@@ -22,9 +23,10 @@ async def stats(place: str, db: Session = Depends(get_db)):
 async def stats(place: str, db: Session = Depends(get_db)):
     return crud.get_current_graph(place, db)
 
+
 @router.get('/{place}/opening_hours', response_model=List[schemas.OpeningHours])
-async def get_opening_hours(place: str, page: int = 1, db: Session = Depends(get_db)):
-    return crud.get_opening_hours(place, page, db)
+async def get_opening_hours(place: str, db: Session = Depends(get_db)):
+    return crud.get_opening_hours(place, db)
 
 
 @router.get('/{place}/opening_hours/{day}/{timeslot}', response_model=List[schemas.OpeningHours])
@@ -40,3 +42,8 @@ async def create_opening_hours(opening_hours: schemas.OpeningHoursBase, db: Sess
 @router.delete('/opening_hours/{id}', response_model=None)
 async def delete_opening_hours(id: int, db: Session = Depends(get_db)):
     return crud.delete_opening_hours(id, db)
+
+
+@router.get('/restaurants', response_model=List[dict])
+async def get_restaurants(db: Session = Depends(get_db)):
+    return crud.get_restaurants(db)
\ No newline at end of file