一对多
from exts import db
#种类模型
class Categroy(db.Model):
id = db.Column(db.Integer,primary_key=True,autoincrement=True,comment='分类ID')
name = db.Column(db.String(10),comment='分类名')
img = db.Column(db.String(250),comment='图片')
goods = db.relationship('Goods',backref='cate') #声明与哪个模型建立关系，不会数据库产生实质操作 backref的值是给"多方"用来查询"一方"的
#商品模型
class Goods(db.Model):
id = db.Column(db.Integer,primary_key=True,autoincrement=True,comment='商品ID')
title = db.Column(db.String(20),default='',comment='商品名')
price = db.Column(db.DECIMAL(10,2),comment='售价')
img = db.Column(db.String(250),comment='图片')
cate_id = db.Column(db.Integer,db.ForeignKey('categroy.id')) # 声明一个字段为外键, 关联categroy.id categroy是数据库的表名
一对多关系的查询:
从一查多:
categroy= Categroy.query.get(1)
goods= categroy.goods# 不需要.all()
从多查一:
goods= Goods.query.get(2)
categroy= goods.cate
多对多
需要定义一个中间表，包含两个模型的外键字段就可以了，并作为一个“复合主键”
在两个需要做多对多的模型中随便选择一个模型，定义一个relationship属性，来绑定三者之间的关系，在使用relationship的时候，需要传入一个secondary=中间表模型名。
# 中间表
article_tag = db.Table(
'article_tag',
db.Column('article_id', db.Integer,db.ForeignKey("article.id"),primary_key=True),
db.Column('tag_id', db.Integer, db.ForeignKey("tag.id"), primary_key=True)
)
# 文章表
class Article(db.Model):
__tablename__ = 'article'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(64))
tags = db.relationship('Tag', secondary=article_tag, backref=db.backref('articles'))
# 标签表
class Tag(db.Model):
__tablename__ = 'tag'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32))
多对多关系的查询
1.查询文章所有的标签:
article = Article.query.filter(Article.title=="xxx").first()
tags = article.tags
2.查询一个标签涉及的所有文章:
tag = Tag.query.get(2)
articles = tag.articles
自定义SQL语句
sql = 'select * from stu where id=1'
results = db.session.execute(sql)
增删改查操作
这里以一对多关系模型为例，演示增删改查操作
# 分类视图 增删改查
class CateView(Resource):
def get(self):
cate_info = Categroy.query.all()
cate_list = []
for i in cate_info:
dict = {'id':i.id,'name':i.name}
cate_list.append(dict)
print(cate_list)
return jsonify({'msg':'获取成功','code':200,'data':cate_list})
def post(self):
req = reqparse.RequestParser() #实例化请求对象
req.add_argument('name',required=True) #添加参数
args = req.parse_args() #校验数据
c1 = Categroy(name=args['name']) #插入数据库
db.session.add(c1)
db.session.commit()
return jsonify({'msg':'添加成功','code':200})
def put(self):
req = reqparse.RequestParser()
req.add_argument('id',type=int, required=True) #接受参数
req.add_argument('name',required=True)
args = req.parse_args()
cate_info = Categroy.query.get(args['id'])
if not cate_info:
return jsonify({'msg': '分类不存在', 'code': 400})
cate_info.name = args['name']
db.session.commit()
return jsonify({'msg': '修改成功', 'code': 200})
def delete(self):
req = reqparse.RequestParser()
req.add_argument('id', type=int,required=True,help='ID必填') # 接受参数
args = req.parse_args()
print(args)
Categroy.query.filter(Categroy.id == args['id']).delete()
db.session.commit()
return jsonify({'msg': '删除成功', 'code': 200})
api.add_resource(CateView,'/cate')
from datetime import datetime
import random,os
# 商品视图 增删改查
class GoodsView(Resource):
def get(self):
req = reqparse.RequestParser()
req.add_argument('cate_id',type=int,required=True)
args = req.parse_args()
cate_info = Categroy.query.get(args['cate_id'])
goods = cate_info.goods
goods_list = []
for i in goods: #int 不能json 转为字符串型 #反向查询分类名字
dict = {'id':i.id,'title':i.title,'price':str(i.price),'cate_name':i.cate.name,'img':i.img}
goods_list.append(dict)
print(goods_list)
return jsonify({'msg':'获取成功','code':200,'data':goods_list})
def post(self):
req = reqparse.RequestParser() #实例化请求对象
req.add_argument('title',required=True) #添加参数
req.add_argument('price',type=float, required=True)
req.add_argument('cate_id',type=int, required=True)
req.add_argument('img')
args = req.parse_args() #校验数据
cate_info = Categroy.query.get(args['cate_id']) #看id存不存在
if not cate_info:
return jsonify({'msg': '分类不存在', 'code': 400})
c1 = Goods(title=args['title'],price=args['price'],img=args['img'],cate=cate_info) #插入数据库
db.session.add(c1)
db.session.commit()
return jsonify({'msg':'添加成功 - {%s}' % c1.title,'code':200})
def put(self): #修改 ： 名字售价、图片、分类
req = reqparse.RequestParser()
req.add_argument('id',type=int, required=True) #接受参数
req.add_argument('title')
req.add_argument('price',type=float)
req.add_argument('cate_id', type=int)
req.add_argument('img')
args = req.parse_args()
cate_info = Goods.query.get(args['id'])
if not cate_info:
return jsonify({'msg': '商品不存在', 'code': 400}) #商品存在则修改
if args['title']:
cate_info.name = args['title']
if args['price']:
cate_info.price = args['price']
if args['cate_id']:
cate_lst = Categroy.query.get(args['cate_id']) #分类是否存在
if cate_lst:
cate_info.cate_id = args['cate_id']
if args['img']:
if not os.path.exists('./static/img/%s' % args['img']):
return jsonify({'msg': '图片不存在', 'code': 400})
cate_info.img = args['img']
db.session.commit()
return jsonify({'msg': '修改成功', 'code': 200})
def delete(self):
req = reqparse.RequestParser()
req.add_argument('id', type=int,required=True,help='ID必填') # 接受参数
args = req.parse_args()
print(args)
Goods.query.filter(Goods.id == args['id']).delete()
db.session.commit()
return jsonify({'msg': '删除成功', 'code': 200})
api.add_resource(GoodsView,'/goods')
# 上传图片视图
class UploadView(Resource):
def post(self):
img = request.files.get('img') # 获取图片
t = datetime.strftime(datetime.now(), '%Y%m%d%H%M%S')
r = str(random.randint(100000, 999999)) # 生成图片文件名
file_name = t + r + '.png'
img.save('./static/img/%s' % file_name) # 存文件
return jsonify({'msg': '上传成功', 'code': 200, 'data': file_name})
api.add_resource(UploadView,'/upload')