wbfdm 1.54.8__py2.py3-none-any.whl → 1.54.9__py2.py3-none-any.whl

wbfdm/contrib/dsws/dataloaders/market_data.py

@@ -5,6 +5,7 @@ from typing import Iterator
 from DatastreamPy import DSUserObjectFault
 from django.conf import settings
 from pandas.tseries.offsets import BDay
+from wbcore.contrib.currency.models import Currency, CurrencyFXRates
 from wbcore.contrib.dataloader.dataloaders import Dataloader
 from wbfdm.dataloaders.protocols import MarketDataProtocol
 from wbfdm.dataloaders.types import MarketDataDict
@@ -34,6 +35,8 @@ class DSWSMarketDataDataloader(MarketDataProtocol, Dataloader):
         to_date: date | None = None,
         exact_date: date | None = None,
         frequency: Frequency = Frequency.DAILY,
+        target_currency: str | None = None,
+        apply_fx_rate: bool = True,
         **kwargs,
     ) -> Iterator[MarketDataDict]:
         """Get prices for instruments.
@@ -55,6 +58,12 @@ class DSWSMarketDataDataloader(MarketDataProtocol, Dataloader):
                 "id",
             )
         }
+        instruments = {entity.id: entity for entity in self.entities}
+        try:
+            target_currency = Currency.objects.get(key=target_currency)
+        except Currency.DoesNotExist:
+            target_currency = None
+
         if (dsws_username := getattr(settings, "REFINITIV_DATASTREAM_USERNAME", None)) and (
             dsws_password := getattr(settings, "REFINITIV_DATASTREAM_PASSWORD", None)
         ):
@@ -80,13 +89,31 @@ class DSWSMarketDataDataloader(MarketDataProtocol, Dataloader):
                     for row in df.to_dict("records"):
                         jsondate = row["Dates"].date()
                         external_id = row["Instrument"]
-
-                        data = {}
+                        fx_rate = 1.0
+                        if target_currency:
+                            instrument = instruments[default_lookup[external_id]["id"]]
+                            if instrument.currency != target_currency:
+                                with suppress(CurrencyFXRates.DoesNotExist):
+                                    fx_rate = instrument.currency.convert(jsondate, target_currency)
+                        data = dict(fx_rate=fx_rate)
                         for market_value in values:
                             data[market_value.value] = row.get(FIELD_MAP[market_value.value], None)
+                            if (
+                                apply_fx_rate
+                                and data[market_value.value]
+                                and market_value.value
+                                not in [
+                                    MarketData.MARKET_CAPITALIZATION.value,
+                                    MarketData.VOLUME.value,
+                                    MarketData.VWAP.value,
+                                ]
+                            ):
+                                data[market_value.value] *= fx_rate
+
                         with suppress(KeyError):
                             if default_symbol := default_lookup[external_id].get("symbol", None):
                                 data["close"] = row[default_symbol]
+
                             yield MarketDataDict(
                                 id=f"{default_lookup[external_id]['id']}_{jsondate}",
                                 valuation_date=jsondate,

wbfdm/contrib/qa/dataloaders/market_data.py

@@ -98,7 +98,9 @@ class DatastreamMarketDataDataloader(MarketDataProtocol, Dataloader):
             fx_code = pk.Table("DS2FxCode")
             fx_rate = pk.Table("DS2FxRate")
             query = (
-                query.select((1 / fx_rate.midrate).as_("fx_rate"))
+                query.select(
+                    (Case().when(pricing.Currency == target_currency, 1).else_(1 / fx_rate.midrate)).as_("fx_rate")
+                )
                 # Join FX code table matching currencies and ensuring SPOT rate type
                 .left_join(fx_code)
                 .on(

{wbfdm-1.54.8.dist-info → wbfdm-1.54.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbfdm
-Version: 1.54.8
+Version: 1.54.9
 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.54.8.dist-info → wbfdm-1.54.9.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=7Vj64nLqJAA3BiNeJPr3uLnnZ5kFOY2XGyAAJRL7vOk,6097
+wbfdm/contrib/dsws/dataloaders/market_data.py,sha256=eup82Tpa6JRzIsKcMr6L33wEPVPwxH1KvjfMijd2mO4,7479
 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=FA7_f_20Gx1EjWOK9HDcWfX_gbBpOYXaxMR_O5T-K-Q,4542
@@ -100,7 +100,7 @@ wbfdm/contrib/qa/dataloaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 wbfdm/contrib/qa/dataloaders/adjustments.py,sha256=DQEexOLA7WyBB1dZJHQd-6zbzyEIURgSSgS7bJRvXzQ,2980
 wbfdm/contrib/qa/dataloaders/corporate_actions.py,sha256=lWT6klrTXKqxiko2HGrxHH8E2C00FJS-AOX3IhglRrI,2912
 wbfdm/contrib/qa/dataloaders/financials.py,sha256=xUHpvhUkvmdPL_RyWCrs7XgChgTklX5qemmaXMedgkY,3475
-wbfdm/contrib/qa/dataloaders/market_data.py,sha256=0LjTN04WM2i9rrJ_M3M04cY0UbxMElsWiaqOC75xpHQ,8221
+wbfdm/contrib/qa/dataloaders/market_data.py,sha256=ei_XNzgMOU-X8dCR06xQGASboCmw05-Hwgn1t17n5ME,8318
 wbfdm/contrib/qa/dataloaders/officers.py,sha256=vytlQJJxmn4Y5HfNh5mHJAvuIrrsQSkNO-sONyhxftY,2940
 wbfdm/contrib/qa/dataloaders/reporting_dates.py,sha256=q25ccB0pbGfLJLV1A1_AY1XYWJ_Fa10egY09L1J-C5A,2628
 wbfdm/contrib/qa/dataloaders/statements.py,sha256=hC6YErJcvBTmaAmzscgeC4sBK3lYE2U5eIKRIE9b_cs,10094
@@ -359,6 +359,6 @@ wbfdm/viewsets/statements/__init__.py,sha256=odxtFYUDICPmz8WCE3nx93EvKZLSPBEI4d7
 wbfdm/viewsets/statements/statements.py,sha256=gA6RCI8-B__JwjEb6OZxpn8Y-9aF-YQ3HIQ7e1vfJMw,4304
 wbfdm/viewsets/technical_analysis/__init__.py,sha256=qtCIBg0uSiZeJq_1tEQFilnorMBkMe6uCMfqar6-cLE,77
 wbfdm/viewsets/technical_analysis/monthly_performances.py,sha256=O1j8CGfOranL74LqVvcf7jERaDIboEJZiBf_AbbVDQ8,3974
-wbfdm-1.54.8.dist-info/METADATA,sha256=76MaKZO9aTh9b4Q6HavxQ-XD7hqH7GeDXmw9Xs65kv0,768
-wbfdm-1.54.8.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-wbfdm-1.54.8.dist-info/RECORD,,
+wbfdm-1.54.9.dist-info/METADATA,sha256=HpfktLlCPZBteZWL7Ta3TmLEZArU9BIja0IK2iKPKpo,768
+wbfdm-1.54.9.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+wbfdm-1.54.9.dist-info/RECORD,,