Added rotues to retrieve data

backend/routes_auth.py

 from flask import Blueprint, current_app, jsonify, request
 from flask_security import auth_required, logout_user
-from models_shared import SweElectionResult
+from models_shared import SweElectionResult, SweDemographicResult
 
 auth = Blueprint("auth", __name__)
 
@@ -11,23 +11,64 @@ def logout():
     return jsonify(success=True)
 
 
-@auth.route("/api/swedish_election", methods=["GET"])
+@auth.route("/api/swedish_election_party_votes", methods=["GET"])
 def get_swe_election_data():
-    """
-    data = request.get_json()
-    county_code = request.args.get('county_code')
+
+    county_code = request.args.get('code')
+
     if not county_code:
         return jsonify({"errors" : ["County code is required"], "success" : False})
-    """
-
-    query_results = SweElectionResult.query.filter_by(db_county_code=1).first()
-    response = {
-        "county_code": query_results.db_county_code,
-        "county_name": query_results.db_county_name,
-        "votes_each_year": query_results.db_votes_each_year
-    }
+
+    try:
+        query_results = SweElectionResult.query.filter_by(db_county_code=int(county_code)).first()
+        response = {
+            "county_code": query_results.db_county_code,
+            "county_name": query_results.db_county_name,
+            "query_data": query_results.db_votes_each_year
+        }
+
+        return jsonify({"info": response, "success": True})
+
+    except:
+            return jsonify({"errors" : ["County code is invalid"], "success" : False})
+
+
+@auth.route("/api/swedish_election_total_results", methods=["GET"])
+def get_swe_election_total_results():
+
+    county_code = request.args.get('code')
+
+    if not county_code:
+        return jsonify({"errors" : ["County code is required"], "success" : False})
+
+    try:
+        query_results = SweElectionResult.query.filter_by(db_county_code=int(county_code)).first()
+        
+        response = {
+            "county_code": query_results.db_county_code,
+            "query_data": query_results.db_total_votes_all_years
+        }
+        return jsonify({"info": response, "success": True})
     
-    print(response)
+    except:
+        return jsonify({"errors" : ["County code is invalid"], "success" : False})
+
+
+@auth.route("/api/swedish_demographic", methods=["GET"])
+def get_swe_demographic_data():
+
+    county_code = request.args.get('code')
 
-    return jsonify({"data": response, "success": True})
-    
\ No newline at end of file
+    if not county_code:
+        return jsonify({"errors" : ["County code is required"], "success" : False})
+
+    try: 
+        query_results = SweDemographicResult.query.filter_by(db_county_code=int(county_code)).first()
+        response = {
+            "county_code": query_results.db_county_code,
+            "query_data": query_results.db_data
+        }
+        return jsonify({"info": response, "success": True})
+    
+    except:
+        return jsonify({"errors" : ["County code is invalid"], "success" : False})
\ No newline at end of file