wbportfolio 1.52.0__py2.py3-none-any.whl → 1.59.4__py2.py3-none-any.whl

wbportfolio/viewsets/configs/titles/transactions.py

@@ -1,9 +0,0 @@
-from wbcore.metadata.configs.titles import TitleViewConfig
-
-from wbportfolio.models import Portfolio
-
-
-class TransactionPortfolioTitleConfig(TitleViewConfig):
-    def get_list_title(self):
-        portfolio = Portfolio.objects.get(id=self.view.kwargs["portfolio_id"])
-        return f"Transactions for Product {str(portfolio)}"

wbportfolio/viewsets/signals.py

@@ -1,43 +0,0 @@
-from django.apps import apps
-from django.contrib.contenttypes.models import ContentType
-from django.db.models import Q
-from django.dispatch import receiver
-from rest_framework.reverse import reverse
-from wbcore.contrib.icons import WBIcon
-from wbcore.metadata.configs import buttons as bt
-from wbcore.signals.instance_buttons import add_instance_button
-
-from wbportfolio.viewsets.products import ProductModelViewSet
-
-
-@receiver(add_instance_button, sender=ProductModelViewSet)
-def add_report_buttons(sender, many, *args, request=None, **kwargs):
-    report_buttons = []
-    try:
-        if instrument_id := kwargs.get("pk", None):
-            content_type_product = ContentType.objects.get(app_label="wbportfolio", model="product")
-            content_type_instrument = ContentType.objects.get(app_label="wbfdm", model="instrument")
-            Report = apps.get_model("wbreport", "Report")
-            for report in Report.objects.filter(object_id=instrument_id, is_active=True).filter(
-                Q(content_type=content_type_product) | Q(content_type=content_type_instrument)
-            ):
-                if primary_version := report.primary_version:
-                    report_buttons.append(
-                        bt.HyperlinkButton(
-                            label=f"{report.title} Report",
-                            endpoint=reverse(
-                                "public_report:report_version", args=[primary_version.lookup], request=request
-                            ),
-                        ),
-                    )
-                    if request.user.profile.is_internal or request.user.is_superuser:
-                        report_buttons.append(
-                            bt.HyperlinkButton(
-                                label=f"{report.title} Widget",
-                                endpoint=reverse("wbreport:report-detail", args=[report.id], request=request),
-                            ),
-                        )
-    except LookupError:
-        pass
-    if report_buttons:
-        return bt.DropDownButton(label="Reports", buttons=tuple(report_buttons), icon=WBIcon.NOTEBOOK.icon)

wbportfolio/viewsets/transactions/trade_proposals.py

