wbfdm 1.49.1__tar.gz → 1.49.2__tar.gz

{wbfdm-1.49.1 → wbfdm-1.49.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbfdm
-Version: 1.49.1
+Version: 1.49.2
 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.49.1 → wbfdm-1.49.2}/wbfdm/contrib/dsws/dataloaders/market_data.py

@@ -75,6 +75,8 @@ class DSWSMarketDataDataloader(MarketDataProtocol, Dataloader):
                         parameters["end"] = to_date.strftime("%Y-%m-%d")
                 for chunked_identifiers in client.get_chunked_list(identifiers, len(fields)):
                     df = client.get_timeserie_df(chunked_identifiers, fields, **parameters)
+                    if exact_date:
+                        df = df[df["Dates"] == exact_date]
                     for row in df.to_dict("records"):
                         jsondate = row["Dates"].date()
                         external_id = row["Instrument"]

{wbfdm-1.49.1 → wbfdm-1.49.2}/wbfdm/models/instruments/instruments.py

@@ -130,6 +130,9 @@ class InstrumentManager(TreeManager):
     def annotate_all(self):
         return self.get_queryset().annotate_all()
 
+    def filter_active_at_date(self, val_date: date):
+        return self.get_queryset().filter_active_at_date(val_date)
+
 
 class SecurityInstrumentManager(InstrumentManager):
     def get_queryset(self) -> InstrumentQuerySet:
@@ -142,19 +145,8 @@ class ClassifiableInstrumentManager(InstrumentManager):
 
 
 class ActiveInstrumentManager(InstrumentManager):
-    def get_queryset_at_date(self, val_date: date):
-        return (
-            super()
-            .get_queryset()
-            .filter(
-                (models.Q(delisted_date__isnull=True) | models.Q(delisted_date__gte=val_date))
-                & models.Q(inception_date__isnull=False)
-                & models.Q(inception_date__lte=val_date)
-            )
-        )
-
     def get_queryset(self):
-        return self.get_queryset_at_date(date.today())
+        return super().get_queryset().filter_active_at_date(date.today())
 
 
 class InvestableUniverseManager(InstrumentManager):
@@ -579,6 +571,18 @@ class Instrument(ComplexToStringMixin, TagModelMixin, ImportMixin, InstrumentPMS
         if favorite_classification := self.classifications.filter(group=get_default_classification_group()).first():
             return favorite_classification
 
+    def get_primary_quote(self):
+        if self.children.exists():
+            # we try the primary children first, then the first children within the investable universe and we fallback to the first children otherwise
+            try:
+                instrument = self.children.get(is_primary=True)
+            except Instrument.DoesNotExist:
+                instrument = self.children.filter(is_investable_universe=True).first()
+                if not instrument:
+                    instrument = self.children.first()
+            return instrument.get_primary_quote()
+        return self
+
     @property
     def active(self) -> bool:
         today = date.today()

{wbfdm-1.49.1 → wbfdm-1.49.2}/wbfdm/models/instruments/querysets.py

@@ -10,6 +10,7 @@ from django.db.models import (
     ExpressionWrapper,
     F,
     OuterRef,
+    Q,
     QuerySet,
     Subquery,
 )
@@ -20,6 +21,12 @@ from wbfdm.enums import MarketData
 
 
 class InstrumentQuerySet(QuerySet):
+    def filter_active_at_date(self, val_date: date):
+        return self.filter(
+            (Q(delisted_date__isnull=True) | Q(delisted_date__gte=val_date))
+            & (Q(inception_date__isnull=True) | Q(inception_date__lte=val_date))
+        )
+
     def annotate_classification_for_group(
         self, classification_group, classification_height: int = 0, **kwargs
     ) -> QuerySet: