BE: stocks updates

andreipradan · Feb 25, 2025 · c33c90e · c33c90e
1 parent 037185f
commit c33c90e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/mainframe/clients/finance/stocks.py b/src/mainframe/clients/finance/stocks.py
@@ -93,7 +93,7 @@ def normalize_key(key):
             return key.lower().replace(" ", "_")
 
         def normalize_value(value):
-            return str(value).replace("$", "").replace("€", "")
+            return str(value).replace("$", "").replace("€", "").replace(",", "")
 
         row["Type"] = self.normalize_type(row["Type"])
         return {normalize_key(k): normalize_value(v) for (k, v) in row.items() if v}

diff --git a/src/mainframe/finance/viewsets/mixins.py b/src/mainframe/finance/viewsets/mixins.py
@@ -63,4 +63,5 @@ def pnl(self, request, *args, **kwargs):
             except self.pnl_importer_error_class as e:
                 logger.error("Could not process file: %s - error: %s", file, e)
                 return Response(f"Invalid file: {file}", status.HTTP_400_BAD_REQUEST)
-            return self.list(request, *args, **kwargs)
+            request.method = "GET"
+            return self.pnl(request, *args, **kwargs)
diff --git a/src/mainframe/finance/viewsets/stocks.py b/src/mainframe/finance/viewsets/stocks.py
@@ -14,6 +14,8 @@
 from mainframe.finance.serializers import PnLSerializer, StockTransactionSerializer
 from mainframe.finance.viewsets.mixins import PnlActionModelViewSet
 
+logger = logging.getLogger(__name__)
+
 
 class StocksViewSet(PnlActionModelViewSet):
     permission_classes = (IsAdminUser,)
@@ -26,7 +28,6 @@ class StocksViewSet(PnlActionModelViewSet):
 
     def create(self, request, *args, **kwargs):
         file = request.FILES["file"]
-        logger = logging.getLogger(__name__)
         try:
             StockTransactionsImporter(file, logger).run()
         except StockImportError as e: