class User(db.Model, UserMixin): __tablename__ = 'users' 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)) location = db.Column(db.String(64)) about_me = db.Column(db.Text()) member_since = db.Column(db.DateTime(), default=datetime.datetime.utcnow) last_seen = db.Column(db.DateTime(), default=datetime.datetime.utcnow) role_id = db.Column(db.Integer, db.ForeignKey('roles.id')) role = db.relationship('Role', back_populates='users') depart_id = db.Column(db.Integer, db.ForeignKey('departs.id')) depart = db.relationship('Depart', back_populates='users') photos = db.relationship('Photo', back_populates='author', cascade='all') advice = db.relationship('Advice', back_populates='author', cascade='all') comments = db.relationship('Comment', back_populates='author', cascade='all') def __init__(self, **kwargs): super(User, self).__init__(**kwargs) self.set_role_by_role_name() self.set_depart_by_depart_name() def __repr__(self): return f'<User {self.username}>' # @property # def password(self): # raise AttributeError('密码不具有可读属性') def set_password(self, password): self.password_hash = generate_password_hash(password) def set_role_by_role_name(self, role_name='User'): self.role = Role.query.filter_by(name=role_name).first() db.session.commit() def set_depart_by_depart_name(self, depart_name='本科生第一支部'): self.depart = Depart.query.filter_by(name=depart_name).first() db.session.commit() def validate_password(self, password): return check_password_hash(self.password_hash, password) @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 ping(self): self.last_seen = datetime.datetime.utcnow() db.session.commit()
class Photo(db.Model): __tablename__ = 'photos' id = db.Column(db.Integer, primary_key=True) description = db.Column(db.Text) filesize = db.Column(db.String(64)) filename = db.Column(db.String(512)) filename_m = db.Column(db.String(512)) filename_s = db.Column(db.String(512)) timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True) public_status = db.Column(db.Integer, default=0) author_id = db.Column(db.Integer, db.ForeignKey('users.id')) author = db.relationship('User', back_populates='photos') task_id = db.Column(db.Integer, db.ForeignKey('tasks.id')) task = db.relationship('Task', back_populates='photos') def __repr__(self): return f'<Photo {self.id}>' def set_task_by_id(self, task_id=1): self.task = Task.query.filter_by(id=task_id).first() db.session.commit() def set_task_by_name_third(self, name_third): self.task = Task.query.filter_by(name_third=name_third).first() db.session.commit()
class Comment(db.Model): __tablename__ = 'comments' id = db.Column(db.Integer, primary_key=True) body = db.Column(db.Text) timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True) advice_id = db.Column(db.Integer, db.ForeignKey('advice.id')) advice = db.relationship('Advice', back_populates='comments') author_id = db.Column(db.Integer, db.ForeignKey('users.id')) author = db.relationship('User', back_populates='comments') replied_id = db.Column(db.Integer, db.ForeignKey('comments.id')) replies = db.relationship('Comment', back_populates='replied', cascade='all') replied = db.relationship('Comment', back_populates='replies', remote_side=[id])
class Advice(db.Model): __tablename__ = 'advice' id = db.Column(db.Integer, primary_key=True) depart_id = db.Column(db.Integer) passed_count = db.Column(db.Integer) url = db.Column(db.Text) filter = db.Column(db.Text) status = db.Column(db.Integer, default=0) timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True) author_id = db.Column(db.Integer, db.ForeignKey('users.id')) author = db.relationship('User', back_populates='advice') comments = db.relationship('Comment', back_populates='advice', cascade='all')
class User(db.Model, UserMixin): 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)) member_since = db.Column(db.DateTime, default=datetime.utcnow) role_id = db.Column(db.Integer, db.ForeignKey('role.id')) role = db.relationship('Role', back_populates='users') def __init__(self, **kwargs): super(User, self).__init__(**kwargs) self.set_role() def set_password(self, password): self.password_hash = generate_password_hash(password) def set_role(self): if self.role is None: if self.email == current_app.config['ADMIN_EMAIL']: self.role = Role.query.filter_by(name='Administrator').first() else: self.role = Role.query.filter_by(name='User').first() db.session.commit() def validate_password(self, password): return check_password_hash(self.password_hash, password) @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
from flask import current_app from flask_login import UserMixin from werkzeug.security import generate_password_hash, check_password_hash from datetime import datetime from main.plugins.extensions import db, whooshee # relationship table 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