@@ -1,139 +0,0 @@
-from contextlib import suppress
-from datetime import date
-
-from django.contrib.messages import info, warning
-from django.shortcuts import get_object_or_404
-from django.utils.functional import cached_property
-from pandas._libs.tslibs.offsets import BDay
-from rest_framework import status
-from rest_framework.decorators import action
-from rest_framework.response import Response
-from wbcompliance.viewsets.risk_management.mixins import RiskCheckViewSetMixin
-from wbcore import serializers as wb_serializers
-from wbcore import viewsets
-from wbcore.metadata.configs.display.instance_display import (
-    Display,
-    create_simple_display,
-)
-from wbcore.permissions.permissions import InternalUserPermissionMixin
-from wbcore.utils.views import CloneMixin
-
-from wbportfolio.models import AssetPosition, TradeProposal
-from wbportfolio.models.transactions.trade_proposals import (
-    replay_as_task,
-)
-from wbportfolio.serializers import (
-    ReadOnlyTradeProposalModelSerializer,
-    TradeProposalModelSerializer,
-    TradeProposalRepresentationSerializer,
-)
-
-from ..configs import (
-    TradeProposalButtonConfig,
-    TradeProposalDisplayConfig,
-    TradeProposalEndpointConfig,
-    TradeProposalPortfolioEndpointConfig,
-)
-from ..mixins import UserPortfolioRequestPermissionMixin
-
-
-class TradeProposalRepresentationViewSet(InternalUserPermissionMixin, viewsets.RepresentationViewSet):
-    IDENTIFIER = "wbportfolio:trade"
-    queryset = TradeProposal.objects.all()
-    serializer_class = TradeProposalRepresentationSerializer
-
-
-class TradeProposalModelViewSet(CloneMixin, RiskCheckViewSetMixin, InternalUserPermissionMixin, viewsets.ModelViewSet):
-    ordering_fields = ("trade_date",)
-    ordering = ("-trade_date",)
-    search_fields = ("comment",)
-    filterset_fields = {"trade_date": ["exact", "gte", "lte"], "status": ["exact"]}
-
-    queryset = TradeProposal.objects.select_related("rebalancing_model", "portfolio")
-    serializer_class = TradeProposalModelSerializer
-    display_config_class = TradeProposalDisplayConfig
-    button_config_class = TradeProposalButtonConfig
-    endpoint_config_class = TradeProposalEndpointConfig
-
-    def get_serializer_class(self):
-        if self.new_mode or (
-            "pk" in self.kwargs and (obj := self.get_object()) and obj.status == TradeProposal.Status.DRAFT
-        ):
-            return TradeProposalModelSerializer
-        return ReadOnlyTradeProposalModelSerializer
-
-    # 2 methods to parametrize the clone button functionality
-    def get_clone_button_serializer_class(self, instance):
-        class CloneSerializer(wb_serializers.Serializer):
-            clone_date = wb_serializers.DateField(
-                default=(instance.trade_date + BDay(1)).date(), label="Trade Date"
-            )  # we need to change the field name from the trade proposa fields, otherwise fontend conflicts
-            clone_comment = wb_serializers.TextField(label="Comment")
-
-        return CloneSerializer
-
-    def get_clone_button_instance_display(self) -> Display:
-        return create_simple_display(
-            [
-                ["clone_comment"],
-                ["clone_date"],
-            ]
-        )
-
-    def add_messages(self, request, instance=None, **kwargs):
-        if instance and instance.status == TradeProposal.Status.SUBMIT:
-            if not instance.portfolio.is_manageable:
-                info(request, "This trade proposal cannot be approved the portfolio is considered unmanaged.")
-            if instance.has_non_successful_checks:
-                warning(
-                    request,
-                    "This trade proposal cannot be approved because there is unsuccessful pre-trade checks. Please rectify accordingly and resubmit a valid trade proposal",
-                )
-
-    @classmethod
-    def _get_risk_checks_button_title(cls) -> str:
-        return "Pre-Trade Checks"
-
-    @action(detail=True, methods=["PATCH"])
-    def reset(self, request, pk=None):
-        trade_proposal = get_object_or_404(TradeProposal, pk=pk)
-        if trade_proposal.status == TradeProposal.Status.DRAFT:
-            trade_proposal.reset_trades()
-            return Response({"send": True})
-        return Response({"status": "Trade proposal is not Draft"}, status=status.HTTP_400_BAD_REQUEST)
-
-    @action(detail=True, methods=["PATCH"])
-    def normalize(self, request, pk=None):
-        trade_proposal = get_object_or_404(TradeProposal, pk=pk)
-        if trade_proposal.status == TradeProposal.Status.DRAFT:
-            trade_proposal.normalize_trades()
-            return Response({"send": True})
-        return Response({"status": "Trade proposal is not Draft"}, status=status.HTTP_400_BAD_REQUEST)
-
-    @action(detail=True, methods=["PATCH"])
-    def replay(self, request, pk=None):
-        trade_proposal = get_object_or_404(TradeProposal, pk=pk)
-        if trade_proposal.portfolio.is_manageable:
-            replay_as_task.delay(trade_proposal.id, user_id=self.request.user.id)
-            return Response({"send": True})
-        return Response({"status": "Trade proposal is not Draft"}, status=status.HTTP_400_BAD_REQUEST)
-
-    @action(detail=True, methods=["PATCH"])
-    def deleteall(self, request, pk=None):
-        trade_proposal = get_object_or_404(TradeProposal, pk=pk)
-        if trade_proposal.status == TradeProposal.Status.DRAFT:
-            trade_proposal.trades.all().delete()
-            return Response({"send": True})
-        return Response({"status": "Trade proposal is not Draft"}, status=status.HTTP_400_BAD_REQUEST)
-
-
-class TradeProposalPortfolioModelViewSet(UserPortfolioRequestPermissionMixin, TradeProposalModelViewSet):
-    endpoint_config_class = TradeProposalPortfolioEndpointConfig
-
-    @cached_property
-    def default_trade_date(self) -> date | None:
-        with suppress(AssetPosition.DoesNotExist):
-            return (self.portfolio.assets.latest("date").date + BDay(1)).date()
-
-    def get_queryset(self):
-        return TradeProposal.objects.filter(portfolio=self.kwargs["portfolio_id"])

wbportfolio/viewsets/transactions/transactions.py

