wbportfolio 1.55.3__py2.py3-none-any.whl → 1.55.5__py2.py3-none-any.whl

wbportfolio/analysis/claims.py

@@ -58,13 +58,12 @@ class ConsolidatedTradeSummary:
         self.pivot_label = pivot_label
         self.queryset = self.queryset.annotate(
             internal_trade=F("trade__marked_as_internal"),
-            date_considered=ExpressionWrapper(
-                Greatest("trade__transaction_date", "date") + 1, output_field=DateField()
-            ),
+            valid_date=Greatest("trade__transaction_date", "date"),
+            date_considered=ExpressionWrapper(F("valid_date") + 1, output_field=DateField()),
             net_value=InstrumentPrice.subquery_closest_value(
-                "net_value", date_name="date_considered", instrument_pk_name="product__pk", date_lookup="exact"
+                "net_value", date_name="valid_date", instrument_pk_name="product__pk", date_lookup="exact"
             ),
-            fx_rate=CurrencyFXRates.get_fx_rates_subquery("date_considered", lookup_expr="exact"),
+            fx_rate=CurrencyFXRates.get_fx_rates_subquery("valid_date", lookup_expr="exact"),
             aum=ExpressionWrapper(F("fx_rate") * F("net_value") * F("shares"), output_field=DecimalField()),
         )
         self.queryset = Account.annotate_root_account_info(self.queryset)
@@ -158,8 +157,8 @@ class ConsolidatedTradeSummary:
         df = df.fillna(0)
         # Prepare dataframe
         df["sum_aum_start"] = df.sum_shares_start * df.price_start
-        df["sum_aum_end"] = df.sum_shares_end * df.price_end
 
