class Role(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30), unique=True)
users = db.relationship('User', back_populates='role')
permissions = db.relationship('Permission', secondary=roles_permissions, back_populates='roles')
@staticmethod
def init_role():
roles_permissions_map = {
'Locked': ['FOLLOW', 'COLLECT'],
'User': ['FOLLOW', 'COLLECT','COMMENT','UPLOAD'],
'Moderator':['FOLLOW','COLLECT','COMMENT','UPLOAD','MODERATE'],
'Administrator':['FOLLOW','COLLECT','COMMENT','UPLOAD','MODERATE','ADMINISTER']
}
for role_name in roles_permissions_map:
role = Role.query.filter_by(name=role_name).first()
if role is None:
role = Role(name=role_name)
db.session.add(role)
role.permissions = []
for permission_name in roles_permissions_map[role_name]:
permission = Permission.query.filter_by(name=permission_name).first()
if permission is None:
permission = Permission(name=permission_name)
db.session.add(permission)
role.permissions.append(permission)
db.session.commit()
class Follow(db.Model):
follower_id = db.Column(db.Integer, db.ForeignKey('user.id'), primary_key=True)
followed_id = db.Column(db.Integer, db.ForeignKey('user.id'), primary_key=True)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
follower = db.relationship('User', foreign_keys=[follower_id], back_populates='following', lazy='joined')
followed = db.relationship('User', foreign_keys=[followed_id], back_populates='followers', lazy='joined')
class Notification(db.Model):
id = db.Column(db.Integer, primary_key=True)
message = db.Column(db.Text, nullable=False)
is_read = db.Column(db.Boolean, default=False)
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
receiver_id = db.Column(db.Integer, db.ForeignKey('user.id'))
receiver = db.relationship('User', back_populates='notifications')
class PostComment(db.Model):
__tablename__ = 'post_comment'
id = db.Column(db.Integer, primary_key=True)
body = db.Column(db.Text)
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
flag = db.Column(db.Integer, default=0) # 举报次数
replied_id = db.Column(db.Integer, db.ForeignKey('post_comment.id'))
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
post_id = db.Column(db.Integer, db.ForeignKey('post.id'))
post = db.relationship('Post', back_populates='comments')
author = db.relationship('User', back_populates='post_comments')
replies = db.relationship('PostComment',
back_populates='replied',
cascade='all')
replied = db.relationship('PostComment',
back_populates='replies',
remote_side=[id])
#链接
# class Link(db.Model):
# id = db.Column(db.Integer, primary_key=True)
# name = db.Column(db.String(30))
# url = db.Column(db.String(255))
class Photo(db.Model):
id = db.Column(db.Integer, primary_key=True)
description = db.Column(db.String(500))
filename = db.Column(db.String(64))
filename_s = db.Column(db.String(64)) #小图
filename_m = db.Column(db.String(64)) #中图
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
author = db.relationship('User', back_populates='photos')
flag = db.Column(db.Integer, default=0) #被举报次数
can_comment = db.Column(db.Boolean, default=True)
comments = db.relationship('PhotoComment', back_populates='photo', cascade='all')
tags = db.relationship('Tag', secondary=tagging, back_populates='photos')
collectors = db.relationship('Collect', back_populates='collected', cascade='all')
class Category(db.Model):
#联合唯一索引
__table_args__ = (UniqueConstraint('name',
'author_id',
name='author_category_unique'), )
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
name = db.Column(db.String(30))
posts = db.relationship('Post', back_populates='category')
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
author = db.relationship('User', back_populates='categories')
def delete(self):
default_category = Category.query.get(1)
posts = self.posts[:]
for post in posts:
post.category = default_category
db.session.delete(self)
db.session.commit()
class PhotoComment(db.Model):
__tablename__ = 'photo_comment'
id = db.Column(db.Integer, primary_key=True)
body = db.Column(db.Text)
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
flag = db.Column(db.Integer, default=0) #举报次数
replied_id = db.Column(db.Integer, db.ForeignKey('photo_comment.id'))
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
photo_id = db.Column(db.Integer, db.ForeignKey('photo.id'))
photo = db.relationship('Photo', back_populates='comments')
author = db.relationship('User', back_populates='photo_comments')
replies = db.relationship('PhotoComment', back_populates='replied', cascade='all')
replied = db.relationship('PhotoComment', back_populates='replies', remote_side=[id])
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(60))
body = db.Column(db.Text)
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
can_comment = db.Column(db.Boolean, default=True)
# slug = db.Column(db.String(60))
flag = db.Column(db.Integer, default=0) #被举报次数
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
author = db.relationship('User', back_populates='posts')
category_id = db.Column(db.Integer, db.ForeignKey('category.id'))
category = db.relationship('Category', back_populates='posts')
comments = db.relationship('PostComment',
back_populates='post',
cascade='all, delete-orphan')
class Tag(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), index=True, unique=True)
photos = db.relationship('Photo', secondary=tagging, back_populates='tags')
from flask import current_app
from datetime import datetime
import os
from sites.extensions import whooshee, db
from sites.models import *
# class Item(db.Model):
# id = db.Column(db.Integer, primary_key=True)
# item_key = db.Column(db.Integer) #[(0,Photo),(1,Post)]
# item_id = db.Column(db.Integer)
tagging = db.Table('tagging',
db.Column('photo_id',db.Integer, db.ForeignKey('photo.id')),
db.Column('tag_id', db.Integer, db.ForeignKey('tag.id'))
)
@whooshee.register_model('description')
class Photo(db.Model):
id = db.Column(db.Integer, primary_key=True)
description = db.Column(db.String(500))
filename = db.Column(db.String(64))
filename_s = db.Column(db.String(64)) #小图
filename_m = db.Column(db.String(64)) #中图
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
author = db.relationship('User', back_populates='photos')
class User(db.Model, UserMixin):
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True, index=True)
email = db.Column(db.String(254), unique=True, index=True)
password_hash = db.Column(db.String(128))
name = db.Column(db.String(30))
website = db.Column(db.String(255))
bio = db.Column(db.String(120))
location = db.Column(db.String(50))
member_since = db.Column(db.DateTime, default=datetime.utcnow)
avatar_s = db.Column(db.String(64))
avatar_m = db.Column(db.String(64))
avatar_l = db.Column(db.String(64))
avatar_raw = db.Column(db.String(64)) #用户上传的头像
#消息开关
receive_comment_notification = db.Column(db.Boolean, default=True)
receive_follow_notification = db.Column(db.Boolean, default=True)
receive_collect_notification = db.Column(db.Boolean, default=True)
#隐私设置
public_collections = db.Column(db.Boolean, default=True)
confirmed = db.Column(db.Boolean, default=False) #是否邮箱验证
locked = db.Column(db.Boolean, default=False) #用户锁定
active = db.Column(db.Boolean, default=True) #封禁用户
role_id = db.Column(db.Integer, db.ForeignKey('role.id'))
role = db.relationship('Role', back_populates='users')
photos = db.relationship('Photo', back_populates='author', cascade='all') #级联设为all，用户被删除，相应图片全删除
photo_comments = db.relationship('PhotoComment', back_populates='author', cascade='all')
posts = db.relationship('Post', back_populates='author', cascade='all')
post_comments = db.relationship('PostComment', back_populates='author', cascade='all')
categories = db.relationship('Category', back_populates='author', cascade='all')
collections = db.relationship('Collect', back_populates='collector', cascade='all')
following = db.relationship('Follow', foreign_keys=[Follow.follower_id], back_populates='follower',
lazy='dynamic', cascade='all')
followers = db.relationship('Follow', foreign_keys=[Follow.followed_id], back_populates='followed',
lazy='dynamic', cascade='all')
notifications = db.relationship('Notification', back_populates='receiver', cascade='all')
def __init__(self, **kwargs):
super(User, self).__init__(**kwargs)
self.generate_avatar()
self.follow(self) #follow self
self.set_role()
self.default_category()
#创建默认文章分类
def default_category(self):
category = Category(name='默认分类',author=self)
db.session.add(category)
db.session.commit()
def generate_avatar(self):
avatar = Identicon()
filenames = avatar.generate(text=self.username)
self.avatar_s = filenames[0]
self.avatar_m = filenames[1]
self.avatar_l = filenames[2]
db.session.commit()
def set_role(self):
if self.role is None:
if self.email == current_app.config['ALBUMY_ADMIN_EMAIL']:
self.role = Role.query.filter_by(name='Administrator').first()
else:
self.role = Role.query.filter_by(name='User').first()
db.session.commit()
@property
def is_admin(self):
return self.role.name == 'Administrator'
def can(self, permission_name):
permission = Permission.query.filter_by(name=permission_name).first()
return permission is not None and self.role is not None and permission in self.role.permissions
def set_password(self, password):
self.password_hash = generate_password_hash(password)
def validate_password(self, password):
return check_password_hash(self.password_hash, password)
#收藏和取消收藏
def collect(self, photo):
if not self.is_collecting(photo):
collect = Collect(collector=self, collected=photo)
db.session.add(collect)
db.session.commit()
def uncollect(self, photo):
collect = Collect.query.with_parent(self).filter_by(collected_id=photo.id).first()
if collect:
db.session.delete(collect)
db.session.commit()
def is_collecting(self, photo):
return Collect.query.with_parent(self).filter_by(collected_id=photo.id).first() is not None
#关注和取消关注
def follow(self, user):
if not self.is_following(user):
follow = Follow(follower=self, followed=user)
db.session.add(follow)
db.session.commit()
def unfollow(self, user):
follow = self.following.filter_by(followed_id=user.id).first()
if follow:
db.session.delete(follow)
db.session.commit()
def is_following(self, user):
if user.id is None: #在实例化的时候，还未提交数据库，user.id为none,为了避免构造方法报错，就直接返回False
return False
return self.following.filter_by(followed_id=user.id).first() is not None
def is_followed_by(self, user):
return self.followers.filter_by(follower_id=user.id).first() is not None
def followed_photos(self):
from sites.models.photo import Photo
return Photo.query.join(Follow, Follow.followed_id == Photo.author_id).filter(Follow.follower_id == self.id)
#用户锁定
def lock(self):
self.locked = True
self.role = Role.query.filter_by(name='Locked').first()
db.session.commit()
def unlock(self):
self.locked = False
self.role = Role.query.filter_by(name='User').first()
db.session.commit()
#用户封禁
@property
def is_active(self):
return self.active
def block(self):
self.active = False
db.session.commit()
def unblock(self):
self.active = True
db.session.commit()
class Collect(db.Model):
collector_id = db.Column(db.Integer, db.ForeignKey('user.id'), primary_key=True)
collected_id = db.Column(db.Integer, db.ForeignKey('photo.id'), primary_key=True)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
collector = db.relationship('User', back_populates='collections', lazy='joined') #lazy设为joined或者False,表示用连结查询，提高性能，减少一次查询
collected = db.relationship('Photo', back_populates='collectors', lazy='joined')
class Permission(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30), unique=True)
roles = db.relationship('Role', secondary=roles_permissions, back_populates='permissions')
def is_active(self):
return self.active
def block(self):
self.active = False
db.session.commit()
def unblock(self):
self.active = True
db.session.commit()
#权限管理(RBAC)
roles_permissions = db.Table('roles_permissions',
db.Column('role_id', db.Integer, db.ForeignKey('role.id')),
db.Column('permission_id',db.Integer,db.ForeignKey('permission.id'))
)
class Permission(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30), unique=True)
roles = db.relationship('Role', secondary=roles_permissions, back_populates='permissions')
class Role(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30), unique=True)
users = db.relationship('User', back_populates='role')
permissions = db.relationship('Permission', secondary=roles_permissions, back_populates='roles')
@staticmethod