class Reply(BaseModel, db.Model):
__tablename__ = 'b_reply'
id = db.Column(db.Integer, primary_key=True) # 回复的id
reply_from_user = db.Column(db.String(32), nullable=False) # 谁回复的
reply_to_user = db.Column(db.String(32), nullable=False) # 回复给谁的
reply_content = db.Column(db.Text, nullable=False) # 回复的内容
reply_status = db.Column(db.Integer,
default=0) # 回复是否通过审核 -1不可用 0:审核中 1:审核通过
reply_comment_id = db.Column(db.Integer,
db.ForeignKey('b_comments.id'),
nullable=False) # 当前回复属于的评论id
def to_dict(self):
user = User.query.filter_by(nick_name=self.reply_from_user).first()
res_dict = {
"id": self.id,
"reply_from_user": self.reply_from_user,
"reply_to_user": self.reply_to_user,
"reply_content": self.reply_content,
"reply_comment_id": self.reply_comment_id,
"reply_create_time": self.create_time,
"reply_update_time": self.update_time,
"reply_user_is_admin": user.is_admin,
"reply_user_avatar_url": user.avatar_url,
}
return res_dict
class MessageBoard(BaseModel, db.Model):
__tablename__ = 'b_board'
id = db.Column(db.Integer, primary_key=True) # 留言板id
board_user = db.Column(db.String(32), nullable=False) # 留言用户
board_desc = db.Column(db.Text, nullable=False) # 留言内容
board_status = db.Column(db.Integer, nullable=False,
default=0) # 留言状态 -1不可用 0:审核中 1:审核通过
board_email = db.Column(db.String(50), nullable=False) # 留言回复邮箱
def to_dict(self):
res_dict = {
"id": self.id,
"board_user": self.board_user,
"board_desc": self.board_desc,
"board_status": self.board_status,
"board_create_time": self.create_time,
"board_update_time": self.update_time,
"board_email": self.board_email,
}
return res_dict
class User(BaseModel, db.Model, UserMixin):
"""
UserMixin表示通过认证的用户
is_authenticated 表示用户已通过认证 返回True 否则False
is_active 表示允许用户登陆 返回True 否则False
is_anonymous 表示如果当前未用户登陆(匿名用户) 返回True 否则False
get_id() 以unicode形式返回用户唯一标识
"""
__tablename__ = 'b_users'
id = db.Column(db.Integer, primary_key=True) # 用户id
nick_name = db.Column(db.String(32), nullable=False) # 用户名
password_hash = db.Column(db.String(128), nullable=False) # 用户密码
mobile = db.Column(db.String(11), unique=True) # 手机号码
email = db.Column(db.String(64), unique=True, nullable=True) # 邮箱
desc = db.Column(db.Text) # 个人简介
location = db.Column(db.String(128)) # 地址
avatar_url = db.Column(db.String(256)) # 用户头像路径
is_admin = db.Column(db.Boolean, default=False) # 是否为管理员
last_login_time = db.Column(db.DateTime,
default=datetime.utcnow) # 最后一次登陆时间
is_delete = db.Column(db.Integer, default=1) # 用户是否被删除 1正常 0被删除
gender = db.Column(
db.Enum(
'MAN', # 男
'WOMAN' # 女
),
default='MAN')
@property
def password(self):
raise AttributeError(u'该属性不可读')
@password.setter
def password(self, value):
"""
generate_password_hash（password，method='pbkdf2：sha256'，salt_length=8）
method指定计算散列值的方法
salt_length 指定盐长度
"""
self.password_hash = generate_password_hash(value)
def check_password(self, password):
"""
接收散列值 和 密码作比较 返回布尔类型
check_password_hash（pwhash，password）
"""
return check_password_hash(self.password_hash, password)
def to_dict(self):
res_dict = {
"id": self.id,
"nick_name": self.nick_name,
"email": self.email,
"desc": self.desc,
"avatar_url": self.avatar_url,
"gender": self.gender,
"is_admin": self.is_admin,
}
return res_dict
class Journey(BaseModel, db.Model):
__tablename__ = 'b_journey'
id = db.Column(db.Integer, primary_key=True) # 历程id
journey_title = db.Column(db.String(32), nullable=False) # 历程标题
journey_desc = db.Column(db.Text, nullable=False) # 历程详情
journey_time = db.Column(db.DateTime, default=datetime.utcnow) # 历程时间
def to_dict(self):
res_dict = {
"id": self.id,
"journey_title": self.journey_title,
"journey_desc": self.journey_desc,
"journey_time": self.journey_time
}
return res_dict
class Category(BaseModel, db.Model):
__tablename__ = 'b_category'
id = db.Column(db.Integer, primary_key=True) # 分类编号
cg_name = db.Column(db.String(64), nullable=False, unique=True) # 分类名称
cg_posts = db.relationship('Post',
back_populates='post_category') # 分类下的文章
def to_dict(self):
res_dict = {
"id": self.id,
"cg_name": self.cg_name,
"cg_posts_count": len(self.cg_posts) if self.cg_posts else 0
}
return res_dict
class Post(BaseModel, db.Model):
__tablename__ = 'b_posts'
id = db.Column(db.Integer, primary_key=True) # 文章编号
post_title = db.Column(db.String(256), nullable=False) # 文章标题
post_user_id = db.Column(db.Integer, nullable=False) # 创建文章用户
post_digest = db.Column(db.String(512), nullable=True) # 文章简介
post_content = db.Column(db.Text, nullable=False) # 文章内容
post_clicks = db.Column(db.Integer, default=0) # 点击量
post_like_num = db.Column(db.Integer, default=0) # 点赞数量
post_index_image_url = db.Column(db.String(256)) # 主页面列表图片地址
post_status = db.Column(db.Integer, default=1) # 文章状态
post_can_comment = db.Column(db.Integer, default=1) # 当前文章是否可以被评论
post_comments = db.relationship('Comment',
backref='comment_post') # 当前文章的评论
post_category = db.relationship('Category', back_populates='cg_posts')
post_category_id = db.Column(db.Integer,
db.ForeignKey('b_category.id'),
nullable=False) # 文章类型
def get_comment_length(self):
comments = []
if self.post_comments is not []:
for comment in self.post_comments:
if comment.comment_status == 1:
comments.append(comment)
return len(comments)
def to_dict(self):
res_dict = {
"id":
self.id,
"post_title":
self.post_title,
"post_user_id":
self.post_user_id,
"post_digest":
self.post_digest if self.post_digest else "",
"post_clicks":
self.post_clicks,
"post_like_num":
self.post_like_num,
"post_index_image_url":
self.post_index_image_url if self.post_index_image_url else "",
"post_category":
self.post_category.to_dict() if self.post_category else None,
"post_comments_count":
self.get_comment_length(),
"post_create_time":
self.create_time,
"post_update_time":
self.update_time,
}
return res_dict
def to_dict_details(self):
res_dict = {
"id":
self.id,
"post_title":
self.post_title,
"post_user_id":
self.post_user_id,
"post_content":
self.post_content,
"post_clicks":
self.post_clicks,
"post_like_num":
self.post_like_num,
"post_can_comment":
self.post_can_comment,
"post_create_time":
self.create_time,
"post_category":
self.post_category.to_dict() if self.post_category else None,
"post_comments_count":
self.get_comment_length(),
}
return res_dict