d0863325a5
But for real, it's a real shitstorm in there.
- Added context by making the init go through a function
and not implicitly happen via imports
- Fixup all context and contextless function mixups
- splitup the models in sensible different files
- give the dump of view functions in views/__init__.py their own file
- add all routes via blueprints, not half of them
- move the slack notifications function and class to its own file,
no idea what it was doing in a views file in the first place.
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
from datetime import datetime
|
|
|
|
from .database import db
|
|
from .user import User
|
|
|
|
|
|
class OrderItem(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
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(
|
|
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))
|
|
|
|
def configure(self, user, order, product):
|
|
self.user = user
|
|
self.order = order
|
|
self.product = product
|
|
|
|
def get_name(self):
|
|
if self.user_id is not None and self.user_id > 0:
|
|
return self.user.username
|
|
return self.name
|
|
|
|
def __repr__(self):
|
|
product_name = None
|
|
if self.product:
|
|
product_name = self.product.name
|
|
return 'Order %d: %s wants %s' % (self.order_id or 0, self.get_name(),
|
|
product_name or 'None')
|
|
|
|
def can_delete(self, order_id, user_id, name):
|
|
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()
|
|
if user and user.is_admin():
|
|
return True
|
|
return False
|