haldis/app/models.py
2015-03-28 00:44:58 +01:00

115 lines
3.3 KiB
Python

from datetime import datetime
from app import db
# Create database models
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True)
admin = db.Column(db.Boolean)
bias = db.Column(db.Integer)
runs = db.relationship('Order', backref='courrier', lazy='dynamic')
orders = db.relationship('OrderItem', backref='user', lazy='dynamic')
def configure(self, username, admin, bias):
self.username = username
self.admin = admin
self.bias = bias
def is_authenticated(self):
return True
def is_active(self):
return True
def is_admin(self):
return self.admin
def is_anonymous(self):
return False
def get_id(self):
try:
return unicode(self.id) # python 2
except NameError:
return str(self.id) # python 3
def __repr__(self):
return '%s' % self.username
class Location(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(120))
address = db.Column(db.String(254))
website = db.Column(db.String(120))
food = db.relationship('Food', backref='location', lazy='dynamic')
orders = db.relationship('Order', backref='location', lazy='dynamic')
def configure(self, name, address, website):
self.name = name
self.address = address
self.website = website
def __repr__(self):
return '%s: %s' % (self.name, self.address)
class Food(db.Model):
id = db.Column(db.Integer, primary_key=True)
location_id = db.Column(db.Integer, db.ForeignKey('location.id'))
name = db.Column(db.String(120))
price = db.Column(db.Integer)
orders = db.relationship('OrderItem', backref='food', lazy='dynamic')
def configure(self, location, name, price):
self.location = location
self.name = name
self.price = price
def __repr__(self):
return '%s' % self.name
class Order(db.Model):
id = db.Column(db.Integer, primary_key=True)
courrier_id = db.Column(db.Integer, db.ForeignKey('user.id'))
location_id = db.Column(db.Integer, db.ForeignKey('location.id'))
starttime = db.Column(db.DateTime)
stoptime = db.Column(db.DateTime)
orders = db.relationship('OrderItem', backref='order', lazy='dynamic')
def configure(self, courrier, location, starttime, stoptime):
self.courrier = courrier
self.location = location
self.starttime = starttime
self.stoptime = stoptime
def __repr__(self):
return 'Order %s' % (self.location.name)
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'))
food_id = db.Column(db.Integer, db.ForeignKey('food.id'))
def configure(self, user, order, food):
self.user = user
self.order = order
self.food = food
def __repr__(self):
return 'OrderItem'
def can_delete(self, order_id, user_id):
if self.user_id != user_id:
return False
if int(self.order_id) != int(order_id):
return False
if self.order.stoptime and self.order.stoptime < datetime.now():
return False
return True