add backend

Author: kittinan

.gitignore

@@ -166,3 +166,4 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+venv

backend/api/requirements.txt

@@ -0,0 +1,6 @@
+pandas==1.1.1
+requests==2.24.0
+flask==1.1.2
+python-dotenv==0.14.0
+cachetools==4.1.1
+airtable-python-wrapper==0.15.0

backend/api/survey-result.py

@@ -0,0 +1,78 @@
+from flask import Flask, Response, jsonify
+from dotenv import load_dotenv, find_dotenv
+from cachetools import cached, LRUCache, TTLCache
+from airtable import Airtable
+from collections import Counter
+
+import pandas as pd
+import requests
+import os
+
+load_dotenv(find_dotenv())
+
+print("Starting Server")
+
+AIRTABLE_ID = os.getenv("AIRTABLE_ID")
+AIRTABLE_KEY = os.getenv("AIRTABLE_KEY")
+
+app = Flask(__name__)
+
+
+def get_airtable_data(table_name: str):
+    """
+    Get data from Airtable with a particular AIRTABLE_ID
+    stated in .env file in this directory
+
+    table_name can be ``talks``, ``jobboard``, ``jobseeker``
+    """
+    request_url = f"https://api.airtable.com/v0/{AIRTABLE_ID}/{table_name}?api_key={AIRTABLE_KEY}"
+    print(request_url)
+    airtable_data = requests.get(request_url).json()
+    return airtable_data
+
+
+def format_dict_counter(dict_counter):
+
+    result = []
+    for k, v in sorted(dict_counter.items(), key=lambda item: item[1], reverse=True):
+        result.append({"budget_type": k, "amount": v})
+
+    return result
+
+
+@app.route("/", defaults={"path": ""})
+@app.route("/<path:path>")
+def catch_all(path):
+
+    #survey = get_airtable_data("survey")
+    #survey_df = pd.DataFrame([r["fields"] for r in survey["records"]])
+
+    survey_table = Airtable(AIRTABLE_ID, 'survey', AIRTABLE_KEY)
+    survey = survey_table.get_all()
+    
+    survey_df = pd.DataFrame([r['fields'] for r in survey])
+    survey_summary_df = survey_df.groupby("เขต").agg(sum).reset_index()
+
+    print(survey_summary_df)
+    survey_summary_df["to_increase"] = survey_summary_df["อยากเพิ่มงบ"].map(
+        lambda x: format_dict_counter(dict(Counter(x)))
+    )
+    survey_summary_df["to_decrease"] = survey_summary_df["อยากลดงบ"].map(
+        lambda x: format_dict_counter(dict(Counter(x)))
+    )
+    result = (
+        survey_summary_df[["เขต", "to_increase", "to_decrease"]]
+        .rename(columns={"เขต": "district"})
+        .to_dict(orient="records")
+    )
+
+    # Add more data
+    for r in result:
+        r["sum_to_increase"] = sum([x["amount"] for x in r["to_increase"]])
+        r["sum_to_decrease"] = sum([x["amount"] for x in r["to_decrease"]])
+
+    return jsonify({"status": True, "data": result})
+
+
+if __name__ == "__main__":
+    app.run(debug=True, port=5003)