wbfdm 1.49.0__py2.py3-none-any.whl → 1.49.2__py2.py3-none-any.whl

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/models/instruments/instrument_requests.py

@@ -1,6 +1,7 @@
 from typing import Any, Dict, Optional
 
 from django.db import models
+from django.utils.translation import gettext_lazy as _
 from django_fsm import FSMField, transition
 from wbcore.contrib.authentication.models import User
 from wbcore.contrib.currency.models import Currency
@@ -112,7 +113,7 @@ class InstrumentRequest(WBModel):
 
     def can_approve(self):
         if self._check_already_existing_instrument():
-            return {"non_field_errors": "An instrument already exists with the proposed identifier"}
+            return {"non_field_errors": [_("An instrument already exists with the proposed identifier")]}
 
     @transition(
         field=status,
@@ -178,7 +179,7 @@ class InstrumentRequest(WBModel):
 
     def can_submit(self) -> Dict[str, Any]:
         if self._check_already_existing_instrument():
-            return {"non_field_errors": "An instrument already exists with the proposed identifier"}
+            return {"non_field_errors": [_("An instrument already exists with the proposed identifier")]}
 
     @classmethod
     def get_representation_endpoint(cls) -> str:

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/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:

{wbfdm-1.49.0.dist-info → wbfdm-1.49.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbfdm
-Version: 1.49.0
+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.0.dist-info → wbfdm-1.49.2.dist-info}/RECORD

@@ -38,7 +38,7 @@ wbfdm/backends/dto.py,sha256=5IdeGVsrk8trEi9rqtq-zisqiEDE_VLBP8RxlfZZnjk,596
 wbfdm/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbfdm/contrib/dsws/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbfdm/contrib/dsws/client.py,sha256=C598w5P85kvX7_gR0Yf-WHRLj77HqztUhofBMl3JWAE,14508
-wbfdm/contrib/dsws/dataloaders/market_data.py,sha256=TVL0Vzmg8hDzjbqBWx1342dvT_qC5gVp8E4i3hVcivA,6003
+wbfdm/contrib/dsws/dataloaders/market_data.py,sha256=7Vj64nLqJAA3BiNeJPr3uLnnZ5kFOY2XGyAAJRL7vOk,6097
 wbfdm/contrib/internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbfdm/contrib/internal/dataloaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbfdm/contrib/internal/dataloaders/market_data.py,sha256=aoWSMamtnMbejYZrHzr37dQCG40nuA3fNBQNOyqhceQ,3260
@@ -230,11 +230,11 @@ wbfdm/models/instruments/classifications.py,sha256=EeZ_P8f1F1NfjUmLf9fDMF0iPM73q
 wbfdm/models/instruments/instrument_lists.py,sha256=GxfFyfYxEcJS36LAarHja49TOM8ffhIivpZX2-tPtZg,4234
 wbfdm/models/instruments/instrument_prices.py,sha256=WCaFMZ9sF7xeGTp9jkq7TA8RfYHrwvTjujzqgCKj9SI,22070
 wbfdm/models/instruments/instrument_relationships.py,sha256=zpCZCnt5CqIg5bd6le_6TyirsSwGV2NaqTVKw3bd5vM,10660
-wbfdm/models/instruments/instrument_requests.py,sha256=wC07CEqQxMGfrnATyr56T6GWxXknai105G2WIw1i3xk,7767
-wbfdm/models/instruments/instruments.py,sha256=jbk3rsTIddstWi6DfFPt_UHtthQb0BUtOZUwA2gocJA,39573
+wbfdm/models/instruments/instrument_requests.py,sha256=XbpofRS8WHadHlTFjvXJyd0o7K9r2pzJtnpjVQZOLdI,7832
+wbfdm/models/instruments/instruments.py,sha256=T0Ey5tMbUnA8DMroY1c959hOL9bI-CwTZnq5XhxCmi4,39934
 wbfdm/models/instruments/options.py,sha256=hFprq7B5t4ctz8nVqzFsBEzftq_KDUSsSXl1zJyh7tE,7094
 wbfdm/models/instruments/private_equities.py,sha256=uzwZi8IkmCKAHVTxnuFya9tehx7kh57sTlTEi1ieDaM,2198
-wbfdm/models/instruments/querysets.py,sha256=c5Op5VxR0A0uRPGHmnLJ6b7d2X5Kb5IX5bnk7MUfTjA,6475
+wbfdm/models/instruments/querysets.py,sha256=gL-uC5MfsX58RInK64Pok49tqeZhZY7xKPvPiwmraSQ,6732
 wbfdm/models/instruments/utils.py,sha256=88jnWINSSC0OwH-mCEOPLZXuhBCtEsxBpSaZ38GteaE,1365
 wbfdm/models/instruments/llm/__init__.py,sha256=dSmxRmEWb0A4O_lUoWuRKt2mBtUuLCTPVVJqGyi_n40,52
 wbfdm/models/instruments/llm/create_instrument_news_relationships.py,sha256=Eza39rlkNJxpURroIsJLImKC6F-KtTmkdjHn1kv4F3Q,3439
@@ -352,6 +352,6 @@ wbfdm/viewsets/statements/__init__.py,sha256=odxtFYUDICPmz8WCE3nx93EvKZLSPBEI4d7
 wbfdm/viewsets/statements/statements.py,sha256=kmtM0uZ3f7eJJe5gVmd-iVra9dHwTB9x12p7f5qTEx8,4084
 wbfdm/viewsets/technical_analysis/__init__.py,sha256=qtCIBg0uSiZeJq_1tEQFilnorMBkMe6uCMfqar6-cLE,77
 wbfdm/viewsets/technical_analysis/monthly_performances.py,sha256=O1j8CGfOranL74LqVvcf7jERaDIboEJZiBf_AbbVDQ8,3974
-wbfdm-1.49.0.dist-info/METADATA,sha256=C8Y0mUVZ76A9TxM7gsYtL6k28hNfD4qHqnQWytjJyAI,737
-wbfdm-1.49.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-wbfdm-1.49.0.dist-info/RECORD,,
+wbfdm-1.49.2.dist-info/METADATA,sha256=L1Z7pyHHJyzLnwLBcKxunvp7Cj0lWXdw58lcNLFDF4o,737
+wbfdm-1.49.2.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+wbfdm-1.49.2.dist-info/RECORD,,