@@ -1,122 +0,0 @@
-from decimal import Decimal
-
-from django.db.models import (
-    Case,
-    DecimalField,
-    ExpressionWrapper,
-    F,
-    OuterRef,
-    Subquery,
-    Sum,
-    When,
-)
-from wbcore import viewsets
-from wbcore.contrib.currency.models import CurrencyFXRates
-from wbcore.permissions.permissions import InternalUserPermissionMixin
-from wbcore.utils.strings import format_number
-
-from wbportfolio.filters import TransactionFilterSet, TransactionPortfolioFilterSet
-from wbportfolio.models import Fees, Trade, Transaction
-from wbportfolio.serializers import (
-    TransactionModelSerializer,
-    TransactionRepresentationSerializer,
-)
-
-from ..configs import (
-    TransactionDisplayConfig,
-    TransactionEndpointConfig,
-    TransactionPortfolioDisplayConfig,
-    TransactionPortfolioEndpointConfig,
-    TransactionPortfolioTitleConfig,
-)
-from ..mixins import UserPortfolioRequestPermissionMixin
-
-
-class TransactionRepresentationViewSet(InternalUserPermissionMixin, viewsets.RepresentationViewSet):
-    IDENTIFIER = "wbportfolio:transaction"
-    filterset_class = TransactionFilterSet
-    search_fields = ("portfolio__name", "underlying_instrument__name", "comment")
-    ordering_fields = (
-        "transaction_date",
-        "currency__key",
-        "underlying_instrument__name",
-        "portfolio__name",
-        "transaction_type",
-        "book_date",
-        "value_date",
-        "currency_fx_rate",
-        "total_value",
-        "total_value_fx_portfolio",
-        "total_value_gross",
-        "total_value_gross_fx_portfolio",
-    )
-    ordering = ["-transaction_date"]
-
-    serializer_class = TransactionRepresentationSerializer
-    queryset = Transaction.objects.all()
-
-
-class TransactionModelViewSet(UserPortfolioRequestPermissionMixin, InternalUserPermissionMixin, viewsets.ModelViewSet):
-    IDENTIFIER = "wbportfolio:transaction"
-
-    serializer_class = TransactionModelSerializer
-
-    filterset_class = TransactionFilterSet
-    ordering_fields = TransactionRepresentationViewSet.ordering_fields
-    search_fields = TransactionRepresentationViewSet.search_fields
-    ordering = TransactionRepresentationViewSet.ordering
-
-    queryset = Transaction.objects.all()
-    display_config_class = TransactionDisplayConfig
-    endpoint_config_class = TransactionEndpointConfig
-
-    def get_queryset(self):
-        return (
-            super()
-            .get_queryset()
-            .annotate(
-                fx_rate=CurrencyFXRates.get_fx_rates_subquery(
-                    "transaction_date", currency="currency", lookup_expr="exact"
-                ),
-                total_value_usd=ExpressionWrapper(F("total_value"), output_field=DecimalField()),
-                total_value_gross_usd=ExpressionWrapper(F("total_value_gross"), output_field=DecimalField()),
-                transaction_underlying_type_trade=Subquery(
-                    Trade.objects.filter(id=OuterRef("pk")).values("transaction_subtype")[:1]
-                ),
-                transaction_underlying_type_fees=Subquery(
-                    Fees.objects.filter(id=OuterRef("pk")).values("transaction_subtype")[:1]
-                ),
-                transaction_underlying_type=Case(
-                    When(
-                        transaction_underlying_type_trade__isnull=False,
-                        then=F("transaction_underlying_type_trade"),
-                    ),
-                    When(transaction_underlying_type_fees__isnull=False, then=F("transaction_underlying_type_fees")),
-                ),
-            )
-            .select_related("underlying_instrument")
-            .select_related("portfolio")
-            .select_related("currency")
-        )
-
-    def get_aggregates(self, queryset, paginated_queryset):
-        return {
-            "total_value_usd": {
-                "Σ": format_number(queryset.aggregate(s=Sum("total_value_usd"))["s"] or Decimal(0)),
-            }
-        }
-
-
-class TransactionPortfolioModelViewSet(TransactionModelViewSet):
-    IDENTIFIER = "wbportfolio:transaction"
-
-    filterset_class = TransactionPortfolioFilterSet
-    display_config_class = TransactionPortfolioDisplayConfig
-    endpoint_config_class = TransactionPortfolioEndpointConfig
-    title_config_class = TransactionPortfolioTitleConfig
-
-    def get_queryset(self):
-        qs = super().get_queryset().filter(portfolio=self.portfolio)
-        if self.is_portfolio_manager:
-            return qs
-        return qs.exclude(transaction_type=Transaction.Type.FEES)