-
Notifications
You must be signed in to change notification settings - Fork 8
/
models.py
25 lines (21 loc) · 739 Bytes
/
models.py
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
import sqlalchemy
from db import db, metadata, sqlalchemy
pizzas = sqlalchemy.Table(
"pizzas",
metadata,
sqlalchemy.Column("id", sqlalchemy.Integer, primary_key=True),
sqlalchemy.Column("name", sqlalchemy.String),
sqlalchemy.Column("price", sqlalchemy.Integer),
sqlalchemy.Column("is_cheese_stuffed", sqlalchemy.Boolean, default=False, server_default='False')
)
class Pizza:
@classmethod
async def get(cls, id):
query = pizzas.select().where(pizzas.c.id == id)
pizza = await db.fetch_one(query)
return pizza
@classmethod
async def create(cls, **pizza):
query = pizzas.insert().values(**pizza)
pizza_id = await db.execute(query)
return pizza_id