+        df["sum_aum_end"] = df.sum_shares_end * df.price_end
         # Sanitize dataframe
         df = df.drop(
             columns=df.columns.difference(["id", "sum_shares_start", "sum_shares_end", "sum_aum_start", "sum_aum_end"])

wbportfolio/filters/assets.py

@@ -127,7 +127,7 @@ class AssetPositionFilter(wb_filters.FilterSet):
     )
 
     underlying_instrument__is_cash = wb_filters.BooleanFilter(
-        label="With Cash", exclude=True, lookup_expr="exact", field_name="underlying_instrument__is_cash"
+        label="Cash", lookup_expr="exact", method="filter_is_cash"
     )
     underlying_instrument__instrument_type = wb_filters.ModelChoiceFilter(
         label="Instrument Type",
@@ -230,6 +230,13 @@ class AssetPositionFilter(wb_filters.FilterSet):
             return queryset.filter(underlying_instrument__in=value.get_classified_instruments())
         return queryset
 
+    def filter_is_cash(self, queryset, name, value):
+        if value is True:
+            return queryset.filter(Q(underlying_quote__is_cash=True) | Q(underlying_quote__is_cash_equivalent=True))
+        if value is False:
+            return queryset.filter(Q(underlying_quote__is_cash=False) & Q(underlying_quote__is_cash_equivalent=False))
+        return queryset
+
     def filter_portfolio_instrument(self, queryset, name, value):
         if value:
             return queryset.filter(

wbportfolio/import_export/handlers/asset_position.py

@@ -9,6 +9,7 @@ from django.core.exceptions import ObjectDoesNotExist
 from django.db import models
 from wbcore.contrib.authentication.authentication import User
 from wbcore.contrib.currency.import_export.handlers import CurrencyImportHandler
+from wbcore.contrib.currency.models import Currency
 from wbcore.contrib.io.exceptions import DeserializationError
 from wbcore.contrib.io.imports import ImportExportHandler
 from wbcore.contrib.notifications.dispatch import send_notification
@@ -36,9 +37,13 @@ class AssetPositionImportHandler(ImportExportHandler):
         portfolio_data = data.pop("portfolio", None)
         underlying_quote_data = data.pop("underlying_quote", data.pop("underlying_instrument", None))
         if "currency" in data:
-            data["currency"] = self.currency_handler.process_object(
-                data["currency"], read_only=True, raise_exception=False
-            )[0]
+            currency = self.currency_handler.process_object(data["currency"], read_only=True, raise_exception=False)[0]
+            # we do not support GBX in our instrument table
+            if currency.key == "GBX":
+                currency = Currency.objects.get(key="GBP")
+                if "initial_price" in data:
+                    data["initial_price"] /= 1000
+            data["currency"] = currency
         data["date"] = datetime.strptime(data["date"], "%Y-%m-%d").date()
         if data.get("asset_valuation_date", None):
             data["asset_valuation_date"] = datetime.strptime(data["asset_valuation_date"], "%Y-%m-%d").date()

wbportfolio/import_export/parsers/sg_lux/equity.py

@@ -82,7 +82,7 @@ def parse(import_source):
     cash = (
         df.loc[
             cash_mask,
-            ["currency__key", "initial_currency_fx_rate", "portfolio", "date", "initial_price", "initial_shares"],
+            ["currency__key", "initial_currency_fx_rate", "portfolio", "date", "initial_shares"],
         ]
         .groupby(
             [
@@ -94,21 +94,20 @@ def parse(import_source):
         .agg(
             {
                 "initial_currency_fx_rate": "mean",
-                "initial_price": "mean",
                 "initial_shares": "sum",
             }
         )
         .reset_index()
     ).copy()
     cash["underlying_quote"] = cash["currency__key"].apply(lambda x: {"currency__key": x, "instrument_type": "cash"})
-
+    cash["initial_price"] = 1.0
     # cash_equivalents_mask, all asset type code that match TRES and which don't have accounting category T111
     cash_equivalents_mask = df["Asset type code"].str.match("TRES") & ~df["Accounting category"].str.match("T111")
 
     cash_equivalents = (
         df.loc[
             cash_equivalents_mask,
-            ["currency__key", "initial_currency_fx_rate", "portfolio", "date", "initial_price", "initial_shares"],
+            ["currency__key", "initial_currency_fx_rate", "portfolio", "date", "initial_shares"],
         ]
         .groupby(
             [
@@ -120,7 +119,6 @@ def parse(import_source):
         .agg(
             {
                 "initial_currency_fx_rate": "mean",
-                "initial_price": "mean",
                 "initial_shares": "sum",
             }
         )
@@ -129,9 +127,11 @@ def parse(import_source):
     cash_equivalents["underlying_quote"] = cash_equivalents["currency__key"].apply(
         lambda x: {"currency__key": x, "instrument_type": "cash_equivalent"}
     )
+    cash_equivalents["initial_price"] = 1.0
 
     # equities = df.loc[df["Accounting category"].str.match("010"), :].copy() # Historically, we filter out equity base on the "accounting category" that matches "010", we had issue with equity having the code "020". We decided to use the column "Asset type code" and filter out with the code "VCOM"
     equities = df.loc[df["Asset type code"].str.match("VMOB"), :].copy()
+
     if not equities.empty:
         equities["underlying_quote"] = equities.apply(lambda x: get_underlying_quote(x), axis=1)
         equities["exchange"] = equities.apply(lambda x: get_exchange(x), axis=1)
@@ -140,7 +140,6 @@ def parse(import_source):
         del equities["Bloom Ticker"]
         del equities["Asset description"]
     df = pd.concat([cash, equities, cash_equivalents])
-
     df["asset_valuation_date"] = df["date"]
     # Rename the columns
 

wbportfolio/tests/analysis/test_claims.py

@@ -0,0 +1,85 @@
+from datetime import date, timedelta
+
+import pytest
+from pandas._libs.tslibs.offsets import BDay
+
+from wbportfolio.analysis.claims import ConsolidatedTradeSummary
+from wbportfolio.models import Claim
+
+
+@pytest.mark.django_db
+class TestConsolidatedTradeSummary:
+    def test_base(self, claim_factory, customer_trade_factory, product, instrument_price_factory):
+        d1 = date(2024, 1, 1)
+        d2 = date(2025, 1, 1)
+        p1 = instrument_price_factory.create(date=d1, instrument=product)  # noqa
+        p2 = instrument_price_factory.create(date=d2, instrument=product)
+        t1 = customer_trade_factory.create(
+            shares=1000, transaction_date=d1, underlying_instrument=product, portfolio=product.primary_portfolio
+        )
+        t2 = customer_trade_factory.create(
+            shares=500, transaction_date=d2, underlying_instrument=product, portfolio=product.primary_portfolio
+        )
+
+        c0 = claim_factory.create()  # noqa
+        c1 = claim_factory.create(shares=1000, status="APPROVED", trade=t1, date=d2, product=product)
+        c2 = claim_factory.create(shares=500, status="APPROVED", trade=t2, date=d1, product=product)
+        cts = ConsolidatedTradeSummary(Claim.objects.all(), d1, d2, "account", "account__title")
+
+        valid_date = dict(cts.queryset.values_list("id", "valid_date"))
+        assert valid_date == {c1.id: d2, c2.id: d2}
+
+        date_considered = dict(cts.queryset.values_list("id", "date_considered"))
+        assert date_considered == {c1.id: d2 + timedelta(days=1), c2.id: d2 + timedelta(days=1)}
+
+        aum = dict(cts.queryset.values_list("id", "aum"))
+        assert aum == {c1.id: p2.net_value * c1.shares, c2.id: p2.net_value * c2.shares}
+
+    def test_get_aum_df(self, claim_factory, product, instrument_price_factory, customer_trade_factory):
+        t1 = customer_trade_factory.create(
+            shares=10,
+            transaction_date=date(2024, 12, 31),
+            underlying_instrument=product,
+            portfolio=product.primary_portfolio,
+        )
+        t2 = customer_trade_factory.create(
+            shares=100,
+            transaction_date=date(2025, 2, 3),
+            underlying_instrument=product,
+            portfolio=product.primary_portfolio,
+        )
+        c1 = claim_factory.create(trade=t1, status="APPROVED")
+        c2 = claim_factory.create(trade=t2, status="APPROVED", account=c1.account)
+        d1 = date(2025, 1, 1)
+        d2 = date(2025, 3, 3)
+        product.prices.all().delete()
+        p1 = instrument_price_factory.create(
+            date=(d1 - BDay(7)).date(), instrument=product
+        )  # we test that the CTS uses the earliest date
+        p2 = instrument_price_factory.create(date=c2.date, instrument=product)  # noqa
+        p3 = instrument_price_factory.create(date=d2, instrument=product)
+
+        cts = ConsolidatedTradeSummary(Claim.objects.all(), d1, d2, "account", "account__title")
+        aum_start = float(round(p1.net_value * c1.shares))
+        aum_end = float(round(p3.net_value * (c1.shares + c2.shares)))
+        assert cts.get_aum_df().to_dict(orient="index") == {
+            c1.account.id: {
+                "sum_shares_start": c1.shares,
+                "sum_shares_end": c1.shares + c2.shares,
+                "sum_aum_start": aum_start,
+                "sum_aum_end": aum_end,
+                "sum_shares_diff": c2.shares,
+                "sum_shares_perf": (c1.shares + c2.shares) / c1.shares - 1,
+                "sum_aum_diff": aum_end - aum_start,
+                "sum_aum_perf": aum_end / aum_start - 1,
+            }
+        }
+
+        p0 = instrument_price_factory.create(date=c1.date, instrument=product)  # noqa
+
+        assert cts.get_nnm_df().to_dict(orient="index") == {
+            c1.account.id: {
+                "sum_nnm_2025-02": float(round(c2.shares * p2.net_value)),
+                "sum_nnm_total": float(round(c2.shares * p2.net_value)),
+            }
+        }, "Ensure the first claim that is considered in t+1 (on the CTS lower bound range) is excluded from the NNM"

{wbportfolio-1.55.3.dist-info → wbportfolio-1.55.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbportfolio
-Version: 1.55.3
+Version: 1.55.5
 Author-email: Christopher Wittlinger <c.wittlinger@stainly.com>
 License-File: LICENSE
 Requires-Dist: cryptography==3.4.*

{wbportfolio-1.55.3.dist-info → wbportfolio-1.55.5.dist-info}/RECORD

@@ -28,7 +28,7 @@ wbportfolio/admin/transactions/dividends.py,sha256=2Hx0yvcMp0VvzYfuO_jl-9KFrpq6R
 wbportfolio/admin/transactions/fees.py,sha256=gpq-_k3ypEGVNuDfFE4OHtn-B9ZtkzA4YYNLtNu-Fc8,1329
 wbportfolio/admin/transactions/trades.py,sha256=LanpSm6iaR9cmNvSJwpRFPkOgN46K_Y9YoMy0MReYuc,1913
 wbportfolio/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wbportfolio/analysis/claims.py,sha256=islJVmHg6jKr6Q1b84Vt5Au2bzy9aDjWGEVGNtPFQd4,10688
+wbportfolio/analysis/claims.py,sha256=wcTniEVdDnmME8LHrYbo-WWpQ5EzNPgAT2KJIGQ0IIA,10688
 wbportfolio/api_clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbportfolio/api_clients/ubs.py,sha256=pB_urTr8x8YCW28c_hK7DeCzAx07VeLZ27TamrZln2k,6598
 wbportfolio/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -82,7 +82,7 @@ wbportfolio/factories/orders/orders.py,sha256=tTw2U-xvUsFpG4zad3KCHZPQx1FVg-zPjw
 wbportfolio/fdm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbportfolio/fdm/tasks.py,sha256=59hXt3TOJaM8KpUaKdBNAGsO7NFMHaAlNA0T3FTUau0,533
 wbportfolio/filters/__init__.py,sha256=m44CdyYMtku8rMGyiX8KSmMdvyqaElKW4UMkEqOC1VM,1374
-wbportfolio/filters/assets.py,sha256=jQoWmK2teQPBpEUNLC3NWl6fhrndl1Tt0sPdQxiPPM4,18538
+wbportfolio/filters/assets.py,sha256=rvBNtQ8OcvYXSmaVdgjH7IITvaBZ558Y9sCj0B_WDpw,18866
 wbportfolio/filters/assets_and_net_new_money_progression.py,sha256=IYl_qMFO9QKhXhCTtz2JrOjlAKdEkykNlu4DO_H7HkQ,1403
 wbportfolio/filters/custodians.py,sha256=pStQgPQPhPpnt57_V7BuXbFXmRiZBEAiEeMFuQmt2NE,287
 wbportfolio/filters/esg.py,sha256=ZEUQh3IQxLSshVrgtPqHC_ToirrllkCXR_KzUlCUtkA,688
@@ -120,7 +120,7 @@ wbportfolio/import_export/backends/wbfdm/dividend.py,sha256=iAQXnYPXmtG_Jrc8THAJ
 wbportfolio/import_export/backends/wbfdm/mixin.py,sha256=JNtjgqGLson1nu_Chqb8MWyuiF3Ws8ox2vapxIRBYKE,400
 wbportfolio/import_export/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbportfolio/import_export/handlers/adjustment.py,sha256=6bdTIYFmc8_HFxcdwtnYwglMyCfAD8XrTIrEb2zWY0g,1757
-wbportfolio/import_export/handlers/asset_position.py,sha256=trI_I4IxZ6fqV9_BpRAvPNkYf07-loQB7f4ipkGkJbE,8695
+wbportfolio/import_export/handlers/asset_position.py,sha256=EWBmuZActTmBk4PJRjVRby-99g27LH6cWLbWve9N8GM,9000
 wbportfolio/import_export/handlers/dividend.py,sha256=F0oLfNt2B_QQAjHBCRpxa5HSkfkAYdal_NjLJGtVckY,4408
 wbportfolio/import_export/handlers/fees.py,sha256=BOFHAvSTlvVLaxnm6KD_fcza1TlPc02HOR9J0_jjswI,2495
 wbportfolio/import_export/handlers/orders.py,sha256=GU3_tIy-tAw9aU-ifsnmMZPBB9sqfkFC_S1d9VziTwg,3136
@@ -161,7 +161,7 @@ wbportfolio/import_export/parsers/sg_lux/customer_trade.py,sha256=gTEUIaxlZvXXCg
 wbportfolio/import_export/parsers/sg_lux/customer_trade_pending_slk.py,sha256=WyRpKXkNhpK4H9aNjW7_ggcCGlN_-9ETlR1QsE9xl_k,5068
 wbportfolio/import_export/parsers/sg_lux/customer_trade_slk.py,sha256=D_1c0v6lTXkopjZzr9pQb575HxrMFe91tKk8xT1X3Ns,3604
 wbportfolio/import_export/parsers/sg_lux/customer_trade_without_pw.py,sha256=nMYV9-OwY_TagU36Ab0dOweZYzVriFWHXPYjauBg2Yk,2034
-wbportfolio/import_export/parsers/sg_lux/equity.py,sha256=SXupaP9W299oFfnucV7FogY_yqusLkbqQFR3nGt-4m0,6117
+wbportfolio/import_export/parsers/sg_lux/equity.py,sha256=IRwjeQmNFGuALUCWpWXWtflUEjl5qKyJumlQIAHmOPk,6076
 wbportfolio/import_export/parsers/sg_lux/fees.py,sha256=AK2TGOjI3JrOjtACEQd-4ffuedtIus-_gWug6wN6RQs,1846
 wbportfolio/import_export/parsers/sg_lux/perf_fees.py,sha256=6HAV3Pu2aaYARjMQvN24Op2rru_YdjeI9J9ADDQVH8s,1513
 wbportfolio/import_export/parsers/sg_lux/portfolio_cash_flow.py,sha256=ftLKWukUl9AcGWHQmJZnShVZpkO4rOl04PGlLcalDdQ,889
@@ -399,6 +399,8 @@ wbportfolio/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 wbportfolio/tests/conftest.py,sha256=007ydJK59J1guakXWYrZ-waDZNgXAXwMYSJ-BTDFVhA,4614
 wbportfolio/tests/signals.py,sha256=aUu4nDWmVAAWyokcWJDj2RsajeIa070qXsZeNq6ynqY,5545
 wbportfolio/tests/tests.py,sha256=3NdpJQlksLclF-7lVP07WJXaEzquhneoDX1lIA_tyXU,1121
+wbportfolio/tests/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wbportfolio/tests/analysis/test_claims.py,sha256=aWiYASn7sLg255Bq38kUd9skpYQKxnROUKau6ha4uww,4036
 wbportfolio/tests/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbportfolio/tests/models/test_account_reconciliation.py,sha256=7IDZh6qbG7-3czbNf5aDRaEDnMBw2Qiqk_h425dFu9w,8569
 wbportfolio/tests/models/test_assets.py,sha256=N8OP2ZTPdq0lGonFRB0q_ruwF8QYpOGLkPWoucr8u_E,10428
@@ -565,7 +567,7 @@ wbportfolio/viewsets/transactions/claim.py,sha256=Pb1WftoO-w-ZSTbLRhmQubhy7hgd68
 wbportfolio/viewsets/transactions/fees.py,sha256=WT2bWWfgozz4_rpyTKX7dgBBTXD-gu0nlsd2Nk2Zh1Q,7028
 wbportfolio/viewsets/transactions/mixins.py,sha256=WipvJoi5hylkpD0y9VATe30WAcwIHUIroVkK10FYw7k,636
 wbportfolio/viewsets/transactions/trades.py,sha256=xBgOGaJ8aEg-2RxEJ4FDaBs4SGwuLasun3nhpis0WQY,12363
-wbportfolio-1.55.3.dist-info/METADATA,sha256=LE4MTUvlSpHHJ82XECeFV1B3sI56oqaK4M_Fk7rkYGw,751
-wbportfolio-1.55.3.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-wbportfolio-1.55.3.dist-info/licenses/LICENSE,sha256=jvfVH0SY8_YMHlsJHKe_OajiscQDz4lpTlqT6x24sVw,172
-wbportfolio-1.55.3.dist-info/RECORD,,
+wbportfolio-1.55.5.dist-info/METADATA,sha256=BSYTmVfS-XgxIctFW8n7654BngmPS2PnUP8UwhUZssg,751
+wbportfolio-1.55.5.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+wbportfolio-1.55.5.dist-info/licenses/LICENSE,sha256=jvfVH0SY8_YMHlsJHKe_OajiscQDz4lpTlqT6x24sVw,172
+wbportfolio-1.55.5.dist-info/RECORD,,