Flask SQLAlchemy多次插入重复条目

时间:2016-03-08 06:33:15

标签: python flask sqlalchemy flask-sqlalchemy

我创建了一个Flask-SQLAlchemy API,其中包含Items与Tags之间的多对多关系。

一个简单的添加没有问题(在适当的表中插入项目和标签),但如果我添加另一个具有相同标签名称的项目,它会创建一组重复的标签名称,而不是仅仅将关联表链接到现有标签条目。

我发现一些情况与我要求的情况非常接近,但并不完全相同,我真的在努力去理解实现这一目标的最佳方法。

所以问题是:在标签表中只有唯一条目的同时仍然正确地将标签与items表相关联的最佳方法是什么?

感谢您的帮助/指导!

以下是我的模特:

items_tags = db.Table('items_tags',
    db.Column('tag_id', db.Integer, db.ForeignKey('tag.id')),
    db.Column('item_id', db.Integer, db.ForeignKey('item.id'))
)

class Item(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(80), unique=False)
    code = db.Column(db.String(), unique=False)
    notes = db.Column(db.String(), unique=False)

    tags = db.relationship('Tag', secondary=items_tags, back_populates="items")

    def json(self):
        list_tag = []
        for tag in self.tags:
            list_tag.append({"tag": tag.name, "id": tag.id })
        return json.loads(json.dumps({"id": self.id, "title": self.title, "code": self.code, "notes": self.notes, "tags" : list_tag }))

    def __init__(self, title, code, notes, tags):
        self.title = title
        self.code = code
        self.notes = notes        
        self.tags = tags

    def __repr__(self):
        return self.title

class Tag(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50))

    items = db.relationship('Item', secondary=items_tags, back_populates="tags")

    def json(self):
        return {'id': self.id, 'tag': self.name}

    def __init__(self, name):
        self.name = name

    def __repr__(self):
        return self.name

这是控制器:

# curl -k -H "Content-Type: application/json" -X POST https://x.x.x.x/api/v1.0/items/ -d '{"title": "title03", "code": "code03", "notes": "notes03", "tags": [{"tag": "tag01"}, {"tag": "tag02"}]}'
@app.route('/api/v1.0/items/', methods=['GET', 'POST'])
def items():
    if request.method == 'GET':
        return jsonify({"items": [item.json() for item in Item.query.all()]})

    if request.method == 'POST':
        if 'tags' in request.json:
            tags = request.json['tags']    
            for tag in tags:
              list_tags = [Tag(tag['tag']) for tag in tags]

        item = Item(request.json['title'], request.json['code'], request.json['notes'], list_tags)
        db.session.merge(item)
        db.session.commit()
        return jsonify({}), 201

1 个答案:

答案 0 :(得分:0)

我错过了这样一个事实:我需要先查询标签是否存在(@dirn正在使用“改变你如何加载现有记录”)。

以下是我为确保未添加重复标记而采取的措施:

if request.method == 'POST':
    if 'tags' in request.json:
        list_tags = []
        for tag in request.json['tags']:
            if Tag.query.filter_by(name=tag['tag']).first():
                list_tags.append(Tag.query.filter_by(name=tag['tag']).first())
            else:
                list_tags.append(Tag(tag['tag']))

    item = Item(request.json['title'], request.json['code'], request.json['notes'], list_tags)
    db.session.merge(item)
    db.session.commit()
    return jsonify({}), 201