61 lines
675 B
Python
61 lines
675 B
Python
import random
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.db.models import Item
|
|
|
|
|
|
def get_items(
|
|
db: Session
|
|
):
|
|
|
|
return db.query(Item).all()
|
|
|
|
|
|
def create_item(
|
|
db: Session,
|
|
title: str
|
|
):
|
|
|
|
item = Item(
|
|
title=title
|
|
)
|
|
|
|
db.add(item)
|
|
|
|
db.commit()
|
|
|
|
db.refresh(item)
|
|
|
|
return item
|
|
|
|
|
|
def delete_item(
|
|
db: Session,
|
|
item_id: int
|
|
):
|
|
|
|
item = db.query(Item).filter(
|
|
Item.id == item_id
|
|
).first()
|
|
|
|
if item:
|
|
|
|
db.delete(item)
|
|
|
|
db.commit()
|
|
|
|
return item
|
|
|
|
|
|
def spin_wheel(
|
|
db: Session
|
|
):
|
|
|
|
items = db.query(Item).all()
|
|
|
|
if not items:
|
|
|
|
return None
|
|
|
|
return random.choice(items) |