From efefc7fc03e551a314d3b1e38454512bdca5c7e4 Mon Sep 17 00:00:00 2001 From: Jake Moshenko Date: Fri, 2 May 2014 17:33:14 -0400 Subject: [PATCH] Add a tool to backfill the upload flags from the registry data. --- tools/backfilluploadingflag.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tools/backfilluploadingflag.py diff --git a/tools/backfilluploadingflag.py b/tools/backfilluploadingflag.py new file mode 100644 index 000000000..6d5ce4a37 --- /dev/null +++ b/tools/backfilluploadingflag.py @@ -0,0 +1,20 @@ +import logging + +from data.database import ImageStorage +from app import storage + +logging.basicConfig(level=logging.DEBUG) +logger = logging.getLogger(__name__) + +for image_storage in ImageStorage.select(ImageStorage.uploading == None).limit(1000): + mark_path = storage.image_mark_path(None, None, None, image_storage.uuid) + json_path = storage.image_json_path(None, None, None, image_storage.uuid) + + logger.debug('Mark path: %s Json path: %s', mark_path, json_path) + + if storage.exists(json_path): + image_storage.uploading = storage.exists(mark_path) + logger.debug('Image existed and was currently uploading: %s', image_storage.uploading) + # image_storage.save() + else: + logger.warning('Image does not exist.')