class Tag(db.Model, TimestampMixin):
__tablename__ = 'tags'
__table_args__ = (db.UniqueConstraint('tag'), )
id = db.Column(db.Integer, primary_key=True)
tag = db.Column(db.String(80), nullable=False)
lang = db.Column(db.String(2), nullable=False)
class BaseModel(db.Model):
"""基类"""
__abstract__ = True
id = db.Column(db.Integer, primary_key=True, autoincrement=True, comment='主键')
create_time = db.Column(db.DateTime, default=datetime.datetime.now, comment='创建时间')
update_time = db.Column(db.DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now, comment='修改时间')
class Posts(db.Model):
__tablename__ = 'posts'
id = db.Column(db.Integer, primary_key=True)
rid = db.Column(db.Integer, index=True, default=0)
content = db.Column(db.Text)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
# 添加关联外键: '表名.字段'
uid = db.Column(db.Integer, db.ForeignKey('users.id'))
class User(BaseModel, ModelMixin):
__tablename__ = 'user'
nickname = db.Column(db.String(32), unique=True, comment='用户名')
email = db.Column(db.String(50), unique=True, comment='邮箱')
password = db.Column(db.String(32), nullable=True, comment='密码')
secret_token = db.Column(db.String(32), nullable=True, comment='访问密匙')
salt = db.Column(db.String(32), nullable=True, comment='密码盐')
class User(db.Model):
__tablename__ = "users"
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(32), nullable=False, unique=True)
password_hash = db.Column(db.String(128), nullable=False)
def __repr__(self):
return self.username
class SearchCompany(db.Model, TimestampMixin):
__tablename__ = 'search_companies'
id = db.Column(db.Integer, primary_key=True)
company_id = db.Column(db.Integer,
db.ForeignKey('companies.id'),
nullable=False)
initial_index = db.Column(db.String(255), nullable=False)
phoneme_index = db.Column(db.String(255), nullable=False)
class CompanyName(db.Model, TimestampMixin):
__tablename__ = 'company_names'
__table_args__ = (db.UniqueConstraint('company_id', 'lang'), )
id = db.Column(db.Integer, primary_key=True)
company_id = db.Column(db.Integer,
db.ForeignKey('companies.id'),
nullable=False)
name = db.Column(db.String(80), nullable=False)
lang = db.Column(db.String(2), nullable=False)
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 BaseModel(db.Model):
"""
Abstract base data model to be extended in all models.
Defines audit fields for a model.
"""
__abstract__ = True
created_at = db.Column(db.DateTime, default=db.func.now())
updated_at = db.Column(db.DateTime,
default=db.func.now(),
onupdate=db.func.now())
class Stats(db.Model):
__tablename__ = 'stats'
id = db.Column(db.Integer, primary_key=True)
code = db.Column(db.String, unique=False, nullable=True)
response_time = db.Column(db.Integer, unique=False, nullable=True)
time = db.Column(db.DateTime, index=False, unique=False, nullable=False)
host_id = db.Column(db.Integer, db.ForeignKey('hosts.id'), nullable=True)
def __init__(self, code, response_time, time, host_id):
self.code = code
self.response_time = response_time
self.time = time
self.host_id = host_id
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 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 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 Book(BaseModel):
"""
Book database model.
Used for storing authors books details.
"""
__tablename__ = 'book'
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
title = db.Column(db.String(255))
year = db.Column(db.Integer)
def __repr__(self):
"""
Book representation
"""
return f"<Book {self.title}>"
class User(BaseModel):
"""
User database model
"""
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
first_name = db.Column(db.String(255), nullable=False)
last_name = db.Column(db.String(255), nullable=False)
email = db.Column(db.String(255), unique=True, nullable=False)
user_type = db.Column(db.Enum(UserType), default='AUTHOR')
@classmethod
def find_by_email(cls, email):
"""
Find user by email
"""
return cls.query.filter_by(email=email).first()
class TaskLog(db.Model):
__tablename__ = 'taskLog'
id = db.Column(db.Integer, primary_key=True)
taskId = db.Column(db.Integer, unique=False)
startTime = db.Column(db.DateTime, unique=False)
endTime = db.Column(db.DateTime, unique=False)
status = db.Column(db.Integer, unique=False)
code = db.Column(db.Integer, unique=False)
result = db.Column(db.String(1000), unique=False)
class User(db.Model):
id = db.Column(db.INTEGER, primary_key=True)
user_code = db.Column(db.String(100))
# 根据生成密码的规则，password代表的是密码hash，并非明文密码
password = db.Column(db.String(100))
@property
# property包装器表明将函数直接当作属性使用
# 必须如此声明，此可以使用_password包装器
# 以及使用setter和getter方法
def _password(self):
return self.password
@_password.setter
def _password(self, password):
self.password = generate_password_hash(password)
# self.save()
def check_password(self, password):
return check_password_hash(self.password, password)
@classmethod
def create(cls, **kwargs):
try:
obj = cls(user_code=kwargs.get('user_code'))
if kwargs.get('password'):
obj.password = kwargs.get('password')
else:
# 使用=，调用的是_password.setter方法
obj._password = '******'
db.session.add(obj)
db.session.commit()
return obj
except Exception as e:
logging.debug(e)
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 CreatedAtMixin(db.Model):
__abstract__ = True
created_at = db.Column(db.TIMESTAMP,
default=dt.datetime.utcnow,
nullable=False)
class UpdatedAtMixin(db.Model):
__abstract__ = True
updated_at = db.Column(db.TIMESTAMP, onupdate=dt.datetime.utcnow)
class User(db.Model):
id = db.Column(db.Integer)
user = db.Column(db.String(24))
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()
"""
Author related model
"""
from apps.extensions import db
from .audit import BaseModel
books = db.Table(
'author_book',
db.Column('author_id',
db.Integer,
db.ForeignKey('author.id'),
primary_key=True),
db.Column('book_id',
db.Integer,
db.ForeignKey('book.id'),
primary_key=True))
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)
class Image(BaseModel, ModelMixin):
"""用户上传图片"""
__tablename__ = 'image'
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), comment='上传的用户', default=None)
default_show = db.Column(db.String(200), comment='默认上传显示的地址')
raw = db.Column(db.Text)
from .users import User
from .posts import Posts
from apps.extensions import db
# 添加用户与帖子的收藏中间关联表 表名.字段
collections = db.Table(
'collections', db.Column('user_id', db.Integer, db.ForeignKey('users.id')),
db.Column('posts_id', db.Integer, db.ForeignKey('posts.id')))
#一个用户可以收藏多个帖子
#一个帖子可以被多个用户收藏
#多对多我们是通过 中间表来实现
class Task(db.Model):
__tablename__ = 'task'
id = db.Column(db.Integer, primary_key=True)
corn = db.Column(db.String(40), unique=False)
name = db.Column(db.String(300), unique=True)
desc = db.Column(db.String(500), unique=False)
created = db.Column(db.DateTime, default=datetime.now)
url = db.Column(db.String(300), unique=False)
method = db.Column(db.String(4), unique=False)
params = db.Column(db.String(300), unique=False)
headers = db.Column(db.String(300), unique=False)
success = db.Column(db.String(300), unique=False)
dependId = db.Column(db.Integer, unique=False)
