class ChangeLogs(db.Model): """ 修改日志 """ __tablename__ = 'change_logs' id = db.Column(db.Integer, autoincrement=True, primary_key=True) author_id = db.Column(db.Integer, db.ForeignKey('user.id')) # 作者 article_id = db.Column(db.Integer, db.ForeignKey('article.id')) # 文章 modify_content = db.Column(db.String(255), nullable=False) # 修改内容 create_time = db.Column(db.DateTime, nullable=False) # 创建日期
class Article(db.Model): """ 文章表 """ __tablename__ = 'article' id = db.Column(db.Integer, autoincrement=True, primary_key=True) title = db.Column(db.String(20), nullable=False) # 文章标题 body = db.Column(db.String(255), nullable=False) # 文章内容 last_change_time = db.Column(db.DateTime, nullable=False, default=datetime.now) # 最后一次修改日期 author_id = db.Column(db.Integer, db.ForeignKey('user.id')) # 作者
class Income(db.Model): __tablename__ = "incomes" id = db.Column(db.Integer, autoincrement=True, primary_key=True) client_id = db.Column(db.Integer, db.ForeignKey("clients.id")) value = db.Column(db.Float, nullable=False) date = db.Column(db.Date, default=(datetime.utcnow() + timedelta(hours=8)).strftime( "%Y-%m-%d"), nullable=False)
class Income(db.Model): """ 收入表 """ __tablename__ = "tb_income" id = db.Column(db.Integer, autoincrement=True, primary_key=True) value = db.Column(db.Float, nullable=False) customer_id = db.Column(db.Integer, db.ForeignKey("tb_customer.id")) date = db.Column(db.Date, nullable=False)
class Order(db.Model): # 设置数据库表名 __tablename__ = 'order' id = db.Column(db.Integer, primary_key=True, autoincrement=True) status = db.Column(db.String(128)) company = db.Column(db.String(255)) create = db.Column(db.DateTime, index=True, default=datetime.utcnow) start = db.Column(db.DateTime, index=True) complete = db.Column(db.DateTime, index=True) price = db.Column(db.Float) email = db.Column(db.String(128)) wxuser_openid = db.Column(db.String(255), db.ForeignKey('wxuser.openid')) # 属于哪个用户 code = db.Column(db.String(255), index=True) payid = db.Column(db.String(255)) result = db.Column(db.String(255)) def to_dict(self): data = { 'id': self.id, 'status': self.status, 'company': self.company, 'create': self.create, 'start': self.start, 'complete': self.complete, 'price': self.price, 'email': self.email, 'code': self.code, 'payid': self.payid, 'result': self.result } return data def from_dict(self, data): for field in [ 'status', 'company', 'price', 'email', 'code', 'wxuser_openid', 'payid', 'result' ]: if field in data: setattr(self, field, data[field]) if field == "status" and data["status"] == "complete": setattr(self, "complete", datetime.now()) @staticmethod def to_collection_dict(query, page=1, per_page=10, **kwargs): # 如果当前没有任何资源时,或者前端请求的 page 越界时,都会抛出 404 错误 # 由 @bp.app_errorhandler(404) 自动处理,即响应 JSON 数据:{ error: "Not Found" } # resources = query.paginate(page, per_page) return [item.to_dict() for item in query]
class Operation(PaginatedAPIMixin, db.Model): __tablename__ = 'operations' id = db.Column(db.Integer, primary_key=True) operator_id = db.Column(db.Integer, db.ForeignKey('users.id')) describe = db.Column(db.Text) timestamp = db.Column(db.DateTime, index=True, default=datetime.now) ip = db.Column(db.Text) def from_dict(self, data): for field in ['operator_id', 'describe', 'ip']: if field in data: setattr(self, field, data[field]) def to_dict(self): data = { 'id': self.id, 'operator_name': self.operator.username, 'timestamp': self.timestamp, 'describe': self.describe, 'ip': self.ip } return data
id = db.Column(db.Integer, autoincrement=True, primary_key=True) name = db.Column(db.String(255), nullable=False) email = db.Column(db.String(255), nullable=False) users = db.relationship('User', secondary=user_customer, backref=db.backref('customers')) incomes = db.relationship("Income", backref=db.backref("customer")) # 用户-客户多对多 user_customer = db.Table( 'user_customer', db.Column('user_id', db.Integer, db.ForeignKey('tb_user.id'), primary_key=True), db.Column('customer_id', db.Integer, db.ForeignKey('tb_customer.id'), primary_key=True)) class Income(db.Model): """ 收入表 """ __tablename__ = "tb_income" id = db.Column(db.Integer, autoincrement=True, primary_key=True) value = db.Column(db.Float, nullable=False)
from datetime import datetime from app.utils.core import db from flask import url_for, current_app from app.utils.elasticsearch import add_to_index, remove_from_index, query_index, es_highlight haowen_tag = db.Table( 'haowen_tag', db.Column('haowen_id', db.Integer, db.ForeignKey('haowen.id'), primary_key=True), db.Column('tag_id', db.Integer, db.ForeignKey('tag.id'), primary_key=True)) class User(db.Model): """ 用户表 """ __tablename__ = 'user' id = db.Column(db.Integer, autoincrement=True, primary_key=True) name = db.Column(db.String(128), nullable=False, server_default="") age = db.Column(db.String(128), nullable=False, server_default="") class UserLoginMethod(db.Model): """ 用户登陆验证表 """ __tablename__ = 'user_login_method' id = db.Column(db.Integer, autoincrement=True, primary_key=True) # 用户登陆方式主键ID
from datetime import datetime, timedelta from app.utils.core import db # 用户-客户多对多 user_client = db.Table('user_client', db.Column('user_id', db.Integer, db.ForeignKey('users.id'), primary_key=True), db.Column('client_id', db.Integer, db.ForeignKey('clients.id'), primary_key=True)) class User(db.Model): __tablename__ = 'users' id = db.Column(db.Integer, autoincrement=True, primary_key=True) name = db.Column(db.String(20), nullable=False, unique=True) email = db.Column(db.String(50), nullable=False) password = db.Column(db.String(255), nullable=False) clients = db.relationship('Client', secondary=user_client, cascade="all,delete", backref="users") class Client(db.Model): __tablename__ = "clients" id = db.Column(db.Integer, autoincrement=True, primary_key=True) name = db.Column(db.String(50), nullable=False, unique=True)
class User(PaginatedAPIMixin, db.Model): # 设置数据库表名,Post模型中的外键 user_id 会引用 users.id __tablename__ = 'users' id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(64), index=True, unique=True) name = db.Column(db.String(64), index=True) email = db.Column(db.String(120), index=True) password_hash = db.Column(db.String(128)) # 不保存原始密码 # about_me = db.Column(db.Text()) member_since = db.Column(db.DateTime(), default=datetime.now) # onupdate=datetime.now 修改任何信息,自动修改时间 # last_seen = db.Column(db.DateTime(), default=datetime.now, onupdate=datetime.now) last_seen = db.Column(db.DateTime(), default=datetime.now) department_id = db.Column(db.Integer, db.ForeignKey('departments.id')) remark = db.Column(db.Text()) operation = db.relationship('Operation', backref='operator', lazy='dynamic', cascade='all, delete-orphan') def __repr__(self): return '<User {}>'.format(self.username) def set_password(self, password): '''设置用户密码,保存为 Hash 值''' self.password_hash = generate_password_hash(password) def check_password(self, password): '''验证密码与保存的 Hash 值是否匹配''' return check_password_hash(self.password_hash, password) def avatar(self, size): '''用户头像''' digest = md5(self.email.lower().encode('utf-8')).hexdigest() return 'https://www.gravatar.com/avatar/{}?d=identicon&s={}'.format( digest, size) def to_dict(self, include_email=False): data = { 'id': self.id, 'username': self.username, 'name': self.name, 'email': self.email, 'remark': self.remark, 'department_id': self.department_id, 'department_name': self.department.name, # 'about_me': self.about_me, 'member_since': self.member_since.isoformat() + 'Z', 'last_seen': self.last_seen.isoformat() + 'Z' } return data def from_dict(self, data, new_user=False): for field in ['username', 'email', 'name', 'department_id', 'remark']: if field in data: setattr(self, field, data[field]) if 'password' in data: self.set_password(data['password']) def ping(self): '''更新用户的最后访问时间''' self.last_seen = datetime.now() db.session.add(self) def get(self, field): return getattr(self, field) def get_jwt(self, expires_in=3600): '''用户登录后,发放有效的 JWT''' now = datetime.utcnow() payload = { 'user_id': self.id, 'user_name': self.username if self.username else self.name, 'name': self.name if self.name else self.username, 'exp': now + timedelta(seconds=expires_in), 'iat': now } return jwt.encode(payload, current_app.config['SECRET_KEY'], algorithm='HS256').decode('utf-8') @staticmethod def verify_jwt(token): '''验证 JWT 的有效性''' try: payload = jwt.decode(token, current_app.config['SECRET_KEY'], algorithms=['HS256']) except (jwt.exceptions.ExpiredSignatureError, jwt.exceptions.InvalidSignatureError, jwt.exceptions.DecodeError) as e: # Token过期,或被人修改,那么签名验证也会失败 return None return User.query.get(payload.get('user_id')) # def get_permissions(self): # self.permissions = Department.query.get(self.department_id).get("permissions") def can(self, operate_permission): #这个方法用来传入一个权限来核实用户是否有这个权限,返回bool值,检查permissions要求的权限角色是否允许 d = Department.query.get(self.department_id) if not d.get("active"): return False self.permissions = d.get("permissions") self.permissions = json.loads(self.permissions) (op_key, op_val), = operate_permission.items() return op_val in self.permissions[op_key]