class Company(db.Model, TimestampMixin): __tablename__ = 'companies' id = db.Column(db.Integer, primary_key=True) names = db.relationship('CompanyName', backref='companies', lazy='joined') tags = db.relationship('CompanyTag', backref='companies', lazy='joined')
class CompanyTag(db.Model, TimestampMixin): __tablename__ = 'company_tags' __table_args__ = (db.UniqueConstraint('company_id', 'tag_id'), ) id = db.Column(db.Integer, primary_key=True) company_id = db.Column(db.Integer, db.ForeignKey('companies.id'), nullable=False) tag_id = db.Column(db.Integer, db.ForeignKey('tags.id'), nullable=False) tag = db.relationship(Tag, lazy='joined')
class Hosts(db.Model): __tablename__ = 'hosts' id = db.Column(db.Integer, primary_key=True) url = db.Column(db.String, unique=False, nullable=True) muted = db.Column(db.Boolean, default=False, nullable=False) apikey_id = db.Column(db.Integer, db.ForeignKey('apikey.id'), nullable=True) stats = db.relationship('Stats', backref='hosts', lazy=True) def __init__(self, url, apikey_id): self.url = url self.apikey_id = apikey_id
class ApiKey(db.Model): __tablename__ = 'apikey' id = db.Column(db.Integer, primary_key=True) apikey_hash = db.Column(db.String, unique=True, nullable=True) email = db.Column(db.String, unique=True, nullable=True) hosts = db.relationship('Hosts', backref='apikey', lazy=True) key = db.Column(db.String, unique=True, nullable=False) chat_id = db.Column(db.Integer, unique=True, nullable=True) def check_apikey(self, api, apikey): return verify_apikey(api, apikey_hash=self.apikey_hash, apikey=apikey) def __init__(self, apikey_hash, key, email, chat_id): self.apikey_hash = apikey_hash self.key = key self.email = email self.chat_id = chat_id
class Author(BaseModel): """ Author database model. Used for storing author profile and personal details. """ __tablename__ = 'author' id = db.Column(db.Integer, primary_key=True, autoincrement=True) first_name = db.Column(db.String(255)) last_name = db.Column(db.String(255)) avatar = db.Column(db.String(255), nullable=True) books = db.relationship('Book', secondary=books, lazy='subquery', backref=db.backref('authors', lazy=True)) birthday = db.Column(db.Date, nullable=False) user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False) @property def name(self): """ Author full name """ first_name = f'{self.first_name} ' if self.first_name is not None else '' last_name = self.last_name if self.last_name is not None else '' return f'{first_name}{last_name}'.strip() def __repr__(self): """ Author representation """ return self.name
class User(UserMixin, db.Model): __tablename__ = 'users' id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(32), unique=True) password_hash = db.Column(db.String(128)) email = db.Column(db.String(64), unique=True) confirmed = db.Column(db.Boolean, default=False) # 头像 icon = db.Column(db.String(64), default='default.jpg') # 在关联模型中添加反向引用 # 参数1:关联的模型 # backref:反向引用字段名 # lazy:加载时机,dynamic不加载数据,但是提供数据的查询 # 一对一关系添加:uselist = False posts = db.relationship('Posts', backref='user', lazy='dynamic') # 添加收藏功能 favorites = db.relationship('Posts', secondary='collections', backref=db.backref('users', lazy='dynamic'), lazy='dynamic') # 保护密码属性 @property def password(self): raise AttributeError('密码是不可读属性') # 设置密码时,保存加密后的hash值 @password.setter def password(self, password): self.password_hash = generate_password_hash(password) # 密码校验,正确返回True,错误返回False def verify_password(self, password): return check_password_hash(self.password_hash, password) # 生成账户激活的token def generate_activate_token(self, expires_in=3600): s = Serializer(current_app.config['SECRET_KEY'], expires_in=expires_in) return s.dumps({'id': self.id}) # 校验账户激活的token @staticmethod def check_activate_token(token): s = Serializer(current_app.config['SECRET_KEY']) try: data = s.loads(token) except: return False u = User.query.get(data.get('id')) if not u: # flash('用户不存在') return False if not u.confirmed: # 账户没有激活 u.confirmed = True db.session.add(u) return True # 判断该用户是否收藏了指定的博客 #后边调用的时候 #通过当前登录用户.方法 此刻应该在模型中写 def is_favorite(self, pid): # 获取所有收藏的博客 favorites = self.favorites.all() posts = list(filter(lambda p: p.id == pid, favorites)) if len(posts) > 0: return True return False # 添加收藏 def add_favorite(self, pid): p = Posts.query.get(pid) #根据id 获取帖子对象 self.favorites.append(p) db.session.commit() # 取消收藏 def del_favorite(self, pid): p = Posts.query.get(pid) self.favorites.remove(p) db.session.commit()