From e7408dcd98a9ed2eed01b9fc7531287eacf29efd Mon Sep 17 00:00:00 2001 From: John Davis Date: Fri, 2 Jun 2023 14:01:37 -0400 Subject: [PATCH] Wrap session.flush in populate_uuid script --- scripts/cleanup_datasets/populate_uuid.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/scripts/cleanup_datasets/populate_uuid.py b/scripts/cleanup_datasets/populate_uuid.py index b41f2f6db38..64614fe264f 100755 --- a/scripts/cleanup_datasets/populate_uuid.py +++ b/scripts/cleanup_datasets/populate_uuid.py @@ -15,6 +15,7 @@ import uuid sys.path.insert(1, os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, "lib"))) import galaxy.config +from galaxy.model.base import transaction from galaxy.model.mapping import init_models_from_config from galaxy.util.script import ( app_properties_from_args, @@ -37,18 +38,21 @@ def main(): app_properties = app_properties_from_args(args) config = galaxy.config.Configuration(**app_properties) model = init_models_from_config(config) + session = model.context() for row in model.context.query(model.Dataset): if row.uuid is None: row.uuid = uuid.uuid4() print("Setting dataset:", row.id, " UUID to ", row.uuid) - model.context.flush() + with transaction(session): + session.commit() for row in model.context.query(model.Workflow): if row.uuid is None: row.uuid = uuid.uuid4() print("Setting Workflow:", row.id, " UUID to ", row.uuid) - model.context.flush() + with transaction(session): + session.commit() print("Complete")