Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import uuid
from sqlalchemy.orm import Session
from app.api.v1.endpoints.beverage.schemas import BeverageCreateSchema
from app.database.models import Beverage
def create_beverage(schema: BeverageCreateSchema, db: Session):
entity = Beverage(**schema.dict())
db.add(entity)
db.commit()
return entity
def get_beverage_by_id(beverage_id: uuid.UUID, db: Session):
entity = db.query(Beverage).filter(Beverage.id == beverage_id).first()
return entity
def get_beverage_by_name(beverage_name: str, db: Session):
entity = db.query(Beverage).filter(Beverage.name == beverage_name).first()
return entity
def get_all_beverages(db: Session):
return db.query(Beverage).all()
def update_beverage(beverage: Beverage, changed_beverage: BeverageCreateSchema, db: Session):
for key, value in changed_beverage.dict().items():
setattr(beverage, key, value)
db.commit()
db.refresh(beverage)
return beverage
def delete_beverage_by_id(beverage_id: uuid.UUID, db: Session):
entity = get_beverage_by_id(beverage_id, db)
if entity:
db.delete(entity)
db.commit()