main.py

import json
import time
import requests
from datetime import date
import pygame
import configparser

from requests import ConnectionError
from requests import Timeout
from requests import HTTPError


class VaccineInfo:

    def __init__(self, pin_code):
        self.pinCode = pin_code

    # Method to get data for vaccine centers at a given pin-code
    def getVaccineData(self):
        retries = 1
        success = False
        while not success:
            try:
                data = requests.get(
                    config.get('VaccineInfo', 'apiURL') + "?pincode=" + self.pinCode
                    + "&date=" + date.today().strftime("%d-%m-%Y"))

                centers = json.loads(data.text)
                return centers["centers"]
            except (ConnectionError, Timeout, HTTPError):
                time.sleep(10)
                retries += 1
                print("Retried", str(retries), "times.")


if __name__ == "__main__":
    # Load config from ini file
    config = configparser.ConfigParser()
    config.read(r'config.ini')

    # Initialize pygame mixer to play Alarm
    pygame.mixer.init()
    pygame.mixer.music.load(config.get('Alarm', 'alarmFile'))

    # Get the list of pin-codes configured in ini file
    pinCodeList = json.loads(config.get('Filters', 'pinCodes'))

    # We are not stopping until we do
    while True:
        for pinCode in pinCodeList:

            # Get the vaccine info for selected pin-code
            info = VaccineInfo(pinCode)

            # Filter the centres that don't have the required fee type
            centers = list(filter(lambda center: center["fee_type"] == config.get('Filters', 'feeType'),
                             info.getVaccineData()))
            for center in centers:
                print("Trying", center["name"], str(center["pincode"]) + ".", "It is", center["fee_type"])

                # Filter the sessions that are not for desired age group
                sessions = list(filter(lambda session: (session["min_age_limit"] <=
                                                        int(config.get('Filters', 'minAge')))
                if "min_age_limit" in session else False, center["sessions"]))
                sessions = list(filter(lambda session: (session["max_age_limit"] >=
                                                        int(config.get('Filters', 'maxAge')))
                if "max_age_limit" in session else False, sessions))

                # Filter the sessions that don't have the required amount of doses
                sessions = list(filter(lambda session: session["available_capacity_dose2"] >=
                                                  int(config.get('Filters', 'minimumDose2')), sessions))
                sessions = list(filter(lambda session: session["available_capacity_dose1"] >=
                                                  int(config.get('Filters', 'minimumDose1')), sessions))

                # Print information of each matched center
                for session in sessions:
                    print("Trying", center["name"], center["pincode"], "for", session["date"] + ".", "It is",
                          center["fee_type"], "and has", session["available_capacity_dose2"], "dose2 and",
                          session["available_capacity_dose1"], "dose1")
                    print("Playing Music..")
                    # Play the alarm sound and wait for a few seconds
                    pygame.mixer.music.play()
                    time.sleep(float(config.get('Alarm', 'alarmTime')))

        # Wait for a few seconds before pulling data again
        time.sleep(float(config.get('VaccineInfo', 'retryInSeconds')))