Add deduplicates migration step.

pull/919/head
Andrey 2023-09-19 13:44:54 +03:00
rodzic 5a6c8f5219
commit 1e848f72d4
2 zmienionych plików z 8 dodań i 2 usunięć

Wyświetl plik

@ -127,13 +127,20 @@ def migrations_run(args: argparse.Namespace) -> None:
db_session = SessionLocal() db_session = SessionLocal()
try: try:
if args.id == 20230904: if args.id == 20230904:
step_order = ["fill_missing_selectors_in_moonworm_tasks"] step_order = [
"fill_missing_selectors_in_moonworm_tasks",
"deduplicate_moonworm_tasks",
]
step_map: Dict[str, Dict[str, Any]] = { step_map: Dict[str, Dict[str, Any]] = {
"upgrade": { "upgrade": {
"fill_missing_selectors_in_moonworm_tasks": { "fill_missing_selectors_in_moonworm_tasks": {
"action": add_selectors.fill_missing_selectors_in_moonworm_tasks, "action": add_selectors.fill_missing_selectors_in_moonworm_tasks,
"description": "Get all moonworm jobs from moonworm journal and add selector tag if it not represent", "description": "Get all moonworm jobs from moonworm journal and add selector tag if it not represent",
}, },
"deduplicate_moonworm_tasks": {
"action": add_selectors.deduplicate_moonworm_task_by_selector,
"description": "Deduplicate moonworm tasks by selector",
},
}, },
"downgrade": {}, "downgrade": {},
} }

Wyświetl plik

@ -55,7 +55,6 @@ def fill_missing_selectors_in_moonworm_tasks() -> None:
logger.warn( logger.warn(
f"Unable to parse abi from task: {task.entry_url.split()[-1]}: {e}" f"Unable to parse abi from task: {task.entry_url.split()[-1]}: {e}"
) )
raise e
continue continue
if "name" not in abi: if "name" not in abi: