From 9125fec3e66bba2e5eced52d1b99aa1977f7b388 Mon Sep 17 00:00:00 2001
From: nprasad2077 <114438461+nprasad2077@users.noreply.github.com>
Date: Fri, 28 Apr 2023 01:28:25 -0500
Subject: [PATCH] new view to return scatter plot data

---
 nba_data/urls.py  |  3 ++-
 nba_data/views.py | 36 ++++++++++++++++++++++++++++++++++--
 2 files changed, 36 insertions(+), 3 deletions(-)

diff --git a/nba_data/urls.py b/nba_data/urls.py
index 14a3e00..36a43eb 100644
--- a/nba_data/urls.py
+++ b/nba_data/urls.py
@@ -1,5 +1,5 @@
 from django.urls import path
-from .views import PlayerDataList, PlayerDataByTeamList, PlayerDataBySeasonList, PlayerDataByNameList, TopScorersbySeasonList, ThreeTwoTopPointTrends, TopAssistsBySeasonList, TopReboundsBySeasonList, TopBlocksBySeasonList, TopStealsBySeasonList, TopOffensiveReboundsBySeasonList, TopDefensiveReboundsBySeasonList, PointsPerGameHistogramBySeasonList
+from .views import PlayerDataList, PlayerDataByTeamList, PlayerDataBySeasonList, PlayerDataByNameList, TopScorersbySeasonList, ThreeTwoTopPointTrends, TopAssistsBySeasonList, TopReboundsBySeasonList, TopBlocksBySeasonList, TopStealsBySeasonList, TopOffensiveReboundsBySeasonList, TopDefensiveReboundsBySeasonList, PointsPerGameHistogramBySeasonList, TopPtsScatterPlotData
 
 urlpatterns = [
     path('api/playerdata/', PlayerDataList.as_view(), name='playerdata_list'),
@@ -15,4 +15,5 @@ urlpatterns = [
     path('api/top_rebounds_offensive/<int:season>/', TopOffensiveReboundsBySeasonList.as_view(), name='top_rebounds_ORB'),
     path('api/top_rebounds_defensive/<int:season>/', TopDefensiveReboundsBySeasonList.as_view(), name='top_rebounds_DRB'),
     path('api/points_per_game_histogram/<int:season>/', PointsPerGameHistogramBySeasonList.as_view(), name='points_per_game_histogram'),
+    path('api/top_pts_scatter_plot/', TopPtsScatterPlotData.as_view(), name='top_pts_scatter_plot'),
 ]
\ No newline at end of file
diff --git a/nba_data/views.py b/nba_data/views.py
index c1d3874..3418e5e 100644
--- a/nba_data/views.py
+++ b/nba_data/views.py
@@ -1,12 +1,12 @@
 from django.shortcuts import render
 from django.http import JsonResponse
 from django.core import serializers
-from .models import PlayerData
+from .models import PlayerData, PlayerTotalsData, PlayerAdvancedData
 from rest_framework import generics
 from .serializers import PlayerDataSerializer, HistogramDataSerializer
 from rest_framework.views import APIView
 from rest_framework.response import Response
-from django.db.models import Avg
+from django.db.models import Avg, Sum, F, FloatField
 from collections import Counter
 import math
 
@@ -174,3 +174,35 @@ class PointsPerGameHistogramBySeasonList(generics.ListAPIView):
                           for k, v in histogram_data.items()]
 
         return histogram_list
+
+
+# Scatter Plot - Top 30 players by total PTS across all seasons in DB. Then map those 30 players on a scatter plot against total PTS/WS for each season.
+
+class TopPtsScatterPlotData(APIView):
+    def get(self, request, *args, **kwargs):
+        # Get the top 25 players by overall PTS
+        top_players = PlayerTotalsData.objects.values('player_name') \
+            .annotate(total_pts=Sum(F('PTS'))) \
+            .order_by('-total_pts')[:25]
+
+        # Get the PTS and WS for the corresponding seasons of the top 25 players
+        data = []
+        for player in top_players:
+            player_name = player['player_name']
+            player_seasons = PlayerTotalsData.objects.filter(player_name=player_name) \
+                .values('season') \
+                .annotate(total_pts=Sum(F('PTS'))) \
+                .order_by('season')
+
+            # Combine the WS data from PlayerAdvancedData
+            for season in player_seasons:
+                season_ws = PlayerAdvancedData.objects.filter(player_name=player_name, season=season['season']) \
+                    .aggregate(total_ws=Sum('ws'))
+                season['total_ws'] = season_ws['total_ws']
+
+            data.append({
+                'player_name': player_name,
+                'seasons': list(player_seasons)
+            })
+
+        return Response(data)