Constituency Map - Competitiveness

.gitignore

@@ -157,3 +157,4 @@ webpack-stats.json
 token.pickle
 
 node_modules
+.vscode

.vscode/settings.json

@@ -2,5 +2,5 @@
     "editor.defaultFormatter": "ms-python.autopep8",
     "pylint.args": [
         "--disable=E0015"
-    ]
-}
+    ],
+}

backend/app/api_views.py

@@ -7,12 +7,14 @@
 from rest_framework.response import Response
 
 from .models import (
+    LSElection,
     TCPDElection,
     SeatShare,
     CampaignFinance,
 )
 
 from .serializers import (
+    LSElectionSerializaer,
     TCPDElectionSerializer,
     SeatShareSerializer,
     CampaignFinanceSerializer,
@@ -40,18 +42,106 @@ def all_seats(request):
     return Response(serializer.data)
 
 
+@api_view(['GET'])
+def all_ls_elections(request):
+    """
+    API endpoint to get each Lok Sahbha election
+    in the database
+    """
+    ls_elections = LSElection.objects.all()
+    serializer = LSElectionSerializaer(ls_elections, many=True)
+    return Response(serializer.data)
+
+
+@api_view(['GET'])
+def get_ls_election_year(request, year):
+    """
+    API endpoint to get each Lok Sahbha election
+    in the database
+    """
+    ls_elections = LSElection.objects.filter(election_year=year)
+    serializer = LSElectionSerializaer(ls_elections, many=True)
+    return Response(serializer.data)
+
+
+@api_view(['GET'])
+def get_specific_ls_election(request, year, state, constituency_no):
+    """
+    API endpoint to get each Lok Sahbha election
+    in the database
+    """
+    ls_elections = LSElection.objects.filter(
+        election_year=year, state_name=state, constituency_no=constituency_no)
+    serializer = LSElectionSerializaer(ls_elections, many=True)
+    return Response(serializer.data)
+
+
 @api_view(['GET'])
 def get_SDE_DATA_IN_F7DSTRBND_1991(request, feature_limit=10):
     """
     API endpoint to get SDE_DATA_IN_F7DSTRBND_1991 geojson
     """
-    geojson_path = os.path.join(settings.GEOJSON_DIR, "SDE_DATA_IN_F7DSTRBND_1991.geojson")
-    with open(geojson_path, encoding='utf-8') as f:
+    geocolors_path = os.path.join(
+        settings.GEOJSON_DIR, "SDE_DATA_IN_F7DSTRBND_1991.geojson")
+    with open(geocolors_path, encoding='utf-8') as f:
+        geojson = json.load(f)
+
+        return Response({
+            "type": geojson["type"],
+            "features": geojson["features"]
+        })
+
+
+@api_view(['GET'])
+def get_India_PC_2019_simplified(request, feature_limit=10):
+    """
+    API endpoint to get SDE_DATA_IN_F7DSTRBND_1991 geojson
+    """
+    geocolors_path = os.path.join(
+        settings.GEOJSON_DIR, "india_pc_2019_simplified.geojson")
+    with open(geocolors_path, encoding='utf-8') as f:
         geojson = json.load(f)
-        num_features = feature_limit if feature_limit is not None else len(geojson["features"])
+
         return Response({
             "type": geojson["type"],
-            "features": geojson["features"][:num_features]
+            "features": geojson["features"]
+        })
+
+
+@api_view(['GET'])
+def get_India_PC_2019(request, feature_limit=10):
+    """
+    API endpoint to get SDE_DATA_IN_F7DSTRBND_1991 geojson
+    """
+    geocolors_path = os.path.join(
+        settings.GEOJSON_DIR, "India_PC_2019.geojson")
+    with open(geocolors_path, encoding='utf-8') as f:
+        geojson = json.load(f)
+
+        return Response({
+            "type": geojson["type"],
+            "features": geojson["features"]
+        })
+
+
+@api_view(['GET'])
+def get_competitiveness_data(request, election_year=2004):
+    """
+    API endpoint to get data and colors for competitiveness map
+    """
+    colors_path = os.path.join(
+        settings.GEOJSON_DIR, "competitivenessColors.json")
+    with open(colors_path, encoding='utf-8') as f:
+        colors_json = json.load(f)
+
+    data_path = os.path.join(
+        settings.GEOJSON_DIR, "competitivenessData.json")
+    with open(data_path, encoding='utf-8') as f:
+        data_json = json.load(f)
+
+        return Response({
+            "colors": colors_json[str(election_year)],
+            "data": data_json[str(election_year)]
         })
 
 
@@ -74,7 +164,7 @@ def campaign_finance(request, donor_name=None, party_name=None):
 @api_view(['GET'])
 def campaign_finance_party_subset(request, party_name):
     """
-    API endpoint to get campaign finance donations made by all donors to 
+    API endpoint to get campaign finance donations made by all donors to
     a specified party
     """
     campaign_finances = CampaignFinance.objects.filter(party_name=party_name)
@@ -85,7 +175,7 @@ def campaign_finance_party_subset(request, party_name):
 @api_view(['GET'])
 def campaign_finance_donor_subset(request, donor_name):
     """
-    API endpoint to get campaign finance donations made to all parties by 
+    API endpoint to get campaign finance donations made to all parties by
     a specified donor
     """
     campaign_finances = CampaignFinance.objects.filter(donor_name=donor_name)