Upload 4 files
Browse files- orders/models.py +25 -0
- orders/routes.py +61 -0
- orders/schemas.py +33 -0
- orders/services.py +44 -0
orders/models.py
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# models.py
|
2 |
+
from sqlalchemy import Column, ForeignKey, Integer, String, Float, JSON, ARRAY
|
3 |
+
from sqlalchemy.orm import relationship
|
4 |
+
from core.database import Base
|
5 |
+
|
6 |
+
class Meal(Base):
|
7 |
+
__tablename__ = "meals"
|
8 |
+
|
9 |
+
id = Column(Integer, primary_key=True, index=True)
|
10 |
+
name = Column(String, index=True)
|
11 |
+
description = Column(String)
|
12 |
+
price = Column(Float)
|
13 |
+
|
14 |
+
orders = relationship("Order", back_populates="meal")
|
15 |
+
|
16 |
+
class Order(Base):
|
17 |
+
__tablename__ = "orders"
|
18 |
+
|
19 |
+
id = Column(Integer, primary_key=True, index=True)
|
20 |
+
user_id = Column(Integer, ForeignKey("users.id"))
|
21 |
+
meal_id = Column(Integer, ForeignKey("meals.id"))
|
22 |
+
quantity = Column(Integer)
|
23 |
+
|
24 |
+
user = relationship("User", back_populates="orders")
|
25 |
+
meal = relationship("Meal", back_populates="orders")
|
orders/routes.py
ADDED
@@ -0,0 +1,61 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from fastapi import APIRouter, status, Depends, HTTPException
|
2 |
+
from typing import List, Optional
|
3 |
+
from core.database import get_db
|
4 |
+
from core.security import get_current_user
|
5 |
+
from orders.services import create_meal, get_meals, get_meal, update_meal, delete_meal, create_user_order, get_user_orders
|
6 |
+
from orders.schemas import OrderCreate, OrderBase, Order, MealBase, MealCreate, MealUpdate, Meal
|
7 |
+
from sqlalchemy.orm import Session
|
8 |
+
from services.recommendation_service import MealRecommender
|
9 |
+
|
10 |
+
|
11 |
+
order_router = APIRouter(
|
12 |
+
prefix="/orders",
|
13 |
+
tags=["Orders"],
|
14 |
+
responses={404: {"description": "Not found"}, 401: {"description": "Unauthorized"}}
|
15 |
+
)
|
16 |
+
|
17 |
+
meal_router = APIRouter(
|
18 |
+
prefix="/meals",
|
19 |
+
tags=["Meals"],
|
20 |
+
responses={404: {"description": "Not found"}, 401: {"description": "Unauthorized"}}
|
21 |
+
)
|
22 |
+
|
23 |
+
|
24 |
+
@meal_router.get("/health", tags=["Health"])
|
25 |
+
def health_check():
|
26 |
+
return {"status": "ok"}
|
27 |
+
|
28 |
+
@meal_router.get("/", response_model=MealBase)
|
29 |
+
async def meal_create(data: MealCreate, db: Session = Depends(get_db)):
|
30 |
+
return create_meal(db, data)
|
31 |
+
|
32 |
+
@meal_router.get("/", response_model=List[MealBase])
|
33 |
+
def meals_get(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
|
34 |
+
return get_meals(db, skip=skip, limit=limit)
|
35 |
+
|
36 |
+
@meal_router.get("/{meal_id}", response_model=MealBase)
|
37 |
+
def meal_get(meal_id: int, db: Session = Depends(get_db)):
|
38 |
+
return get_meal(db, meal_id)
|
39 |
+
|
40 |
+
@meal_router.put("/{meal_id}", response_model=MealBase)
|
41 |
+
def meal_update(meal_id: int, data: MealUpdate, db: Session = Depends(get_db)):
|
42 |
+
return update_meal(db, meal_id, data)
|
43 |
+
|
44 |
+
@meal_router.delete("/{meal_id}", response_model=MealBase)
|
45 |
+
def meal_delete(meal_id: int, db: Session = Depends(get_db)):
|
46 |
+
return delete_meal(db, meal_id)
|
47 |
+
|
48 |
+
@order_router.post("/", response_model=OrderBase)
|
49 |
+
def create_order(order: OrderCreate, current_user: OrderBase = Depends(get_current_user), db: Session = Depends(get_db)):
|
50 |
+
return create_user_order(db, order, current_user.id)
|
51 |
+
|
52 |
+
@order_router.get("/", response_model=List[OrderBase])
|
53 |
+
def read_user_orders(skip: int = 0, limit: int = 100, current_user: OrderBase = Depends(get_current_user), db: Session = Depends(get_db)):
|
54 |
+
return get_user_orders(db, user_id=current_user.id, skip=skip, limit=limit)
|
55 |
+
|
56 |
+
@meal_router.get("/recommendations/", response_model=List[MealBase])
|
57 |
+
async def get_recommendations(current_user: OrderBase = Depends(get_current_user), db: Session = Depends(get_db)):
|
58 |
+
recommender = MealRecommender()
|
59 |
+
recommendations = recommender.get_recommendations(current_user)
|
60 |
+
return recommendations
|
61 |
+
|
orders/schemas.py
CHANGED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from pydantic import BaseModel
|
2 |
+
|
3 |
+
|
4 |
+
class MealBase(BaseModel):
|
5 |
+
name: str
|
6 |
+
description: str
|
7 |
+
price: float
|
8 |
+
|
9 |
+
class MealCreate(MealBase):
|
10 |
+
pass
|
11 |
+
|
12 |
+
class MealUpdate(MealBase):
|
13 |
+
pass
|
14 |
+
|
15 |
+
class Meal(MealBase):
|
16 |
+
id: int
|
17 |
+
|
18 |
+
class Config:
|
19 |
+
orm_mode = True
|
20 |
+
|
21 |
+
class OrderBase(BaseModel):
|
22 |
+
quantity: int
|
23 |
+
|
24 |
+
class OrderCreate(OrderBase):
|
25 |
+
meal_id: int
|
26 |
+
|
27 |
+
class Order(OrderBase):
|
28 |
+
id: int
|
29 |
+
user_id = int
|
30 |
+
meal_id = int
|
31 |
+
|
32 |
+
class Config:
|
33 |
+
orm_mode = True
|
orders/services.py
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from sqlalchemy.orm import Session
|
2 |
+
from typing import List
|
3 |
+
from orders.schemas import MealCreate, MealUpdate, OrderCreate
|
4 |
+
from orders.models import Meal, Order
|
5 |
+
|
6 |
+
def get_meals(db: Session, skip: int = 0, limit: int = 100):
|
7 |
+
return db.query(Meal).offset(skip).limit(limit).all()
|
8 |
+
|
9 |
+
def create_meal(db: Session, meal: MealCreate):
|
10 |
+
db_meal = Meal(**meal.dict())
|
11 |
+
db.add(db_meal)
|
12 |
+
db.commit()
|
13 |
+
db.refresh(db_meal)
|
14 |
+
return db_meal
|
15 |
+
|
16 |
+
def get_meal(db: Session, meal_id: int):
|
17 |
+
return db.query(Meal).filter(Meal.id == meal_id).first()
|
18 |
+
|
19 |
+
def update_meal(db: Session, meal_id: int, meal: MealUpdate):
|
20 |
+
db_meal = db.query(Meal).filter(Meal.id == meal_id).first()
|
21 |
+
if db_meal:
|
22 |
+
update_data = meal.dict(exclude_unset=True)
|
23 |
+
for key, value in update_data.items():
|
24 |
+
setattr(db_meal, key, value)
|
25 |
+
db.commit()
|
26 |
+
db.refresh(db_meal)
|
27 |
+
return db_meal
|
28 |
+
|
29 |
+
def delete_meal(db: Session, meal_id: int):
|
30 |
+
db_meal = db.query(Meal).filter(Meal.id == meal_id).first()
|
31 |
+
if db_meal:
|
32 |
+
db.delete(db_meal)
|
33 |
+
db.commit()
|
34 |
+
return db_meal
|
35 |
+
|
36 |
+
def create_user_order(db: Session, order: OrderCreate, user_id: int):
|
37 |
+
db_order = Order(**order.dict(), user_id=user_id)
|
38 |
+
db.add(db_order)
|
39 |
+
db.commit()
|
40 |
+
db.refresh(db_order)
|
41 |
+
return db_order
|
42 |
+
|
43 |
+
def get_user_orders(db: Session, user_id: int, skip: int = 0, limit: int = 100):
|
44 |
+
return db.query(Order).filter(Order.user_id == user_id).offset(skip).limit(limit).all()
|