wbfdm 1.44.5__py2.py3-none-any.whl → 1.45.1__py2.py3-none-any.whl

wbfdm/admin/classifications.py

@@ -1,5 +1,6 @@
 from django.contrib import admin
 from reversion_compare.admin import CompareVersionAdmin
+
 from wbfdm.models import Classification, ClassificationGroup
 
 

wbfdm/admin/esg.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+
 from wbfdm.models import Controversy
 
 

wbfdm/admin/exchanges.py

@@ -1,5 +1,6 @@
 from django import forms
 from django.contrib import admin
+
 from wbfdm.models import Exchange
 
 

wbfdm/admin/instrument_lists.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+
 from wbfdm.models import InstrumentList, InstrumentListThroughModel
 
 

wbfdm/admin/instrument_prices.py

@@ -1,6 +1,7 @@
 import pandas as pd
 from django.contrib import admin
 from wbcore.admin import ExportCsvMixin, ImportCsvMixin
+
 from wbfdm.models import Instrument, InstrumentPrice
 
 

wbfdm/admin/instrument_requests.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+
 from wbfdm.models import InstrumentRequest
 
 

wbfdm/admin/instruments.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+
 from wbfdm.models import Instrument, InstrumentType
 from wbfdm.models.instruments.instruments import import_prices_as_task
 

wbfdm/admin/instruments_relationships.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+
 from wbfdm.models import (
     InstrumentClassificationThroughModel,
     InstrumentFavoriteGroup,

wbfdm/admin/options.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+
 from wbfdm.models import Option, OptionAggregate
 
 

wbfdm/analysis/esg/enums.py

@@ -1,4 +1,5 @@
 from wbcore.utils.enum import ChoiceEnum
+
 from wbfdm.enums import ESG
 
 from .utils import get_esg_df
@@ -52,9 +53,7 @@ class ESGAggregation(ChoiceEnum):
             "UNADJUSTED_GENDER_PAY_GAP": AggregationMethod.WEIGHTED_AVG_NORMALIZED,
             "BOARD_GENDER_DIVERSITY": AggregationMethod.WEIGHTED_AVG_NORMALIZED,
             "CONTROVERSIAL_WEAPONS_EXPOSURE": AggregationMethod.PERCENTAGE_SUM,
-        }.get(
-            self.name, AggregationMethod.WEIGHTED_AVG_NORMALIZED
-        )  # default to weighted avg normalized
+        }.get(self.name, AggregationMethod.WEIGHTED_AVG_NORMALIZED)  # default to weighted avg normalized
 
     def get_esg_code(self) -> ESG:
         return {

wbfdm/analysis/esg/esg_analysis.py

@@ -5,6 +5,7 @@ import numpy as np
 import pandas as pd
 from django.utils.functional import cached_property
 from wbcore.contrib.currency.models import Currency, CurrencyFXRates
+
 from wbfdm.enums import ESG
 from wbfdm.models import Instrument
 

wbfdm/analysis/esg/utils.py

@@ -1,4 +1,5 @@
 import pandas as pd
+
 from wbfdm.enums import ESG
 from wbfdm.models.instruments.instruments import InstrumentQuerySet
 

wbfdm/analysis/financial_analysis/financial_metric_analysis.py

@@ -1,6 +1,7 @@
 from datetime import date
 
 import pandas as pd
+
 from wbfdm.enums import Financial, PeriodType, SeriesType
 from wbfdm.models import Instrument
 from wbfdm.utils import rename_period_index_level_to_repr

wbfdm/analysis/financial_analysis/financial_ratio_analysis.py

@@ -3,6 +3,7 @@ from datetime import date
 
 import pandas as pd
 from django.db.models import TextChoices
+
 from wbfdm.enums import Financial, MarketData, PeriodType
 from wbfdm.models import Instrument
 

wbfdm/analysis/financial_analysis/financial_statistics_analysis.py

@@ -3,6 +3,7 @@ from math import pow
 
 import numpy as np
 import pandas as pd
+from dateutil.relativedelta import relativedelta
 
 
 class FinancialStatistics:
@@ -241,7 +242,9 @@ class FinancialStatistics:
     def get_annualized_daily_volatility(self) -> float:
         """Computed the annualized volatility"""
         df = self.compute_performance(freq=1)
-        df = df.loc[df.index[-1].replace(year=df.index[-1].year - 1) :]
+        last_date = df.index[-1]
+        last_year = last_date - relativedelta(months=12)
+        df = df.loc[df.index >= last_year]
         return df.std()
 
     def prepare_daily_returns_instrument_vs_benchmark(self, benchmark_prices_df: pd.Series) -> pd.DataFrame:

wbfdm/analysis/financial_analysis/statement_with_estimates.py

@@ -1,4 +1,5 @@
 import pandas as pd
+
 from wbfdm.enums import CalendarType, Financial, MarketData
 from wbfdm.models import Instrument
 
@@ -334,10 +335,8 @@ class StatementWithEstimates:
                     * 100,
                     quarterly_df.get(Financial.EBIT.value, quarterly_empty_series).rolling(4, min_periods=1).sum()
                     / (
-                        (
-                            quarterly_df.get(Financial.TOTAL_ASSETS.value, quarterly_empty_series)
-                            - quarterly_df.get(Financial.CURRENT_LIABILITIES.value, quarterly_empty_series)
-                        )
+                        quarterly_df.get(Financial.TOTAL_ASSETS.value, quarterly_empty_series)
+                        - quarterly_df.get(Financial.CURRENT_LIABILITIES.value, quarterly_empty_series)
                     )
                     * 100,
                 ],

