haldis/app/models/orderitem.py

62 lines
2.1 KiB
Python
Raw Normal View History

2019-09-10 01:06:11 +02:00
"Script for everything OrderItem related in the database"
from datetime import datetime
from .database import db
2019-09-08 00:41:50 +02:00
from .order import Order
from .product import Product
from .user import User
class OrderItem(db.Model):
2019-09-10 01:06:11 +02:00
"Class used for configuring the OrderItem model in the database"
id = db.Column(db.Integer, primary_key=True)
2019-09-05 03:33:29 +02:00
user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
order_id = db.Column(db.Integer, db.ForeignKey("order.id"), nullable=False)
product_id = db.Column(
2019-09-05 03:33:29 +02:00
db.Integer, db.ForeignKey("product.id"), nullable=True
) # TODO make false after init migration
paid = db.Column(
db.Boolean, default=False, nullable=True
) # TODO make false after init migration
extra = db.Column(db.String(254), nullable=True)
name = db.Column(db.String(120))
2019-09-08 00:41:50 +02:00
def configure(self, user: User, order: Order, product: Product) -> None:
2019-09-10 01:06:11 +02:00
"Configure the OrderItem"
# pylint: disable=W0201
self.user = user
self.order = order
self.product = product
2019-09-08 00:41:50 +02:00
def get_name(self) -> str:
2019-09-10 01:06:11 +02:00
"Get the name of the user which 'owns' the item"
if self.user_id is not None and self.user_id > 0:
return self.user.username
return self.name
2019-09-08 00:41:50 +02:00
def __repr__(self) -> str:
product_name = None
if self.product:
product_name = self.product.name
2019-09-05 03:33:29 +02:00
return "Order %d: %s wants %s" % (
self.order_id or 0,
self.get_name(),
product_name or "None",
)
2019-09-10 01:06:11 +02:00
# pylint: disable=W0613
2019-09-08 00:41:50 +02:00
def can_delete(self, order_id: int, user_id: int, name: str) -> bool:
2019-09-10 01:06:11 +02:00
"Check if a user can delete an item"
if int(self.order_id) != int(order_id):
return False
if self.order.stoptime and self.order.stoptime < datetime.now():
return False
if self.user is not None and self.user_id == user_id:
return True
if user_id is None:
return False
user = User.query.filter(User.id == user_id).first()
2019-09-12 18:42:29 +02:00
if user and (user.is_admin() or user == self.order.courrier):
return True
return False