时间戳

首页 > 解决方案 > 仅通过 API 获取唯一约束的 peewee IntegrityError

python - 仅通过 API 获取唯一约束的 peewee IntegrityError

问题描述

尝试使用 peeweee 更新 PostgreSQL 数据库上的记录时出现 IntegrityError。但是，只有在我尝试 HTTP PUT 方法时才会发生这种情况。

使用 Flask 和 Flask-Restful 创建 API 资源。GET、DELETE 到单个博客帖子效果很好。发布到存储桶完美地工作

我可以通过 REPL 运行完全相同的代码，它工作得很好。更奇怪的是API突然坏了，我昨天测试了这个功能，它很好。现在我无法确定发生了什么变化。

这是我在 models.py 中的 peewee 模型


class BlogPost(Model):
    title = CharField(default='', unique=True)
    content = TextField(default='')
    created = DateTimeField(default=datetime.datetime.now)

    class Meta:
        database = DATABASE

这是 resources.blogposts.py 中单个 BlogPost 的资源


class BlogPost(Resource):
    def __init__(self):
        self.reqparse = reqparse.RequestParser()
        self.reqparse.add_argument(
            'title',
            required=False,
            help='No title provided',
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'content',
            required=False,
            nullable=True,
            location=['form', 'json'],
            default=''
        )
        super().__init__()

    @marshal_with(blogpost_fields)
    def get(self, id):
        return (blogpost_or_404(id))

    @marshal_with(blogpost_fields)
    @auth.login_required
    def put(self, id):
        args = self.reqparse.parse_args()
        try:
            blogpost = models.BlogPost.get(models.BlogPost.id==id)
        except models.BlogPost.DoesNotExist:
            return make_response(json.dumps(
                    {'error': 'That blogpost does not exist or is not editable'}
                ), 403)
        else:
            query = blogpost.update(**args)
            query.execute()
            blogpost = (blogpost_or_404(id))
            return (blogpost, 200, {
                'Location': url_for('resources.blogposts.blogpost', id=id)
               })

    @auth.login_required
    def delete(self, id):
        try:
            blogpost = models.BlogPost.select().where(
                models.BlogPost.id==id
            ).get()
        except models.BlogPost.DoesNotExist:
            return make_response(json.dumps(
                    {'error': 'That blogpost does not exist or is not editable'}
                ), 403)
        else:
            query = blogpost.delete().where(models.BlogPost.id==id)
            query.execute()
            return '', 204, {'Location': url_for('resources.blogposts.blogposts')}

blogposts_api = Blueprint('resources.blogposts', __name__)
api = Api(blogposts_api)
api.add_resource(
    BlogPost,
    'api/v1/blogposts/<int:id>',
    endpoint='blogpost'
)

如果我对http://localhost:8000/api/v1/blogposts/8执行 GET，我会在表中得到我唯一的 BlogPost（我删除了除此之外的所有内容以进行测试）


{
    "id": 8,
    "title": "Test3",
    "content": "This is to test changes to BlogPost 1",
    "created": "Wed, 19 Jun 2019 12:44:31 -0000"
}

但是，如果我 PUT 到同一个 URL，我会得到一个唯一约束。我绝对没有这个标题的数据库条目。


{
    "title": "9sdnfsudngfisdngondasgjns",
    "content": "lkbksigsndignsoidugnlis",
}

我可以在 REPL 中执行此操作，并且效果非常好，它应该与我在上面所做的完全相同：


blogpost = models.BlogPost.get(models.BlogPost.id==8)
blogpost.update(
    title="9sdnfsudngfisdngondasgjns",
    content="lkbksigsndignsoidugnlis"
)
blogpost.execute()

实际错误：

peewee.IntegrityError：重复键值违反唯一约束“blogpost_title”详细信息：键（标题）=（9sdnfsudngfisdngondasgjns）已经存在。

编辑：在 peewee 错误 psycopg2 也抛出此错误之前：

UniqueViolation：重复的键值违反了唯一约束“user_username_key”详细信息：键（用户名）=（9sdnfsudngfisdngondasgjns）已经存在。

标签： pythonpostgresqlflaskflask-restfulpeewee

解决方案

这真的很奇怪，但我想通了。我必须将我的 put 方法更新为与我的 delete 方法相同的语法。基本上，我必须将上下文传递给实际的更新方法：


@marshal_with(blogpost_fields)
    @auth.login_required
    def put(self, id):
        args = self.reqparse.parse_args()
        try:
            blogpost = models.BlogPost.select().where(
                models.BlogPost.id==id).get()
        except models.BlogPost.DoesNotExist:
            return make_response(json.dumps(
                    {'error': 'That blogpost does not exist or is not editable'}
                ), 403)
        else:
            query = blogpost.update(**args).where(models.BlogPost.id==id)
            query.execute()
            blogpost = (blogpost_or_404(id))
            return (blogpost, 200, {
                'Location': url_for('resources.blogposts.blogpost', id=id)
               })

    @auth.login_required
    def delete(self, id):
        try:
            blogpost = models.BlogPost.select().where(
                models.BlogPost.id==id).get()
        except models.BlogPost.DoesNotExist:
            return make_response(json.dumps(
                    {'error': 'That blogpost does not exist or is not editable'}
                ), 403)
        else:
            query = blogpost.delete().where(models.BlogPost.id==id)
            query.execute()
            return '', 204, {'Location': url_for('resources.blogposts.blogposts')}

推荐阅读