Flask 数据库表关系(一对多和多对多）及增删改查操作

关系查询

一、一对多关系的建立: 一方建立关系, 多方建立外键

#种类模型
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')