vaccine_slot_availability/main.py at master · keshav-sarraf/vaccine_slot_availability · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
from flask import Flask, jsonify, render_template, request

from api_service import get_all_dist_codes_api
from db_service import add_subscriber_to_topic, get_all_dist_codes_db, get_slots_by_dist_id_db, \
    get_dist_id_from_name_db, delete_subscriber_from_topic, send_test_notification, get_trend_for_dist_id
from utils import get_filtered_dists

app = Flask(__name__, static_url_path='/static')

# To run first thing and cache the results
get_all_dist_codes_db()


@app.route('/districts', methods=['GET'])
def get_districts():
    query = request.args.get('q')
    dist_codes = get_all_dist_codes_db()
    filtered_dists = get_filtered_dists(query, dist_codes)
    filtered_dists_list = list(map(lambda x: "{} | {}".format(x["dist_name"], x["state_name"]), filtered_dists))
    filtered_dists_list = filtered_dists_list[0:12]
    return jsonify(filtered_dists_list)


@app.route('/slots/district/<dist_name>')
def get_vaccination_slots(dist_name):
    dist_id = get_dist_id_from_name_db(dist_name)
    vaccination_calendar = get_slots_by_dist_id_db(dist_id)
    return jsonify(vaccination_calendar)


@app.route('/')
def landing_page():
    return render_template("landing.html")


@app.route('/notification-subscription', methods=['POST'])
def accept_notification_subscription():
    payload = request.json

    dist_name = payload["dist_name"]
    notification_token = payload["notification_token"]
    # age_group = payload["age_group"]

    dist_id = get_dist_id_from_name_db(dist_name)
    result = add_subscriber_to_topic(notification_token, dist_id)

    return jsonify(result)


@app.route('/notification-subscription', methods=['DELETE'])
def accept_notification_un_subscription():
    payload = request.json

    dist_name = payload["dist_name"]
    notification_token = payload["notification_token"]
    # age_group = payload["age_group"]

    dist_id = get_dist_id_from_name_db(dist_name)
    result = delete_subscriber_from_topic(notification_token, dist_id)

    return jsonify(result)


@app.route('/notification-test', methods=['POST'])
def test_notification():
    payload = request.json

    dist_name = payload["dist_name"]
    notification_token = payload["notification_token"]
    result = send_test_notification(notification_token, dist_name)

    return jsonify(result)


@app.route('/trend/district/<dist_name>')
def get_trend_for_dist(dist_name):
    dist_id = get_dist_id_from_name_db(dist_name)
    trend = get_trend_for_dist_id(dist_id)
    return jsonify(trend)

'''
for notification:
    email,
    device info ?

    dist_id:
        dist_name
        notifiers:[
            p1:
                is_notified?
                push_id:
                timestamp_of_subscription
        ]


'''

# TODO: change api to pick from firestore
# TODO: create apis to refresh firestore periodically
# TODO: create api to accept pin code
# TODO: create api to accept coordinates
# TODO: create api to accept subscription
# TODO: create job to send notification if slots available
# TODO: remove from list after sending notification
# TODO: if sharing url with location, then populate the list automatically
# TODO: create a page to show how to enable notifications
# TODO: validate token before sending notification

if __name__ == '__main__':
    app.run(debug=True)