Skip to content

Commit a838731

Browse files
committed
get spectidic metadata of model
1 parent e683931 commit a838731

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

migrate_geography_metadata.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
UseCase,
4646
UseCaseMetadata,
4747
)
48+
from api.utils.enums import MetadataModels
4849

4950

5051
def parse_geography_value(value: str) -> List[str]:
@@ -76,7 +77,9 @@ def migrate_dataset_geographies(
7677

7778
# Get geography metadata item
7879
try:
79-
geo_metadata_item = Metadata.objects.get(label="Geography")
80+
geo_metadata_item = Metadata.objects.get(
81+
label="Geography", model=MetadataModels.DATASET
82+
)
8083
except Metadata.DoesNotExist:
8184
print("⚠️ No Geography metadata item found. Skipping dataset migration.")
8285
return
@@ -169,7 +172,9 @@ def migrate_usecase_geographies(
169172

170173
# Get geography metadata item
171174
try:
172-
geo_metadata_item = Metadata.objects.get(label="Geography")
175+
geo_metadata_item = Metadata.objects.get(
176+
label="Geography", model=MetadataModels.USECASE
177+
)
173178
except Metadata.DoesNotExist:
174179
print("⚠️ No Geography metadata item found. Skipping usecase migration.")
175180
return

0 commit comments

Comments
 (0)