class Role(db.Model):
__tablename__ = "t_role"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32), unique=True, nullable=False)
desc = db.Column(db.String(64))
users = db.relationship('User', backref='role')
menus = db.relationship('Menu', secondary=trm)
def to_dict(self):
return {
'id': self.id,
'name': self.name,
'desc': self.desc,
'menu': self.get_menu_dict()
}
def get_menu_dict(self):
menu_list = []
menus = sorted(self.menus, key=lambda temp: temp.id)
for m in menus:
if m.level == 1:
first_dict = m.to_dict()
first_dict['children'] = []
for s in menus:
if s.level == 2 and s.pid == m.id:
first_dict['children'].append(s.to_dict())
menu_list.append(first_dict)
return menu_list
class User(db.Model, BaseModel):
__tablename__ = 't_table'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32), unique=True, nullable=False)
pwd = db.Column(db.String(128))
nick_name = db.Column(db.String(32))
phone = db.Column(db.String(11))
email = db.Column(db.String(32))
craete_time = db.Column(db.DateTime, default=datetime.now)
update_time = db.Column(db.DateTime, default=datetime.now)
rid = db.Column(db.Integer, db.ForeignKey('t_role.id'))
@property
def password(self):
return pwd
@password.setter
def password(self, t_pwd):
self.pwd = generate_password_hash(t_pwd)
def check_password(self, t_pwd):
return check_password_hash(self.pwd, t_pwd)
def to_dict(self):
return {
'id': self.id,
'name': self.name,
'nick_name': self.nick_name,
'phone': self.phone,
'email': self.email,
'rolename': self.role.name if self.role else ''
}
class Menu(db.Model):
__tablename__ = 't_menu'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32), unique=True, nullable=False)
level = db.Column(db.Integer)
path = db.Column(db.String(32))
pid = db.Column(db.Integer, db.ForeignKey('t_menu.id'))
children = db.relationship('Menu')
roles = db.relationship('Role', secondary=trm)
def to_dict(self):
return {
'id': self.id,
'name': self.name,
'level': self.level,
'path': self.path,
'pid': self.pid
}
def get_children_list(self):
obj_children = self.children
data = []
for o in obj_children:
data.append(o.to_dict())
return data
class Order(db.Model, BaseModel):
__tablename__ = 't_order'
id = db.Column(db.Integer, primary_key=True)
uid = db.Column(db.Integer, db.ForeignKey('t_table.id'))
price = db.Column(db.Float)
number = db.Column(db.Integer)
pay_status = db.Column(db.Integer) # 0 没支付,1 已支付
is_send = db.Column(db.Integer) # 0 没发送,1 已发送
fapiao_title = db.Column(db.String(32))
fapiao_company = db.Column(db.String(128))
fapiao_content = db.Column(db.String(521))
addrs = db.Column(db.String(521))
user = db.relationship('User', foreign_keys=[uid])
order_detail = db.relationship('OrderDetail', backref='order')
express = db.relationship('Express', backref='order')
def to_dict(self):
return {
'id': self.id,
'uid': self.uid,
'uname': self.user.nick_name,
'price': self.price,
'number': self.number,
'pay_status': self.pay_status,
'is_send': self.is_send,
'fapiao_title': self.fapiao_title,
'fapiao_content': self.fapiao_content,
'addrs': self.addrs
}
class GoodsAttr(db.Model):
__tablename__ = 't_goods_attr'
gid = db.Column(db.Integer, db.ForeignKey('t_goods.id'), primary_key=True)
aid = db.Column(db.Integer,
db.ForeignKey('t_attribute.id'),
primary_key=True)
val = db.Column(db.String(512))
_type = db.Column(db.String(8))
class Express(db.Model):
__tablename__ = 't_express'
id = db.Column(db.Integer, primary_key=True)
content = db.Column(db.String(64))
update_time = db.Column(db.String(32))
oid = db.Column(db.Integer, db.ForeignKey('t_order.id'))
def to_dict(self):
return {
'id': self.id,
'content': self.content,
'update_time': self.update_time,
'oid': self.oid
}
class Attribute(db.Model):
__tablename__ = 't_attribute'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32))
val = db.Column(db.String(255))
cid = db.Column(db.Integer, db.ForeignKey('t_category.id'))
_type = db.Column(db.Enum('static', 'dynamic'))
def to_dict(self):
return {
'id': self.id,
'name': self.name,
'val': self.val,
'cid': self.cid,
'type': self._type,
}
class Goods(db.Model):
__tablename__ = 't_goods'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(512))
price = db.Column(db.Float)
number = db.Column(db.Integer)
introduce = db.Column(db.Text)
big_log = db.Column(db.String(256))
small_log = db.Column(db.String(256))
state = db.Column(db.Integer) # 0 未通过 1 审核中 2 已审核
is_promate = db.Column(db.Integer)
hot_number = db.Column(db.Integer)
weight = db.Column(db.Integer)
cid_one = db.Column(db.Integer, db.ForeignKey('t_category.id'))
cid_two = db.Column(db.Integer, db.ForeignKey('t_category.id'))
cid_three = db.Column(db.Integer, db.ForeignKey('t_category.id'))
category = db.relationship('Category', foreign_keys=[cid_three])
def to_dict(self):
return {
'id': self.id,
'name': self.name,
'price': self.price,
'number': self.number,
'introduce': self.introduce,
'big_log': self.big_log,
'small_log': self.small_log,
'state': self.state,
'is_promate': self.is_promate,
'hot_number': self.hot_number,
'weight': self.weight,
'cid_one': self.cid_one,
'cid_two': self.cid_two,
'cid_three': self.cid_three,
'attr': [a.to_dict() for a in self.category.attr]
}
class Category(db.Model):
__tablename__ = 't_category'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32), nullable=False)
level = db.Column(db.Integer)
pid = db.Column(db.Integer, db.ForeignKey('t_category.id'))
children = db.relationship('Category')
attr = db.relationship('Attribute', backref='category')
def to_dict(self):
return {
'id': self.id,
'name': self.name,
'level': self.level,
'pid': self.pid,
}
class Picture(db.Model):
__tablename__ = 't_picture'
id = db.Column(db.Integer, primary_key=True)
path = db.Column(db.String(512))
gid = db.Column(db.Integer, db.ForeignKey('t_goods.id'))