diff --git a/swh/deposit/migrations/0018_migrate_swhids.py b/swh/deposit/migrations/0018_migrate_swhids.py --- a/swh/deposit/migrations/0018_migrate_swhids.py +++ b/swh/deposit/migrations/0018_migrate_swhids.py @@ -17,6 +17,7 @@ REVISION, SNAPSHOT, ) +from swh.model.model import Origin from swh.storage import get_storage as get_storage_client @@ -123,7 +124,7 @@ dir_id = obj_dir.object_id origin = obj_dir.metadata["origin"] - check_origin = storage.origin_get({"url": origin}) + check_origin = storage.origin_get([Origin(url=origin)])[0] if not check_origin: logger.warning("Deposit id %s: Origin %s not found!", deposit.id, origin) continue @@ -277,7 +278,7 @@ external_id = deposit.external_id origin = resolve_origin(deposit.id, provider_url, external_id) - check_origin = storage.origin_get({"url": origin}) + check_origin = storage.origin_get([Origin(url=origin)])[0] if not check_origin: logger.warning("Deposit id %s: Origin %s not found!", deposit.id, origin) continue