class Official(db.Model, UserMixin): officialuid = db.Column(db.Integer, primary_key=True) firstname = db.Column(db.String(50), nullable=False) surname = db.Column(db.String(50), nullable=False) email = db.Column(db.String(240), unique=True, nullable=False) password = db.Column(db.String(255), nullable=False) def get_id(self): try: return str(self.officialuid) except AttributeError: raise NotImplementedError('No `UId` attribute - override `get_id`') def __repr__(self): return f"User('{self.email}')" def verify_password(self, password): return self.password == password
class Game(db.Model): id = db.Column(db.Integer, primary_key=True) player1 = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False) score1 = db.Column(db.Integer, unique=False, nullable=False, default=0) player2 = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False) # need to take a look at this score2 = db.Column(db.Integer, unique=False, nullable=False, default=0) done = db.Column(db.Boolean, unique=False, nullable=False, default=False) over = db.Column(db.Boolean, unique=False, nullable=False, default=False)
class Client(db.Model): __tablename__ = "Clients" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(100), nullable=False) street = db.Column(db.String(256), nullable=False) street_number = db.Column(db.String(6), nullable=False) postal_code = db.Column(db.Integer, nullable=False) city = db.Column(db.String(256), nullable=False) firm = db.Column(db.Boolean, nullable=True) vat_number = db.Column(db.String(20), nullable=True)
class Bill(db.Model): __tablename__ = "Bills" id = db.Column(db.Integer, primary_key=True) client_id = db.Column(db.Integer, db.ForeignKey(Client.id)) subject = db.Column(db.String(256), nullable=False) date = db.Column(db.Date, nullable=False) expiration = db.Column(db.Date, nullable=False) price = db.Column(db.Float(precision=2), nullable=False) cash = db.Column(db.Boolean, nullable=False) paid = db.Column(db.Boolean, nullable=False)
class User(db.Model, UserMixin): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(20), unique=True, nullable=False) email = db.Column(db.String(120), unique=True, nullable=False) password = db.Column(db.String(60), nullable=False) games_won = db.Column(db.Integer(), nullable=False, default=0) friend_list = db.relationship("User", secondary=friends, primaryjoin=(friends.c.self_id == id), secondaryjoin=(friends.c.friend_id == id), backref=db.backref('friends', lazy='dynamic'), lazy='dynamic') games = db.relationship("Game", secondary=game_table, backref=db.backref('games', lazy='dynamic')) def add_friend(self, user): # allows user to add a friend and updates db if not self.is_friends(user) and self.id != user.id: self.friend_list.append(user) user.friend_list.append(self) def is_friends(self, user): # checks if users are already friends return self.friend_list.filter( friends.c.friend_id == user.id).count() > 0 def get_reset_token(self, expires_sec=1800): # for resetting password s = Serializer(application.config['SECRET_KEY'], expires_sec) return s.dumps({'user_id': self.id}).decode('utf-8') @staticmethod def verify_reset_token(token): # verifies correct token for reset password s = Serializer(application.config['SECRET_KEY']) try: user_id = s.loads(token)['user_id'] except: return None return User.query.get(user_id) def __repr__(self): return f"User('{self.username}'')"
class ChessPuzzle(db.Model): __tablename__ = 'Puzzles' puzzle_id = db.Column(db.Integer, primary_key=True, autoincrement=True) description = db.Column(db.String(200)) fen = db.Column(db.String(250), nullable=False) solution = db.Column(db.String(250), nullable=False) type_id = db.Column(db.Integer, db.ForeignKey('PuzzleTypes.type_id')) def __eq__(self, other): eq_puzzle_id = self.puzzle_id == other.puzzle_id eq_description = self.description == other.description eq_fen = self.fen == other.fen eq_solution = self.solution == other.solution eq_type = self.type_id == other.type_id return eq_puzzle_id and eq_description and eq_fen and eq_solution and eq_type def __str__(self): return " id: %s\n descr: %s\n fen: %s\n solution: %s\n type_id: %s\n\n" \ % (self.puzzle_id, self.description, self.fen, self.solution, self.type_id)
class Task(db.Model): id = db.Column(db.Integer, primary_key=True) title = db.Column(db.String(30)) description = db.Column(db.String(200)) complete = db.Column(db.Boolean, nullable=False, default=False) due = db.Column(db.DateTime, nullable=True) user = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
class Officials(UserMixin, db.Model): OfficialUId = db.Column(UUID(as_uuid=True), unique=True, primary_key=True) FirstName = db.Column(db.String(50), nullable=False) Surname = db.Column(db.String(50), nullable=False) email = db.Column(db.String(255), unique=True, nullable=False) PwdHash = db.Column(db.String(255), nullable=False) IsAdmin = db.Column(TINYINT(1), default=0) @property def password(self): raise AttributeError('password is not a readable attribute') def get_id(self): try: return text_type(self.OfficialUId) except AttributeError: raise NotImplementedError( 'No `UserUId` attribute - override `get_id`') def verify_password(self, password): return self.PwdHash == password def check_admin_status(self): if IsAdmin == 1: return True else: return False @login_manager.user_loader def load_user(OfficialUId): return Officials.query.get(str(OfficialUId)) def __repr__(self): return f"User('{self.FirstName}','{self.Surname}', '{self.Email}', '{self.PwdHash}', '{self.IsAdmin}')"
class Astronaut(db.Model): """ Define the model of Astronaut user Args: Form: Inherits the db.Model library Returns: None Attributes: id - Primary key, must be an integer name - Must be string of max length 30 username - Can't be null, can be only one in the database, up to 80 characters email - Must be unique, can't be null and it's string of max 120 characters password - Password is a string of max 180 characters, can't be null admin_id - Integer which is an ID of admin, who created current user """ id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(30)) username = db.Column(db.String(80), unique=True, nullable=False) email = db.Column(db.String(120), unique=True, nullable=False) password = db.Column(db.String(180), unique=False, nullable=False) admin_id = db.Column(db.Integer, db.ForeignKey('admin.id'), nullable=False) # astronauts need to have the admin ID's that created them admin = db.relationship('Admin', backref=db.backref('admins')) #Method returning string as representation of the object def __repr__(self): return '<Astronaut %r>' % self.username
class User(db.Model, UserMixin): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(20), unique=True, nullable=False) email = db.Column(db.String(120), unique=True, nullable=False) password = db.Column(db.String(60), nullable=False) posts = db.relationship('Task', backref='author', lazy=True) def get_reset_token(self, expires_sec=1800): s = Serializer(application.config['SECRET_KEY'], expires_sec) return s.dumps({'user_id': self.id}).decode('utf-8') @staticmethod def verify_reset_token(token): s = Serializer(application.config['SECRET_KEY']) try: user_id = s.loads(token)['user_id'] except: return None return User.query.get(user_id) def __repr__(self): return f"User('{self.username}')"
class Device(db.Model): __tablename__ = "Device" dev_id = db.Column(db.Integer, primary_key=True) device_name = db.Column(db.String) latitude = db.Column(db.Float) longitude = db.Column(db.Float) altitude = db.Column(db.Float) location = db.Column(db.String) user_id = db.Column(db.Integer, db.ForeignKey("TTN_User.user_id")) users = db.relationship("TTN_User", foreign_keys=user_id) def __repr__(self) -> str: return f"<Device {self.dev_id}>"
class User(db.Model, UserMixin): id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(50), unique=True) email = db.Column(db.String(255), unique=True) password = db.Column(db.String(255)) active = db.Column(db.Boolean()) confirmed_at = db.Column(db.DateTime()) roles = db.relationship('Role', secondary=roles_users, backref=db.backref('users', lazy='dynamic'))
class Gateway(db.Model): __tablename__ = "Gateway" gateway_id = db.Column(db.Integer, primary_key=True) gtw_id = db.Column(db.String) latitude = db.Column(db.Float) longitude = db.Column(db.Float) altitude = db.Column(db.Float) location = db.Column(db.String) def __repr__(self) -> str: return f"<Gateway {self.gateway_id}>"
class User(db.Model, UserMixin): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(20), unique=True, nullable=False) firstName = db.Column(db.String(60), nullable=False) address = db.Column(db.String(100), nullable=False) email = db.Column(db.String(120), unique=True, nullable=False) password = db.Column(db.String(60), nullable=False) def __repr__(self): return f"User('{self.username}', '{self.email}',)"
class Connection(db.Model): __tablename__ = "Connection" conn_id = db.Column(db.Integer, primary_key=True) dev_id = db.Column(db.Integer, db.ForeignKey("Device.dev_id")) service_id = db.Column(db.Integer, db.ForeignKey("Service.service_id")) gateway_id = db.Column(db.Integer, db.ForeignKey("Gateway.gateway_id")) rssi = db.Column(db.Float) snr = db.Column(db.Float) devices = db.relationship("Device", foreign_keys=dev_id) services = db.relationship("Service", foreign_keys=service_id) gateways = db.relationship("Gateway", foreign_keys=gateway_id) def __repr__(self) -> str: return f"<Connection {self.conn_id}>"
class User(db.Model): __tablename__ = "Users" id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(40), nullable=False, unique=False) email = db.Column(db.String(40), unique=True, nullable=False) password = db.Column(db.String(200), unique=False, nullable=False) created_on = db.Column(db.DateTime, index=False, unique=False, nullable=True) admin = db.Column(db.Boolean, nullable=False, unique=False) def set_password(self, password): """Create hashed password.""" self.password = generate_password_hash(password, method="sha256") def check_password(self, password): """Check hashed password.""" return check_password_hash(self.password, password) def __repr__(self): return "<User {}>".format(self.username)
class PoliticalParty(db.Model): UId = db.Column(UUID(as_uuid=True), unique=True, primary_key=True) Name = db.Column(db.String(255), nullable=False) def __repr__(self): return f"PoliticalParty('{self.Name}')"
from Application import db, login_manager, application from flask_login import UserMixin from itsdangerous import TimedJSONWebSignatureSerializer as Serializer @login_manager.user_loader # loads in user def load_user(user_id): return User.query.get(int(user_id)) # Association Tables friends = db.Table( 'friends', db.Column('self_id', db.Integer, db.ForeignKey('user.id')), db.Column('friend_id', db.Integer, db.ForeignKey('user.id'))) game_table = db.Table( 'games', db.Column("user_id", db.Integer, db.ForeignKey('user.id')), db.Column("game_id", db.Integer, db.ForeignKey('game.id'))) # Models class Game(db.Model): id = db.Column(db.Integer, primary_key=True) player1 = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False) score1 = db.Column(db.Integer, unique=False, nullable=False, default=0) player2 = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False) # need to take a look at this score2 = db.Column(db.Integer, unique=False, nullable=False, default=0) done = db.Column(db.Boolean, unique=False, nullable=False, default=False) over = db.Column(db.Boolean, unique=False, nullable=False, default=False)
class Election(db.Model, UserMixin): title = db.Column(db.String(255), nullable=False, primary_key=True) election_date = db.Column(db.DATE, nullable=False) start_time = db.Column(db.TIME, nullable=False) end_time = db.Column(db.TIME, nullable=False)
class Candidates(db.Model, UserMixin): title = db.Column(db.String(60), primary_key=True)
db.session.commit() return account.id @staticmethod def deleteAccount(id): account = Account.query.get(id) db.session.delete(account) db.session.commit() # ---------------------------------------------------------------------------------------- # Flask Security # ---------------------------------------------------------------------------------------- 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 Role(db.Model, RoleMixin): id = db.Column(db.Integer(), primary_key=True) name = db.Column(db.String(80), unique=True) description = db.Column(db.String(255)) class User(db.Model, UserMixin): id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(50), unique=True) email = db.Column(db.String(255), unique=True) password = db.Column(db.String(255)) active = db.Column(db.Boolean())
class Role(db.Model, RoleMixin): id = db.Column(db.Integer(), primary_key=True) name = db.Column(db.String(80), unique=True) description = db.Column(db.String(255))
class Account(db.Model): __tablename__ = credentials.tablename id = db.Column(db.Integer, primary_key=True) provider = db.Column(db.String) username = db.Column(db.String) password = db.Column(db.String) passwordlength = db.Column(db.Integer) definedcharacter = db.Column(db.String) question = db.Column(db.String) answer = db.Column(db.String) lastmodify = db.Column(db.DateTime) userid = db.Column(db.ForeignKey('user.id')) # Get dict from Account object. def getDict(self, keyList): origin = self.__dict__ object = dict() for key in keyList: object[key] = origin[key] return object # Get a dict object of an account with the given id (primary key). @staticmethod def getDictOf(id, keyList): return Account.query.get(id).getDict(keyList) # Get all Account objects from database as a dictionary. # Dictionaries get values of keyList. @staticmethod def getObjList(keyList): accounts = Account.query.filter_by(userid=current_user.id).all() list = [] for account in accounts: object = account.getDict(keyList) list.append(object) return list # Modify a database entry @staticmethod def modifyDBAccount(accountDict): account = Account.query.get(accountDict['id']) account.lastmodify = datetime.datetime.today() a = account.__dict__ for key in accountDict: if key in a and a[key] != accountDict[key]: account.__setattr__(key, accountDict[key]) db.session.commit() @staticmethod def insertAccount(accountDict): accountDict['lastmodify'] = datetime.datetime.today() accountDict['userid'] = current_user.id account = Account() for key in accountDict: account.__setattr__(key, accountDict[key]) db.session.add(account) db.session.commit() return account.id @staticmethod def deleteAccount(id): account = Account.query.get(id) db.session.delete(account) db.session.commit()
class PuzzleType(db.Model): __tablename__ = 'PuzzleTypes' type_id = db.Column(db.Integer, primary_key=True, autoincrement=True) description = db.Column(db.String(100), nullable=False)
class Service(db.Model): __tablename__ = "Service" service_id = db.Column(db.Integer, primary_key=True) time = db.Column(db.String) status = db.Column(db.Integer) water_ml = db.Column(db.Integer) countdown_timer = db.Column(db.Integer) water_counter = db.Column(db.Integer) voltage_max = db.Column(db.Float) voltage_min = db.Column(db.Float) current_max = db.Column(db.Float) current_min = db.Column(db.Float) dev_id = db.Column(db.Integer, db.ForeignKey("Device.dev_id")) devices = db.relationship("Device", foreign_keys=dev_id) def __repr__(self) -> str: return f"<Service {self.service_id}>"