wbfdm 1.52.0__tar.gz → 1.52.2rc0__tar.gz

{wbfdm-1.52.0 → wbfdm-1.52.2rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbfdm
-Version: 1.52.0
+Version: 1.52.2rc0
 Summary: The workbench module ensures rapid access to diverse financial data (market, fundamental, forecasts, ESG), with features for storing instruments, classifying them, and conducting financial analysis.
 Author-email: Christopher Wittlinger <c.wittlinger@stainly.com>
 Requires-Dist: roman==4.*

{wbfdm-1.52.0 → wbfdm-1.52.2rc0}/wbfdm/contrib/qa/sync/utils.py

@@ -9,6 +9,7 @@ from django.db import IntegrityError, connections
 from django.db.models import Max, QuerySet
 from django.template.loader import get_template
 from jinjasql import JinjaSql  # type: ignore
+from mptt.exceptions import InvalidMove
 from psycopg.errors import UniqueViolation
 from tqdm import tqdm
 from wbcore.contrib.currency.models import Currency
@@ -241,22 +242,26 @@ def update_instruments(sql_name: str, parent_source: str | None = None, context=
 
 def update_or_create_item(
     external_id: int, get_item: Callable, source: str, parent_source: str | None = None
-) -> Instrument:
+) -> Instrument | None:
     defaults = convert_data(get_item(external_id), parent_source=parent_source)
 
     dl_parameters = defaults.pop("dl_parameters", {})
     defaults.pop("source", None)
     defaults.pop("source_id", None)
+    try:
+        instrument, _ = Instrument.objects.update_or_create(
+            source=source,
+            source_id=external_id,
+            defaults=defaults,
+        )
+        instrument.dl_parameters.update(dl_parameters)
+        _save_single_instrument(instrument)
+        return instrument
 
-    instrument, _ = Instrument.objects.update_or_create(
-        source=source,
-        source_id=external_id,
-        defaults=defaults,
-    )
-    instrument.dl_parameters.update(dl_parameters)
-    _save_single_instrument(instrument)
-
-    return instrument
+    except InvalidMove:
+        logger.warning(
+            f"We encountered a MPTT ill-formed node for source ID {external_id}. Rebuilding the tree might be necessary."
+        )
 
 
 def get_item(external_id: int, template_name: str) -> dict: