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()