mirror of
https://github.com/eitchtee/WYGIWYH.git
synced 2026-09-10 03:41:50 +02:00
Herculino Trotta
e2f26b3629
The rules endpoints were one instance of a pattern repeated across every
SharedObject-backed app. Route the rest through the same helper.
DCA entries were the worst case and are strictly wider than the reported
rules bug: DCAEntry has an unscoped default manager, so filtering by
strategy__id alone reached entries on strategies the caller could not see
at all. No public or shared strategy was needed -- only a guessable
integer. strategy_entry_add/edit/delete now resolve through the parent
strategy with via="strategy".
Every SharedObject delete view carried an inverted condition:
if obj.owner != request.user and request.user in obj.shared_with.all():
obj.shared_with.remove(request.user)
else:
obj.delete()
An object its owner had made public matched neither branch's intent and
fell through to delete(), so any authenticated user could destroy it.
Confirmed reachable for accounts, account groups, categories, tags,
entities and DCA strategies. The owner now deletes, a shared user revokes
only their own access, and anyone else gets a 403.
The API viewsets had no object-level check at all. DjangoModelPermissions
gated them shut for ordinary users, who hold no model permissions, so this
was not reachable in a default install -- but the check belongs there
regardless, and a user granted change_account in the admin could write any
visible account. SharedObjectPermission adds it for the SharedObject
viewsets, leaving reads to SharedObjectManager.
account_toggle_untracked only flips the calling user's own row in the
untracked_by m2m, so it takes READ rather than EDIT.
Refs GHSA-83g9-vjqf-2j5q
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
from rest_framework import viewsets
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
from apps.dca.models import DCAStrategy, DCAEntry
|
|
from apps.api.permissions import SHARED_OBJECT_PERMISSIONS
|
|
from apps.api.serializers import DCAStrategySerializer, DCAEntrySerializer
|
|
|
|
|
|
class DCAStrategyViewSet(viewsets.ModelViewSet):
|
|
permission_classes = SHARED_OBJECT_PERMISSIONS
|
|
queryset = DCAStrategy.objects.all()
|
|
serializer_class = DCAStrategySerializer
|
|
filterset_fields = {
|
|
"name": ["exact", "icontains"],
|
|
"target_currency": ["exact"],
|
|
"payment_currency": ["exact"],
|
|
"notes": ["exact", "icontains"],
|
|
"created_at": ["exact", "gte", "lte", "gt", "lt"],
|
|
"updated_at": ["exact", "gte", "lte", "gt", "lt"],
|
|
}
|
|
search_fields = ["name", "notes"]
|
|
ordering_fields = "__all__"
|
|
|
|
def get_queryset(self):
|
|
return DCAStrategy.objects.all()
|
|
|
|
@action(detail=True, methods=["get"])
|
|
def investment_frequency(self, request, pk=None):
|
|
strategy = self.get_object()
|
|
return Response(strategy.investment_frequency_data())
|
|
|
|
@action(detail=True, methods=["get"])
|
|
def price_comparison(self, request, pk=None):
|
|
strategy = self.get_object()
|
|
return Response(strategy.price_comparison_data())
|
|
|
|
@action(detail=True, methods=["get"])
|
|
def current_price(self, request, pk=None):
|
|
strategy = self.get_object()
|
|
price_data = strategy.current_price()
|
|
if price_data:
|
|
price, date = price_data
|
|
return Response({"price": price, "date": date})
|
|
return Response({"price": None, "date": None})
|
|
|
|
|
|
class DCAEntryViewSet(viewsets.ModelViewSet):
|
|
permission_classes = SHARED_OBJECT_PERMISSIONS
|
|
shared_object_via = "strategy"
|
|
queryset = DCAEntry.objects.all()
|
|
serializer_class = DCAEntrySerializer
|
|
filterset_fields = {
|
|
"strategy": ["exact"],
|
|
"date": ["exact", "gte", "lte", "gt", "lt"],
|
|
"amount_paid": ["exact", "gte", "lte", "gt", "lt"],
|
|
"amount_received": ["exact", "gte", "lte", "gt", "lt"],
|
|
"expense_transaction": ["exact", "isnull"],
|
|
"income_transaction": ["exact", "isnull"],
|
|
"notes": ["exact", "icontains"],
|
|
"created_at": ["exact", "gte", "lte", "gt", "lt"],
|
|
"updated_at": ["exact", "gte", "lte", "gt", "lt"],
|
|
}
|
|
search_fields = ["notes"]
|
|
ordering_fields = "__all__"
|
|
ordering = ["-date"]
|
|
|
|
def get_queryset(self):
|
|
# Filter entries by strategies the user has access to
|
|
accessible_strategies = DCAStrategy.objects.all()
|
|
return DCAEntry.objects.filter(strategy__in=accessible_strategies)
|