GPI-1232 Synchronous importer for local sizes

This commit is contained in:
Mieszko Banczerowski
2020-09-16 13:33:17 +02:00
parent 0cf447bdcf
commit 17b0542fdf
3 changed files with 26 additions and 10 deletions

View File

@@ -87,3 +87,16 @@ class CollectionImporter(Importer):
self._notification_failure(id_, UnknownError())
finally:
self._notification_partially_finished(id_)
class SynchroneousImporter(Importer):
async def _import_elements(self, ids_, context_):
try:
for id_ in ids_:
await self._import_element(id_, context_)
self._notification_finished()
self._complete()
except asyncio.CancelledError:
logger.debug("Importing %s cancelled", self._name)
finally:
self._import_in_progress = False

View File

@@ -13,7 +13,7 @@ from galaxy.api.types import (
Subscription, SubscriptionGame
)
from galaxy.task_manager import TaskManager
from galaxy.api.importer import Importer, CollectionImporter
from galaxy.api.importer import Importer, CollectionImporter, SynchroneousImporter
logger = logging.getLogger(__name__)
@@ -104,7 +104,7 @@ class Plugin:
self._user_presence_import_finished,
self.user_presence_import_complete
)
self._local_size_importer = Importer(
self._local_size_importer = SynchroneousImporter(
self._external_task_manager,
"local size",
self.get_local_size,