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 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)
email = db.Column(db.String(50), nullable=False)
trade = db.Column(db.Boolean, default=False)
incomes = db.relationship('Income', cascade="all,delete", backref="client")
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 UserLoginMethod(db.Model):
"""
用户登陆验证表
"""
__tablename__ = 'user_login_method'
id = db.Column(db.Integer, autoincrement=True, primary_key=True) # 用户登陆方式主键ID
user_id = db.Column(db.Integer, nullable=False) # 用户主键ID
login_method = db.Column(db.String(36), nullable=False) # 用户登陆方式，WX微信，P手机
identification = db.Column(db.String(36), nullable=False) # 用户登陆标识，微信ID或手机号
access_code = db.Column(db.String(36), nullable=True) # 用户登陆通行码，密码或token
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 User(db.Model):
"""
用户表
"""
__tablename__ = 'tb_user'
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
username = db.Column(db.String(20), nullable=False)
password = db.Column(db.String(255), nullable=False)
phone = db.Column(db.String(255), nullable=False)
email = db.Column(db.String(255), nullable=False)
address = db.Column(db.String(255), nullable=False)
class Customer(db.Model):
"""
客户表
"""
__tablename__ = "tb_customer"
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"))
class Teacher(db.Model):
"""
教师表
"""
__tablename__ = 'ht_teacher'
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
name = db.Column(db.String(40), nullable=False) # 姓名
remark = db.Column(db.String(255), nullable=True)
update_time = db.Column(db.DateTime, nullable=True)
def list(self):
return {
"id": self.id,
"name": self.name
}
class Department(PaginatedAPIMixin, db.Model):
__tablename__ = 'departments'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(120), index=True, unique=True)
describe = db.Column(db.Text)
timestamp = db.Column(db.DateTime, index=True, default=datetime.now)
members = db.relationship('User',
backref='department',
lazy='dynamic',
cascade='all, delete-orphan')
active = db.Column(db.Boolean, default=True)
permissions = db.Column(db.Text)
def __repr__(self):
return '<Department {}>'.format(self.id)
def to_dict(self):
data = {
'id': self.id,
'name': self.name,
'timestamp': self.timestamp,
'describe': self.describe,
'members_count': self.members.count(),
'active': self.active,
'permissions': self.permissions
}
return data
def get(self, field):
return getattr(self, field)
def from_dict(self, data):
for field in ['name', 'describe', 'permissions']:
if field in data:
setattr(self, field, data[field])
class Tag(db.Model):
__tablename__ = 'tag'
id = db.Column(db.Integer,
autoincrement=True,
primary_key=True,
nullable=False)
name = db.Column(db.Text)
delete_tag = db.Column(db.Integer, default=0)
onindex = db.Column(db.Integer, default=0)
url = db.Column(db.String(30), nullable=False, server_default="")
def __repr__(self):
return '<Tag {}>'.format(self.name)
def to_dict(self):
data = {
"id": str(self.id),
"name": self.name,
"url": self.url,
"onindex": self.onindex
}
return data
def create_from_dict(self, data):
for field in ['name', 'url', 'onindex']:
if field in data:
setattr(self, field, data[field])
class User(db.Model):
"""
用户表
"""
__tablename__ = 'user'
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
name = db.Column(db.String(20), nullable=False) # 用户姓名
age = db.Column(db.Integer, nullable=False) # 用户年龄
class Student(db.Model):
"""
学员表
"""
__tablename__ = 'ht_student'
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
name = db.Column(db.String(30), nullable=False) # 姓名
phone = db.Column(db.String(20), nullable=True)
birthday = db.Column(db.String(20), nullable=True)
age = db.Column(db.Integer, nullable=True)
used_hour = db.Column(db.Integer, nullable=True)
left_hour = db.Column(db.Integer, nullable=True)
remark = db.Column(db.String(255), nullable=True)
type = db.Column(db.Integer, nullable=True)
status = db.Column(db.Integer, nullable=True)
create_time = db.Column(db.DateTime, nullable=True)
update_time = db.Column(db.DateTime, nullable=True)
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 EtAccount(db.Model):
__tablename__ = 'et_accounts'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(99), nullable=False, info='运营账号名称')
password = db.Column(db.String(99), nullable=False, info='账号密码')
role_id = db.Column(db.Integer, nullable=False, info='角色ID')
status = db.Column(db.Integer,
server_default=db.FetchedValue(),
info='状态: 1正常 2禁用')
add_time = db.Column(db.DateTime,
nullable=False,
server_default=db.FetchedValue(),
info='账号添加时间')
last_login = db.Column(db.DateTime,
nullable=False,
server_default=db.FetchedValue(),
info='账号添加时间')
ip = db.Column(db.String(20), nullable=False, info='ip')
class Class(db.Model):
"""
班级表
"""
__tablename__ = 'ht_class'
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
class_name = db.Column(db.String(200), nullable=False)
min_num = db.Column(db.Integer, nullable=True)
max_num = db.Column(db.Integer, nullable=True)
total_hour = db.Column(db.Integer, nullable=True)
teached_hour = db.Column(db.Integer, nullable=True)
teacher_id = db.Column(db.String(200), nullable=True)
remark = db.Column(db.String(255), nullable=True)
begin_time = db.Column(db.DateTime, nullable=True)
end_time = db.Column(db.DateTime, nullable=True)
classroom = db.Column(db.String(20), nullable=True)
status = db.Column(db.Integer, nullable=True)
target = db.Column(db.String(30), nullable=True)
create_time = db.Column(db.DateTime, nullable=True)
update_time = db.Column(db.DateTime, nullable=True)
class UserLoginMethod(db.Model):
"""
用户登陆验证表
"""
__tablename__ = 'user_login_method'
id = db.Column(db.Integer, autoincrement=True,
primary_key=True) # 用户登陆方式主键ID
# user_id = db.Column(db.Integer, nullable=False) # 用户主键ID
login_method = db.Column(db.String(36), nullable=False) # 用户登陆方式，WX微信，P手机
identification = db.Column(db.String(36),
nullable=False) # 用户登陆标识，微信ID或手机号
access_code = db.Column(db.String(36), nullable=True) # 用户登陆通行码，密码或token
nickname = db.Column(db.String(128), nullable=True, server_default="")
sex = db.Column(db.String(1), nullable=False, server_default="0")
admin = db.Column(db.String(1), nullable=False, server_default="0")
def to_dict(self):
return {
'id': self.id,
'login_method': self.login_method,
'identification': self.identification,
'access_code': self.access_code,
'nickname': self.nickname,
'sex': self.sex,
'admin': self.admin
}
class WXUser(db.Model):
# 设置数据库表名
__tablename__ = 'wxuser'
id = db.Column(db.Integer, autoincrement=True)
nickname = db.Column(db.String(128))
openid = db.Column(db.String(255), primary_key=True)
gender = db.Column(db.String(64))
country = db.Column(db.String(128))
province = db.Column(db.String(128))
city = db.Column(db.String(128))
def to_dict(self):
data = {
'id': self.id,
'nickname': self.nickname,
'openid': self.openid,
'gender': self.gender,
'country': self.country,
'province': self.province,
'city': self.city
}
return data
def from_dict(self, data):
for field in [
'nickname', 'openid', 'gender', 'country', 'province', 'city'
]:
if field in data:
setattr(self, field, data[field])
self.openid = data["openId"]
self.nickname = data["nickName"]
class Log(db.Model):
"""
日志表
"""
__tablename__ = 'ht_log'
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
type = db.Column(db.Integer, nullable=True)
time = db.Column(db.DateTime, nullable=True)
teacher_id = db.Column(db.Integer, nullable=True)
student_id = db.Column(db.Integer, nullable=True)
class_id = db.Column(db.Integer, nullable=True)
operator_id = db.Column(db.Integer, nullable=True)
remark = db.Column(db.String(255), nullable=True)
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 User(db.Model):
"""
用户表
"""
__tablename__ = 'ht_user'
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
name = db.Column(db.String(255), nullable=False) # 用户姓名
nick_name = db.Column(db.String(255), nullable=False) # 用户昵称
password = db.Column(db.String(255), nullable=False) # 用户密码
email = db.Column(db.String(255), nullable=True) # 用户密码
phone = db.Column(db.String(15), nullable=True) # 用户手机号
access_code = db.Column(db.String(255), nullable=True) # token
update_time = db.Column(db.DateTime, nullable=True) # 更新时间
class WXUser(db.Model):
# 设置数据库表名
__tablename__ = 'wxuser'
id = db.Column(db.Integer, index=True, autoincrement=True)
nickname = db.Column(db.String(128))
openid = db.Column(db.String(255), index=True, primary_key=True)
gender = db.Column(db.String(64))
country = db.Column(db.String(128))
province = db.Column(db.String(128))
city = db.Column(db.String(128))
email = db.Column(db.String(255), index=True)
def to_dict(self):
data = {
'id': self.id,
'nickname': self.nickname,
'openid': self.openid,
'gender': self.gender,
'country': self.country,
'province': self.province,
'city': self.city
}
return data
def from_dict(self, data):
for field in [
'nickname', 'openid', 'gender', 'country', 'province', 'city'
]:
if field in data:
setattr(self, field, data[field])
self.openid = data["openId"]
self.nickname = data["nickName"]
def get_jwt(self, expires_in=3600):
'''用户登录后，发放有效的 JWT'''
payload = {
"iss": 'wxapp',
"iat": int(time.time()),
"exp": int(time.time()) + 86400 * 7,
"aud": 'flask',
"openid": self.openid,
"nickname": self.nickname,
"scopes": ['open']
}
return jwt.encode(payload,
current_app.config['SECRET_KEY'],
algorithm='HS256')
@staticmethod
def verify_jwt(token):
'''验证 JWT 的有效性'''
try:
payload = jwt.decode(token,
current_app.config['SECRET_KEY'],
algorithms=['HS256'],
audience="flask")
except (jwt.exceptions.ExpiredSignatureError,
jwt.exceptions.InvalidSignatureError,
jwt.exceptions.DecodeError) as e:
# Token过期，或被人修改，那么签名验证也会失败
return None
return WXUser.query.get(payload.get('openid'))
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]
