vaccine_availability.py

"""
python vaccine_availability.py
"""

# standard imports
import requests
import datetime
import json
# import pandas as pd
import smtplib


def logger(line):
    with open('log.txt', 'a+') as f:
        f.write(line+"\n")


"""
To get the state code
for state_code in range(1,40):
    # print("State code: ", state_code)
    logger("State code: "+ str(state_code))
    response = requests.get(
        "https://cdn-api.co-vin.in/api/v2/admin/location/districts/{}".format(state_code))
    json_data = json.loads(response.text)
    for i in json_data["districts"]:
        # print(i["district_id"],'\t', i["district_name"])
        logger(str(i["district_id"])+'\t'+str(i["district_name"]))
    # print("\n")
    logger("\n")
"""

DIST_ID = 446


numdays = 20
age = 19


# Print available centre description (y/n)?
print_flag = 'y'
base = datetime.datetime.today()
date_list = [base + datetime.timedelta(days=x) for x in range(numdays)]
date_str = [x.strftime("%d-%m-%Y") for x in date_list]


def getSlots(DIST_ID=446, numdays=20, age=19):
    flag_available = False
    Available_Slots = []
    for INP_DATE in date_str:
        URL = "https://cdn-api.co-vin.in/api/v2/appointment/sessions/public/calendarByDistrict?district_id={}&date={}".format(
            DIST_ID, INP_DATE)
        response = requests.get(URL)
        if response.ok:
            resp_json = response.json()
            # print(json.dumps(resp_json, indent = 2))

            if resp_json["centers"]:
                # print("Available on: {}".format(INP_DATE))
                logger("Checking on: {}".format(INP_DATE))
                if(print_flag == 'y' or print_flag == 'Y'):
                    for center in resp_json["centers"]:
                        for session in center["sessions"]:
                            if not int(session["available_capacity"]) == 0:
                                if session["min_age_limit"] <= age:
                                    flag_available = True
                                    dict_to_add = {"Date": INP_DATE,
                                                   "Name": center["name"],
                                                   "Block Name": center["block_name"],
                                                   "Fee Type": center["fee_type"],
                                                   "Available Capacity": session["available_capacity"],
                                                   "Vaccine": session["vaccine"]}
                                    # print(dict_to_add)
                                    Available_Slots.append(dict_to_add)
                                    # print(Available_Slots)
                                    logger("\t" + str(center["name"]))
                                    logger("\t" + str(center["block_name"]))
                                    logger("\t Price: " +
                                           str(center["fee_type"]))
                                    logger("\t Available Capacity: " +
                                           str(session["available_capacity"]))
                                    """
                                    print("\t", center["name"])
                                    print("\t", center["block_name"])
                                    print("\t Price: ", center["fee_type"])
                                    print("\t Available Capacity: ",
                                          session["available_capacity"])
                                    """
                                    if(session["vaccine"] != ''):
                                        logger("\t Vaccine: " +
                                               str(session["vaccine"]))
                                    # print("\n\n")
                                    logger("\n\n")
    return flag_available, Available_Slots


"""
if flag_available == False:
            logger("No available slots on {}".format(INP_DATE))
            return flag_available, Available_Slots

        else:
            # print("No available slots on {}".format(INP_DATE))
            logger("No available slots on {}".format(INP_DATE))
            return flag_available, Available_Slots
"""


def send_mail(body, receiver_email='swaymsdennings@gmail.com',  subject='VACCINE AVAILABILITY NOTIFICATION'):
    server = smtplib.SMTP('smtp.gmail.com', 587)
    server.ehlo()
    server.starttls()
    server.ehlo()
    sender_email = 'pythonhai.000@gmail.com'
    server.login(sender_email, 'machinelearning@#$000')

    msg = f"Subject: {subject}\n\n{body}"

    server.sendmail(
        sender_email,
        receiver_email,
        msg
    )
    # print('Email has been sent !')

    server.quit()


def convert_to_str(Available_Slots):

    string = ""
    for slots in Available_Slots:
        for key in slots:
            val = slots[key]
            string += str(key) + " : " + str(val) + "\n"
        string += "\n\n"
    return string


if __name__ == '__main__':
    flag_available, Available_Slots = getSlots(DIST_ID=255, numdays=20, age=21)
    msg = "No available slots found"

    body = convert_to_str(Available_Slots) if len(Available_Slots) > 0 else msg
    MAILS = ['swaymsdennings@gmail.com', 'njrfarhandasilva10@gmail.com']

    for mail in MAILS:
        send_mail(body, receiver_email=mail,
                  subject='VACCINE AVAILABILITY NOTIFICATION')