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]
