class InventoryOrder(db.Model): __tablename__ = 'InventoryOrder' __table_args__ = {} #column definitions cost = db.Column(u'cost', db.DECIMAL(precision=7, scale=2), nullable=False) cur_quantity = db.Column(u'cur_quantity', db.Integer(), nullable=False) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) inventory_id = db.Column(u'inventory_id', db.Integer(), db.ForeignKey('Inventory.id')) orig_quantity = db.Column(u'orig_quantity', db.Integer(), nullable=False) def __init__(self, cost, cur_quantity, inv_id, orig_quant): self.cost = cost self.cur_quantity = cur_quantity self.inventory_id = inv_id self.orig_quantity = orig_quant current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions Inventory = db.relationship('Inventory', primaryjoin='InventoryOrder.inventory_id==Inventory.id')
class Giftcard(db.Model): __tablename__ = 'Giftcard' __table_args__ = {} #column definitions billing_id = db.Column(u'billing_id', db.Integer(), db.ForeignKey('Billing.id')) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) number = db.Column(u'number', db.String(length=24), nullable=False, unique=True) pin = db.Column(u'pin', db.Integer(), nullable=False) def __init__(self, billing_id, number, pin): self.billing_id = billing_id self.number = number self.pin = pin current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions Billing = db.relationship('Billing', primaryjoin='Giftcard.billing_id==Billing.id')
class OrderPayment(db.Model): __tablename__ = 'OrderPayment' id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) order_id = db.Column(u'order_id', db.Integer(), db.ForeignKey('Order.id')) billing_id = db.Column(u'billing_id', db.Integer(), db.ForeignKey('Billing.id')) price = db.Column(u'price', db.DECIMAL(precision=7, scale=2), nullable=False) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) def __init__(self, order_id, billing_id, price): self.order_id = order_id self.billing_id = billing_id self.price = price current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions Order = db.relationship('Order', primaryjoin='OrderPayment.order_id==Order.id') Billing = db.relationship('Billing', primaryjoin='OrderPayment.billing_id==Billing.id')
class Creditcard(db.Model): __tablename__ = 'Creditcard' id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) address_id = db.Column(u'address_id', db.Integer(), db.ForeignKey('UserAddress.id')) billing_id = db.Column(u'billing_id', db.Integer(), db.ForeignKey('Billing.id')) name = db.Column(u'name', db.String(length=64), nullable=False) cc_number = db.Column(u'cc_number', db.BigInteger(), nullable=False) sec_number = db.Column(u'sec_number', db.Integer(), nullable=False) exp_date = db.Column(u'exp_date', db.Date(), nullable=False) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) def __init__(self, name, address_id, billing_id, cc_number, sec_number, exp_date): self.address_id = address_id self.billing_id = billing_id self.name = name self.cc_number = cc_number self.sec_number = sec_number self.exp_date = exp_date current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions UserAddress = db.relationship('UserAddress', primaryjoin='Creditcard.address_id==UserAddress.id') Billing = db.relationship('Billing', primaryjoin='Creditcard.billing_id==Billing.id')
class Book(db.Model): __tablename__ = 'Book' __table_args__ = {} #column definitions date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) isbn = db.Column(u'isbn', db.BigInteger(), nullable=False, unique=True) price = db.Column(u'price', db.DECIMAL(precision=7, scale=2), nullable=False) publisher_id = db.Column(u'publisher_id', db.Integer(), db.ForeignKey('Publisher.id')) title = db.Column(u'title', db.Text(), nullable=False) def __init__(self, isbn, title, price, pub_id): current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time self.isbn = isbn self.title = title self.price = price self.publisher_id = pub_id def __repr__(self): return "<Book id: {0}, title: {1}>".format(self.id, self.title) #db.relationship definitions Publisher = db.relationship('Publisher', primaryjoin='Book.publisher_id==Publisher.id') Authors = db.relationship('Author', secondary=Book_Author, backref=db.backref('books')) Categories = db.relationship('Category', secondary=Book_Category, backref=db.backref('books'))
class CartItem(db.Model): __tablename__ = 'CartItem' __table_args__ = {} id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) cart_id = db.Column(u'cart_id', db.Integer(), db.ForeignKey('Cart.id')) book_id = db.Column(u'book_id', db.Integer(), db.ForeignKey('Book.id')) quantity = db.Column(u'quantity', db.Integer()) status = db.Column(u'status', db.String(length=10), nullable=False) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) def __init__(self, cart_id, book_id, quantity, status): self.cart_id = cart_id self.book_id = book_id self.quantity = quantity self.status = status current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions Book = db.relationship('Book', primaryjoin='CartItem.book_id==Book.id') Cart = db.relationship('Cart', primaryjoin='CartItem.cart_id==Cart.id')
class Order(db.Model): __tablename__ = 'Order' id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) user_id = db.Column(u'user_id', db.Integer(), db.ForeignKey('User.id')) cart_id = db.Column(u'cart_id', db.Integer(), db.ForeignKey('Cart.id')) shipping = db.Column(u'shipping', db.DECIMAL(precision=7, scale=2), nullable=False) tax = db.Column(u'tax', db.DECIMAL(precision=7, scale=2), nullable=False) status = db.Column(u'status', db.String(length=10), nullable=False) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) def __init__(self, user_id, cart_id, shipping, tax, status): self.user_id = user_id self.cart_id = cart_id self.shipping = shipping self.tax = tax self.subtotal = 0 self.status = status current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions User = db.relationship('User', primaryjoin='Order.user_id==User.id') Cart = db.relationship('Cart', primaryjoin='Order.cart_id==Cart.id')
class OrderItem(db.Model): __tablename__ = 'OrderItem' id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) order_id = db.Column(u'order_id', db.Integer(), db.ForeignKey('Order.id')) book_id = db.Column(u'book_id', db.Integer(), db.ForeignKey('Book.id')) quantity = db.Column(u'quantity', db.Integer()) cost = db.Column(u'cost', db.DECIMAL(precision=7, scale=2), nullable=False) price = db.Column(u'price', db.DECIMAL(precision=7, scale=2), nullable=False) status = db.Column(u'status', db.String(length=10), nullable=False) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) def __init__(self, order_id, book_id, quantity, cost, price, status): self.order_id = order_id self.book_id = book_id self.quantity = quantity self.cost = cost self.price = price self.status = status current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions Order = db.relationship('Order', primaryjoin='OrderItem.order_id==Order.id') Book = db.relationship('Book', primaryjoin='OrderItem.book_id==Book.id')
class User(BaseModel, UserMixin): """ This is the user model (used for both login and registration) will consider separating login to a separate table/db """ id = db.Column(db.Integer(), primary_key=True) student_id = db.Column(db.String(255), unique=True, nullable=True) email = db.Column(db.String(255), unique=True) first_name = db.Column(db.String(255)) last_name = db.Column(db.String(255)) password = db.Column(db.String(255), default="") active = db.Column(db.Boolean(), default=False) confirmed_at = db.Column(db.DateTime()) roles = db.relationship('Role', secondary=roles_users, backref=db.backref('users', lazy='dynamic')) registrations = db.relationship("Registration") _default_fields = ["student_id", "email", "first_name", "last_name"] def get_auth_token(self, allow_refresh=False): """ Create an authentication token for this user :param allow_refresh: Doesn't do anything at the moment except add to payload :return: The string token """ payload = { 'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=3), 'iat': datetime.datetime.utcnow(), 'sub': self.email, 'refresh': allow_refresh } token = jwt.encode(payload, app.config.get('SECRET_KEY'), 'HS256').decode() return token @staticmethod def decode_auth_token(token): """ Decode an authentication token provided by JWT :param token: The string/bytes token object :return: The user associated and related payload """ try: payload = jwt.decode(token.encode(), app.config.get("SECRET_KEY")) user = User.query.filter(User.email == payload["sub"]).first() if not user: raise InvalidTokenError() return user, payload except jwt.ExpiredSignatureError: raise Unauthorized("This token has expired") except jwt.InvalidTokenError: traceback.print_exc() raise Unauthorized("Unable to authenticate with that information")
class Cart(db.Model): __tablename__ = 'Cart' __table_args__ = {} #column definitions date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) user_id = db.Column(u'user_id', db.Integer(), db.ForeignKey('User.id')) status = db.Column(u'status', db.String(length=10), nullable=False) def __init__(self, user_id, status): self.user_id = user_id self.status = status current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time
class Club(db.Model): id = db.Column(db.Integer(), primary_key=True) name = db.Column(db.String(255), unique=True) abbreviation = db.Column(db.String(255), unique=True) logo = db.Column(db.String(255)) description = db.Column(db.String(255), default="") registrations = db.relationship("Registration") _default_fields = ["name", "id", "description", "logo", "abbreviation"]
class User(db.Model): __tablename__ = 'User' __table_args__ = {} #column definitions date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) email = db.Column(u'email', db.String(length=128), nullable=False, unique=True) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) name = db.Column(u'name', db.String(length=64)) password = db.Column(u'password', db.String(length=64), nullable=False) phone = db.Column(u'phone', db.BigInteger()) username = db.Column(u'username', db.String(length=32), nullable=False, unique=True) authenticated = False def __init__(self, email, username, password, name=None, phone=None): self.name = name if name != '' else None self.username = username self.email = email self.password = generate_encrypted_password(password) self.phone = str_to_digits(phone) current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time self.authenticated = False def __repr__(self): return "<User id: {id}, username: {usn}, name: {name},>".format( id=self.id, name=self.name, usn=self.username, auth=self.authenticated) #Methods for Flask-Login def get_id(self): return unicode(self.id) def is_anonymous(self): return not isinstance(self.id, long) def is_authenticated(self): return self.authenticated def is_active(self): return self.is_authenticated() def is_admin(self): #TODO: Test whehter admin relationship exists return False #db.relationship definitions Carts = db.relationship('Cart', primaryjoin='User.id==Cart.user_id', backref=db.backref('User'))
class Billing(db.Model): __tablename__ = 'Billing' __table_args__ = {} #column definitions date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) type = db.Column(u'type', db.String(length=12)) user_id = db.Column(u'user_id', db.Integer(), db.ForeignKey('User.id')) def __init__(self, type, user_id): self.type = type self.user_id = user_id current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions User = db.relationship('User', primaryjoin='Billing.user_id==User.id')
class Admin(db.Model): __tablename__ = 'Admin' __table_args__ = {} #column definitions date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) level = db.Column(u'level', db.String(length=8), nullable=False) user_id = db.Column(u'user_id', db.Integer(), db.ForeignKey('User.id')) def __init__(self, level, user_id): self.level = level self.user_id = user_id current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions User = db.relationship('User', primaryjoin='Admin.user_id==User.id', backref=db.backref('User', lazy='joined'))
class Registration(BaseModel): id = db.Column(db.Integer, primary_key=True) user_id = db.Column(db.Integer(), db.ForeignKey('user.id')) club_id = db.Column(db.Integer(), db.ForeignKey('club.id')) evidence = db.Column(db.String(255)) expiry = db.Column(db.DateTime) __table_args__ = (Index('_user_club_registration_composite', 'user_id', 'club_id'), ) _default_fields = ["id", "expiry"] user = db.relationship("User") club = db.relationship("Club") def to_dict(self): response = BaseModel.to_dict(self) response["user"] = BaseModel.to_dict(self.user) response["club"] = BaseModel.to_dict(self.club) return response
class UserAddress(db.Model): __tablename__ = 'UserAddress' __table_args__ = {} #column definitions address1 = db.Column(u'address1', db.String(length=64), nullable=False) address2 = db.Column(u'address2', db.String(length=32)) city = db.Column(u'city', db.String(length=32), nullable=False) country = db.Column(u'country', db.String(length=32), nullable=False) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) name = db.Column(u'name', db.String(length=64), nullable=False) state = db.Column(u'state', db.String(length=20)) status = db.Column(u'status', db.String(length=10), nullable=False) user_id = db.Column(u'user_id', db.Integer(), db.ForeignKey('User.id')) zip = db.Column(u'zip', db.Integer()) def __init__(self, user_id, name, adr1, adr2, city, state, zip, country): self.user_id = user_id self.name = name self.address1 = adr1 self.address2 = adr2 self.city = city self.state = state self.zip = zip self.country = country self.status = 'OK' current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions User = db.relationship('User', primaryjoin='UserAddress.user_id==User.id', backref=db.backref('UserAddress'))
class Category(db.Model): __tablename__ = 'Category' __table_args__ = {} #column definitions date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) name = db.Column(u'name', db.String(length=64), nullable=False, unique=True) def __init__(self, name): self.name = name current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time #db.relationship definitions Books = db.relationship('Book', secondary=Book_Category)
class Author(db.Model): __tablename__ = 'Author' __table_args__ = {} #column definitions birth_date = db.Column(u'birth_date', db.Date()) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) name = db.Column(u'name', db.Text(), nullable=False) def __init__(self, name, b_date): current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time self.name = name self.birth_date = b_date #db.relationship definitions Books = db.relationship('Book', secondary=Book_Author)
class Publisher(db.Model): __tablename__ = 'Publisher' __table_args__ = {} #column definitions city = db.Column(u'city', db.String(length=20)) country = db.Column(u'country', db.String(length=20)) date_created = db.Column(u'date_created', db.DateTime(), nullable=False) date_modified = db.Column(u'date_modified', db.DateTime(), nullable=False) established_date = db.Column(u'established_date', db.Date()) id = db.Column(u'id', db.Integer(), primary_key=True, nullable=False) name = db.Column(u'name', db.String(length=32), nullable=False) state = db.Column(u'state', db.String(length=20)) def __init__(self, name, city, state, country, est_date): self.name = name self.city = city self.state = state self.country = country self.established_date = est_date current_time = now().strftime("%Y-%m-%d %H:%M") self.date_created = current_time self.date_modified = current_time
import datetime import traceback import jwt from flask import current_app as app from flask_login import UserMixin from jwt import InvalidTokenError from model import BaseModel from model.database import db # Define models from utilities.exception_router import Unauthorized roles_users = db.Table( 'roles_users', db.Column('user_id', db.Integer(), db.ForeignKey('user.id')), db.Column('role_id', db.Integer(), db.ForeignKey('role.id'))) class User(BaseModel, UserMixin): """ This is the user model (used for both login and registration) will consider separating login to a separate table/db """ id = db.Column(db.Integer(), primary_key=True) student_id = db.Column(db.String(255), unique=True, nullable=True) email = db.Column(db.String(255), unique=True) first_name = db.Column(db.String(255)) last_name = db.Column(db.String(255))
#This file was modified after being generated by sqlautocode to work with #Flask-SQLAlchemy from datetime import datetime from model.database import db from functions.utils import str_to_digits, generate_encrypted_password #Alias a function now = datetime.now #Setup helper tables Book_Author = db.Table(u'Book_Author', db.Column(u'id', db.Integer(), primary_key=True, nullable=False), db.Column(u'book_id', db.Integer(), db.ForeignKey('Book.id')), db.Column(u'author_id', db.Integer(), db.ForeignKey('Author.id')), db.Column(u'date_created', db.DateTime(), nullable=False), db.Column(u'date_modified', db.DateTime(), nullable=False), ) Book_Category = db.Table(u'Book_Category', db.Column(u'id', db.Integer(), primary_key=True, nullable=False), db.Column(u'book_id', db.Integer(), db.ForeignKey('Book.id')), db.Column(u'category_id', db.Integer(), db.ForeignKey('Category.id')), db.Column(u'date_created', db.DateTime(), nullable=False), db.Column(u'date_modified', db.DateTime(), nullable=False), ) class Admin(db.Model): __tablename__ = 'Admin'
class Role(BaseModel, RoleMixin): id = db.Column(db.Integer(), primary_key=True) name = db.Column(db.String(80), unique=True) description = db.Column(db.String(255)) _default_fields = ["id", "name", "description"]