PR for statistic endpoint

web_app/api/user.py

@@ -1,12 +1,14 @@
 from fastapi import APIRouter, Request, HTTPException
-from web_app.db.crud import UserDBConnector
+from web_app.db.crud import PositionDBConnector, UserDBConnector
 from web_app.api.serializers.transaction import UpdateUserContractRequest
 from web_app.api.serializers.user import CheckUserResponse, UpdateUserContractResponse, GetUserContractAddressResponse
 
 router = APIRouter()  # Initialize the router
 
 user_db = UserDBConnector()
 
+position_db = PositionDBConnector()
+
 
 @router.get("/api/get-user-contract", tags=["User Operations"], summary="Get user's contract status", response_description="Returns 0 if the user is None or if the contract is not deployed. Returns the transaction hash if the contract is deployed.")
 async def get_user_contract(wallet_id: str) -> str:
@@ -86,3 +88,27 @@ async def get_user_contract_address(wallet_id: str) -> GetUserContractAddressRes
         return {"contract_address": contract_address}
     else:
         return {"contract_address": None}
+
+
+@router.get("/api/get_stats", tags=["Statistics"], summary="Get all user amounts and the number of unique users.")
+async def get_stats():
+    """
+    Returns all user amounts for open positions and the number of unique users.
+
+    ### Returns:
+    - Dictionary containing:
+        - total_amounts: dict where keys are user IDs and values are total amounts
+        - unique_users: The count of unique users with open positions
+    """
+    try:
+        # Fetch total amounts for each user where position status is OPENED
+        total_amounts = position_db.get_all_amounts_for_opened_positions()
+
+        # Get the number of unique users
+        unique_users_count = len(total_amounts)
+
+        # Return total amounts and unique users count
+        return {"total_amounts": total_amounts, "unique_users": unique_users_count}
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail="Error retrieving statistics.")

web_app/db/crud.py

@@ -326,3 +326,40 @@ def open_position(self, position_id: uuid) -> Position | None:
             position.status = Status.OPENED.value
             self.write_to_db(position)
         return position.status
+
+    def get_unique_users(self) -> list[str]:
+        """
+        Fetches all unique users from the Position table.
+        :return: list of unique user ids
+        """
+        with self.Session() as db :
+            try:
+                # Query for distinct user IDs from the Position table
+                unique_users = db.query(Position.user_id).distinct().all()
+                return [user[0] for user in unique_users]  # Extract user_id from tuple
+            except SQLAlchemyError as e:
+                logger.error(f"Error fetching unique users: {str(e)}")
+                return []
+
+    def get_all_amounts_for_opened_positions(self) -> dict[str, float]:
+        """
+        Fetches the total amount for all users where position status is 'OPENED'.
+
+        :return: A dictionary with user IDs as keys and the total amount as values.
+        """
+        with self.Session() as db:
+            try:
+                # Query all positions with status OPENED and group by user_id
+                opened_positions = (
+                    db.query(Position.user_id, db.func.sum(Position.amount).label('total_amount'))
+                    .filter(Position.status == Status.OPENED.value)
+                    .group_by(Position.user_id)
+                    .all()
+                )
+
+                # Return a dictionary with user IDs and their respective total amounts
+                return {user_id: total_amount for user_id, total_amount in opened_positions}
+
+            except SQLAlchemyError as e:
+                logger.error(f"Error fetching amounts for opened positions: {str(e)}")
+                return {}