wbfdm/analysis/financial_analysis/utils.py

@@ -8,6 +8,7 @@ from wbcore.pandas.utils import (
     override_number_to_x,
     override_number_with_currency,
 )
+
 from wbfdm.enums import CalendarType, Financial, MarketData, PeriodType
 from wbfdm.models import Instrument
 from wbfdm.utils import rename_period_index_level_to_repr

wbfdm/analysis/technical_analysis/technical_analysis.py

@@ -4,6 +4,7 @@ from typing import TYPE_CHECKING, Literal
 import pandas as pd
 from pandas.tseries.offsets import BDay
 from stockstats import StockDataFrame
+
 from wbfdm.analysis.technical_analysis.traces import TechnicalAnalysisTraceFactory
 from wbfdm.enums import MarketData
 

wbfdm/contrib/dsws/client.py

@@ -1,14 +1,29 @@
 import math
 import re
-from datetime import date
+from datetime import date, datetime
 from typing import Generator, List, Optional
 
 import DatastreamPy as dsweb
 import numpy as np
 import pandas as pd
+import pytz
+from django.core.cache import cache
+from django.utils import timezone
 from wbcore.contrib.currency.models import Currency, CurrencyFXRates
 
 
+class CachedTokenDataClient(dsweb.DataClient):
+    def _get_token(self, isProxy=False):
+        if (token := cache.get("dsws_token")) and (token_expiry := cache.get("dsws_token_expiry")):
+            self.token = token
+            self.tokenExpiry = timezone.make_aware(datetime.fromtimestamp(token_expiry), timezone=pytz.UTC)
+        else:
+            super()._get_token()
+            expiry_elapse = (self.tokenExpiry - timezone.now()).seconds
+            cache.set("dsws_token", self.token, timeout=expiry_elapse)
+            cache.set("dsws_token_expiry", datetime.timestamp(self.tokenExpiry), timeout=expiry_elapse)
+
+
 class Client:
     MAXIMUM_ITEMS_PER_BUNDLE: int = 500
     MAXIMUM_REQUESTS_PER_BUNDLE: int = 20
@@ -19,7 +34,7 @@ class Client:
     IBUNIT_DEFAULT_UNIT: float = 1e6
 
     def __init__(self, username: Optional[str] = None, password: Optional[str] = None):
-        self.connection = dsweb.DataClient(None, username, password)
+        self.connection = CachedTokenDataClient(None, username, password)
         super().__init__()
 
     @classmethod

wbfdm/contrib/internal/dataloaders/market_data.py

@@ -3,6 +3,7 @@ from decimal import Decimal
 from typing import Iterator
 
 from wbcore.contrib.dataloader.dataloaders import Dataloader
