class User(db.Model): id = db.Column(db.Integer, primary_key=True) is_admin = db.Column(db.Boolean) location = db.Column(db.String(120)) name = db.Column(db.String()) phone = db.Column(db.String(255)) email = db.Column(db.String(), unique=True) notifications = db.Column(db.Boolean) have_their_id = db.Column(db.Boolean) requests = db.relationship('Request', back_populates='user') sockets = db.relationship('Socket', back_populates='user') items = db.relationship('Item', backref='user') def __init__(self, email, is_admin=False, name=None, phone=None): self.email = email self.is_admin = is_admin if name is not None: self.name = name if phone is not None: self.phone = phone self.notifications = False self.have_their_id = False def requires_id(self): for item in self.items: if item.entry.requires_checkout \ or item.entry.requires_lottery: return True return False
class Request(db.Model): id = db.Column(db.Integer, primary_key=True) items = db.relationship('RequestItem', backref='request') status = db.Column(db.Enum(RequestStatus)) timestamp = db.Column(db.DateTime) user_id = db.Column(db.Integer, db.ForeignKey('user.id')) requires_id = db.Column(db.Boolean) requires_lottery = db.Column(db.Boolean) user = db.relationship('User', back_populates='requests') proposal = db.Column(db.String()) def __init__(self, items, user_id, proposal=''): self.status = RequestStatus.SUBMITTED self.items = items self.requires_id = self.check_requires_id() self.requires_lottery = self.check_requires_lottery() self.user_id = user_id self.user = User.query.get(user_id) self.timestamp = datetime.now() self.proposal = proposal def __str__(self): return self.user.email + ' ' + str(self.status) \ + ' ' + ', '.join([str(i) for i in self.items]) def check_requires_id(self): for item in self.items: if (item.entry.item_type == ItemType.LOTTERY or item.entry.item_type == ItemType.CHECKOUT): return True return False def check_requires_lottery(self): for item in self.items: if item.entry.item_type == ItemType.LOTTERY: return True return False
class User(db.Model): id = db.Column(db.Integer, primary_key=True) quill_id = db.Column(db.String(), unique=True) verified_email = db.Column(db.Boolean) verification_token = db.Column(db.String(), unique=True) is_admin = db.Column(db.Boolean) location = db.Column(db.String(120)) name = db.Column(db.String()) phone = db.Column(db.String(255)) email = db.Column(db.String()) password_hash = db.Column(db.String()) notifications = db.Column(db.Boolean) have_their_id = db.Column(db.Boolean) requests = db.relationship('Request', back_populates='user') sockets = db.relationship('Socket', back_populates='user') items = db.relationship('Item', backref='user') def __init__(self, quill_id, email, password_hash, verification_token, is_admin): self.quill_id = quill_id self.email = email self.password_hash = password_hash self.is_admin = is_admin self.name = '' self.location = '' self.phone = '' self.notifications = False self.have_their_id = False self.verified_email = False self.verification_token = verification_token def requires_id(self): for item in self.items: if item.entry.requires_checkout \ or item.entry.requires_lottery: return True return False
class RequestItem(db.Model): entry_id = db.Column(db.Integer, db.ForeignKey('inventory_entry.id'), primary_key=True) request_id = db.Column(db.Integer, db.ForeignKey('request.id'), primary_key=True) quantity = db.Column(db.Integer) entry = db.relationship('InventoryEntry') def __init__(self, entry, quantity): self.entry = entry self.quantity = quantity def __str__(self): return str(self.quantity) + 'x ' + self.entry.name
class InventoryEntry(db.Model): id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(120)) description = db.Column(db.String()) category = db.Column(db.String()) link = db.Column(db.String()) max_request_quantity = db.Column(db.Integer) image_src = db.Column(db.String()) item_type = db.Column(db.Enum(ItemType)) is_visible = db.Column(db.Boolean) is_unique = db.Column(db.Boolean) items = db.relationship('Item', backref='entry') requests = db.relationship('Request', secondary=RequestItem.__table__, lazy='select', viewonly=True) def __init__(self, name, description, link, category, tags, image, qty, item_type = ItemType.FREE, max_request_qty = 3): self.name = name self.description = description self.link = link self.category = category self.tags = tags self.image_src = image self.items = [] for i in xrange(int(qty)): self.items.append( Item(self, self.name + " " + str(i+1))) self.max_request_quantity = max_request_qty self.item_type = item_type self.is_visible = True self.is_unique = False @property def quantity(self): """Returns quantity of items that have not been 'claimed' by a request""" requests = RequestItem.query \ .filter_by(entry_id=self.id) \ .join(hardwarecheckout.models.request.Request) \ .filter_by(status=hardwarecheckout.models.request.RequestStatus.APPROVED) \ .with_entities(func.sum(RequestItem.quantity)).scalar() if not requests: requests = 0 return Item.query.filter_by(entry_id = self.id, user = None).count() - requests @property def submitted_request_quantity(self): """Returns number of submitted requests for this entry""" requests = RequestItem.query \ .filter_by(entry_id=self.id) \ .join(hardwarecheckout.models.request.Request) \ .filter_by(status=hardwarecheckout.models.request.RequestStatus.SUBMITTED).count() return requests @property def requires_checkout(self): return self.item_type == ItemType.CHECKOUT @property def requires_lottery(self): return self.item_type == ItemType.LOTTERY def __str__(self): return str(self.name) + " [" + str(self.id) + "]"