From 64069b9b5d374894b4f6040befe69f15420d7cca Mon Sep 17 00:00:00 2001 From: Kevin Morris Date: Sun, 16 Jan 2022 02:00:41 -0800 Subject: [PATCH] change(python): use a transaction query in get_pkg_or_base Signed-off-by: Kevin Morris --- aurweb/packages/util.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/aurweb/packages/util.py b/aurweb/packages/util.py index fd8fdf85..21b5fab8 100644 --- a/aurweb/packages/util.py +++ b/aurweb/packages/util.py @@ -103,11 +103,13 @@ def get_pkg_or_base( if provider: raise HTTPException(status_code=HTTPStatus.NOT_FOUND) - instance = db.query(cls).filter(cls.Name == name).first() + with db.begin(): + instance = db.query(cls).filter(cls.Name == name).first() + if not instance: raise HTTPException(status_code=HTTPStatus.NOT_FOUND) - return db.refresh(instance) + return instance def get_pkgbase_comment(pkgbase: models.PackageBase, id: int) \