+
 from wbfdm.dataloaders.protocols import MarketDataProtocol
 from wbfdm.dataloaders.types import MarketDataDict
 from wbfdm.enums import MarketData

wbfdm/contrib/metric/admin/instruments.py

@@ -1,4 +1,5 @@
 from django.contrib import admin
+
 from wbfdm.admin.instruments import InstrumentModelAdmin as BaseInstrumentModelAdmin
 from wbfdm.models.instruments import Instrument
 

wbfdm/contrib/metric/backends/base.py

@@ -9,6 +9,7 @@ from django.db.models import F, Model, QuerySet, Value
 from rest_framework.serializers import Field
 from wbcore import serializers as wb_serializers
 from wbcore.contrib.currency.models import CurrencyFXRates
+
 from wbfdm.models import Instrument, InstrumentPrice
 
 from ..dto import Metric, MetricField, MetricKey
@@ -65,10 +66,10 @@ class AbstractBackend(Generic[T]):
                 }
             )
             for extra_subfield in metric_key.extra_subfields:
-                fields[
-                    f"{metric_key.key}_{extra_subfield.key}"
-                ] = wb_serializers.ModelSerializer.serializer_field_mapping[extra_subfield.field_type](
-                    read_only=True, label=extra_subfield.label, **self.get_serializer_field_attr(extra_subfield)
+                fields[f"{metric_key.key}_{extra_subfield.key}"] = (
+                    wb_serializers.ModelSerializer.serializer_field_mapping[
+                        extra_subfield.field_type
+                    ](read_only=True, label=extra_subfield.label, **self.get_serializer_field_attr(extra_subfield))
                 )
         return fields
 

wbfdm/contrib/metric/backends/performances.py

@@ -6,6 +6,7 @@ import numpy as np
 import pandas as pd
 from dateutil.relativedelta import relativedelta
 from django.db.models import Avg, BooleanField, DateField, F, QuerySet
+
 from wbfdm.models import Instrument, InstrumentPrice, RelatedInstrumentThroughModel
 
 from ..decorators import register
@@ -252,9 +253,9 @@ class InstrumentPerformanceMetricBackend(InstrumentMetricBaseBackend):
 
         with suppress(KeyError):
             start_date = Dataloader.get_performance_date_map(pivot_date)[metric_field.key.replace("previous_", "")]
-            attrs[
-                "help_text"
-            ] = f"The {metric_field.label} performance is computed from {start_date:%Y-%m-%d} to {pivot_date:%Y-%m-%d}"
+            attrs["help_text"] = (
+                f"The {metric_field.label} performance is computed from {start_date:%Y-%m-%d} to {pivot_date:%Y-%m-%d}"
+            )
         return attrs
 
 

wbfdm/contrib/metric/backends/statistics.py

@@ -5,6 +5,7 @@ from typing import Generator
 import pandas as pd
 from django.db.models import Avg, Sum
 from wbcore.serializers.fields.number import DisplayMode
+
 from wbfdm.enums import Financial, PeriodType, SeriesType
 from wbfdm.models import Instrument, InstrumentPrice
 

wbfdm/contrib/metric/factories.py

@@ -1,6 +1,7 @@
 import factory
 from django.contrib.contenttypes.models import ContentType
 from faker import Faker
+
 from wbfdm.contrib.metric.models import InstrumentMetric
 from wbfdm.factories import InstrumentFactory
 from wbfdm.models import Instrument

wbfdm/contrib/metric/filters.py

@@ -1,6 +1,7 @@
 from django.contrib.contenttypes.models import ContentType
 from django.utils.translation import gettext as _
 from wbcore import filters
+
 from wbfdm.contrib.metric.models import InstrumentMetric
 
 from .registry import backend_registry

wbfdm/contrib/metric/serializers.py

@@ -1,6 +1,7 @@
 from rest_framework.reverse import reverse
 from wbcore import serializers as wb_serializers
 from wbcore.content_type.serializers import ContentTypeRepresentationSerializer
+
 from wbfdm.contrib.metric.models import InstrumentMetric
 from wbfdm.serializers.instruments import InstrumentRepresentationSerializer
 

wbfdm/contrib/metric/tasks.py

@@ -2,6 +2,7 @@ from datetime import date
 
 from celery import shared_task
 from django.contrib.contenttypes.models import ContentType
+
 from wbfdm.contrib.metric.orchestrators import MetricOrchestrator
 
 
@@ -11,7 +12,7 @@ def compute_metrics_as_task(
     key: str | None = None,
     basket_content_type_id: int | None = None,
     basket_id: int | None = None,
-    **kwargs
+    **kwargs,
 ):
     """
     Run the orchestrator as a async task periodically for all keys and baskets

wbfdm/contrib/metric/tests/backends/test_performances.py

@@ -6,6 +6,7 @@ import pytest
 from django.contrib.contenttypes.models import ContentType
 from faker import Faker
 from pandas.tseries.offsets import BDay
+
 from wbfdm.models import Instrument, RelatedInstrumentThroughModel
 
 fake = Faker()

wbfdm/contrib/metric/tests/backends/test_statistics.py

@@ -3,6 +3,7 @@ from unittest.mock import patch
 
 import pytest
 from faker import Faker
+
 from wbfdm.dataloaders.proxies import InstrumentDataloaderProxy
 
 fake = Faker()

wbfdm/contrib/metric/tests/test_models.py

@@ -1,5 +1,6 @@
 import pytest
 from django.contrib.contenttypes.models import ContentType
+
 from wbfdm.models import Instrument
 
 from ..dto import Metric

wbfdm/contrib/metric/tests/test_viewsets.py

@@ -5,6 +5,7 @@ from faker import Faker
 from rest_framework.test import APIRequestFactory
 from wbcore import viewsets
 from wbcore.utils.strings import get_aggregate_symbol
+
 from wbfdm.contrib.metric.backends.performances import (
     PERFORMANCE_METRIC,
     InstrumentPerformanceMetricBackend,

wbfdm/contrib/metric/viewsets/configs/display.py

@@ -9,6 +9,7 @@ from wbcore.metadata.configs.display.instance_display.shortcuts import (
 )
 from wbcore.metadata.configs.display.instance_display.utils import repeat_field
 from wbcore.metadata.configs.display.view_config import DisplayViewConfig
+
 from wbfdm.contrib.metric.dto import MetricField
 
 

wbfdm/contrib/metric/viewsets/mixins.py

@@ -6,6 +6,7 @@ from django.utils.functional import cached_property
 from wbcore import filters
 from wbcore.metadata.metadata import WBCoreMetadata
 from wbcore.utils.strings import get_aggregate_symbol
+
 from wbfdm.contrib.metric.models import InstrumentMetric
 from wbfdm.contrib.metric.registry import backend_registry
 from wbfdm.contrib.metric.viewsets.configs.display import (

wbfdm/contrib/metric/viewsets/viewsets.py

@@ -1,5 +1,6 @@
 from django.db.models import Case, Exists, F, IntegerField, OuterRef, When
 from wbcore import viewsets
+
 from wbfdm.contrib.metric.filters import InstrumentMetricFilterSet
 from wbfdm.contrib.metric.models import InstrumentMetric
 from wbfdm.contrib.metric.serializers import (

wbfdm/contrib/msci/dataloaders/esg.py

@@ -3,6 +3,7 @@ from typing import Iterator
 
 from django.conf import settings
 from wbcore.contrib.dataloader.dataloaders import Dataloader
+
 from wbfdm.dataloaders.protocols import ESGProtocol
 from wbfdm.dataloaders.types import ESGDataDict
 from wbfdm.enums import ESG

wbfdm/contrib/msci/dataloaders/esg_controversies.py

@@ -3,6 +3,7 @@ from typing import Iterator
 
 from django.conf import settings
 from wbcore.contrib.dataloader.dataloaders import Dataloader
+
 from wbfdm.dataloaders.protocols import ESGControversyProtocol
 from wbfdm.dataloaders.types import ESGControversyDataDict
 

wbfdm/contrib/msci/sync.py

@@ -1,4 +1,5 @@
 from django.db.models import Q, QuerySet
+
 from wbfdm.models import Instrument
 from wbfdm.sync.abstract import Sync
 

wbfdm/contrib/qa/dataloaders/adjustments.py

@@ -3,18 +3,18 @@ from datetime import date
 from itertools import batched
 from typing import Iterator
 
+import pypika as pk
 from django.db import ProgrammingError, connections
+from pypika import functions as fn
+from pypika.enums import Order, SqlTypes
+from pypika.terms import LiteralValue
 from wbcore.contrib.dataloader.dataloaders import Dataloader
 from wbcore.contrib.dataloader.utils import dictfetchall
+
 from wbfdm.contrib.qa.dataloaders.utils import SOURCE_DS2
 from wbfdm.dataloaders.protocols import AdjustmentsProtocol
 from wbfdm.dataloaders.types import AdjustmentDataDict
 
-import pypika as pk
-from pypika import functions as fn
-from pypika.enums import SqlTypes, Order
-from pypika.terms import LiteralValue
-
 
 class DatastreamAdjustmentsDataloader(AdjustmentsProtocol, Dataloader):
     def adjustments(self, from_date: date | None = None, to_date: date | None = None) -> Iterator[AdjustmentDataDict]:

wbfdm/contrib/qa/dataloaders/corporate_actions.py

@@ -3,18 +3,18 @@ from datetime import date
 from itertools import batched
 from typing import Iterator
 
+import pypika as pk
 from django.db import ProgrammingError, connections
+from pypika import functions as fn
+from pypika.enums import SqlTypes
+from pypika.terms import LiteralValue
 from wbcore.contrib.dataloader.dataloaders import Dataloader
 from wbcore.contrib.dataloader.utils import dictfetchall
+
 from wbfdm.contrib.qa.dataloaders.utils import SOURCE_DS2
 from wbfdm.dataloaders.protocols import CorporateActionsProtocol
 from wbfdm.dataloaders.types import CorporateActionDataDict
 
-import pypika as pk
-from pypika import functions as fn
-from pypika.enums import SqlTypes
-from pypika.terms import LiteralValue
-
 
 class DatastreamCorporateActionsDataloader(CorporateActionsProtocol, Dataloader):
     def corporate_actions(

wbfdm/contrib/qa/dataloaders/financials.py

@@ -7,6 +7,7 @@ from django.template.loader import get_template
 from jinjasql import JinjaSql  # type: ignore
 from wbcore.contrib.dataloader.dataloaders import Dataloader
 from wbcore.contrib.dataloader.utils import dictfetchall
+
 from wbfdm.dataloaders.protocols import FinancialsProtocol
 from wbfdm.dataloaders.types import FinancialDataDict
 from wbfdm.enums import (

wbfdm/contrib/qa/dataloaders/market_data.py

@@ -1,19 +1,21 @@
 from contextlib import suppress
-from functools import reduce
-from itertools import batched
 from datetime import date
 from enum import Enum
+from functools import reduce
+from itertools import batched
 from typing import Iterator
-import pypika as pk
-from pypika import Column, MSSQLQuery, functions as fn
-from pypika.enums import SqlTypes, Order
-from pypika.terms import ValueWrapper, LiteralValue
 
+import pypika as pk
 from django.db import ProgrammingError, connections
+from pypika import Column, MSSQLQuery
+from pypika import functions as fn
+from pypika.enums import Order, SqlTypes
+from pypika.terms import LiteralValue, ValueWrapper
 from wbcore.contrib.dataloader.dataloaders import Dataloader
 from wbcore.contrib.dataloader.utils import dictfetchall
-from wbfdm.dataloaders.protocols import MarketDataProtocol
+
 from wbfdm.contrib.qa.dataloaders.utils import create_table
+from wbfdm.dataloaders.protocols import MarketDataProtocol
 from wbfdm.dataloaders.types import MarketDataDict
 from wbfdm.enums import Frequency, MarketData
 

wbfdm/contrib/qa/dataloaders/officers.py

@@ -1,20 +1,18 @@
-from typing import Iterator
-
-from django.db import connections, ProgrammingError
-from wbcore.contrib.dataloader.dataloaders import Dataloader
-from wbcore.contrib.dataloader.utils import dictfetchall
-from wbfdm.dataloaders.protocols import OfficersProtocol
-from wbfdm.dataloaders.types import OfficerDataDict
-from itertools import batched
 from contextlib import suppress
+from itertools import batched
+from typing import Iterator
 
 import pypika as pk
+from django.db import ProgrammingError, connections
 from pypika import functions as fn
+from pypika.analytics import RowNumber
 from pypika.enums import SqlTypes
 from pypika.terms import LiteralValue
+from wbcore.contrib.dataloader.dataloaders import Dataloader
+from wbcore.contrib.dataloader.utils import dictfetchall
 
-
-from pypika.analytics import RowNumber
+from wbfdm.dataloaders.protocols import OfficersProtocol
+from wbfdm.dataloaders.types import OfficerDataDict
 
 
 class RKDOfficersDataloader(OfficersProtocol, Dataloader):

wbfdm/contrib/qa/dataloaders/reporting_dates.py

@@ -4,6 +4,7 @@ from django.db import connections
 from jinjasql import JinjaSql
 from wbcore.contrib.dataloader.dataloaders import Dataloader
 from wbcore.contrib.dataloader.utils import dictfetchall
+
 from wbfdm.dataloaders.protocols import ReportDateProtocol
 from wbfdm.dataloaders.types import ReportDateDataDict
 

wbfdm/contrib/qa/dataloaders/statements.py

@@ -6,6 +6,7 @@ from django.db import connections
 from jinjasql import JinjaSql  # type: ignore
 from wbcore.contrib.dataloader.dataloaders import Dataloader
 from wbcore.contrib.dataloader.utils import dictfetchall
+
 from wbfdm.dataloaders.protocols import StatementsProtocol
 from wbfdm.dataloaders.types import StatementDataDict
 from wbfdm.enums import DataType, Financial, PeriodType, StatementType

wbfdm/contrib/qa/dataloaders/utils.py

@@ -1,11 +1,11 @@
 from typing import TYPE_CHECKING
 
-from pypika.terms import ValueWrapper
 from pypika import Column, MSSQLQuery, Table
+from pypika.terms import ValueWrapper
 
 if TYPE_CHECKING:
-    from pypika.terms import Term
     from pypika.queries import CreateQueryBuilder
+    from pypika.terms import Term
 
 
 def compile_source(source: str) -> "Term":

wbfdm/dataloaders/proxies.py

@@ -2,6 +2,7 @@ from datetime import date
 from typing import Iterator
 
 from wbcore.contrib.dataloader.dataloaders import DataloaderProxy
+
 from wbfdm.dataloaders.protocols import (
     AdjustmentsProtocol,
     CorporateActionsProtocol,

wbfdm/factories/classifications.py

@@ -1,5 +1,6 @@
 import factory
 import rstr
+
 from wbfdm.models import (
     Classification,
     ClassificationGroup,

wbfdm/factories/controversies.py

@@ -1,4 +1,5 @@
 import factory
+
 from wbfdm.enums import (
     ESGControveryFlag,
     ESGControverySeverity,

wbfdm/factories/exchanges.py

@@ -1,5 +1,6 @@
 import factory
 import rstr
+
 from wbfdm.models import Exchange
 
 

wbfdm/factories/instrument_list.py

@@ -1,5 +1,6 @@
 import factory
 from slugify import slugify
+
 from wbfdm.models import InstrumentList, InstrumentListThroughModel
 
 

wbfdm/factories/instrument_prices.py

@@ -5,6 +5,7 @@ from decimal import Decimal
 import factory
 from pandas.tseries.offsets import BDay
 from wbcore.contrib.currency.models import CurrencyFXRates
+
 from wbfdm.models import InstrumentPrice
 
 

wbfdm/factories/instruments.py

@@ -1,6 +1,7 @@
 import factory
 import rstr
 from slugify import slugify
+
 from wbfdm.models import Instrument, InstrumentType
 
 

wbfdm/factories/instruments_relationships.py

@@ -1,4 +1,5 @@
 import factory
+
 from wbfdm.models import InstrumentFavoriteGroup, RelatedInstrumentThroughModel
 
 

wbfdm/factories/options.py

@@ -3,6 +3,7 @@ from datetime import timedelta
 import factory
 from faker import Faker
 from pandas.tseries.offsets import BDay
+
 from wbfdm.models import Option, OptionAggregate
 
 fake = Faker()