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]