script.py

import requests
import json
import os
import time
import datetime
from dotenv import load_dotenv
import re

load_dotenv()

API_TOKEN = os.getenv("API_TOKEN")
DATA_DIRECTORY = os.getenv("DATA_DIRECTORY")

VACCINE_TRACKER_BASE_URL = "https://cdn-api.co-vin.in/api/v2/appointment/sessions/public/calendarByPin"
TELEGRAM_BOT_BASE_URL = "https://api.telegram.org/bot"
REQUEST_HEADERS = {
	'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/50.0.2661.102 Safari/537.36'
}

USER_DATA_FILENAME = DATA_DIRECTORY + "/user_data.json"
DATA_OFFSET_FILE = DATA_DIRECTORY + "/data_offset.json"

def read_offset_data():
	offset_data = None
	if os.path.exists(DATA_OFFSET_FILE):
		infile = open(DATA_OFFSET_FILE, 'r')
		offset_data = infile.read()
	return offset_data

def save_offset_data(latest_offset):
	outfile = open(DATA_OFFSET_FILE, 'w')
	outfile.write(str(latest_offset))

def save_user_data(user_id_pincode_dict):
	with open(USER_DATA_FILENAME, 'w') as outfile:
		json.dump(user_id_pincode_dict, outfile, indent=2)

def read_user_data():
	user_id_pincode_dict = {}
	if os.path.exists(USER_DATA_FILENAME):
		with open(USER_DATA_FILENAME, 'r') as infile:  
			user_id_pincode_dict = json.load(infile)
	return user_id_pincode_dict

def is_valid_pincode(pincode):
	pattern = "^[1-9]{1}[0-9]{2}\\s{0,1}[0-9]{3}$"
	match_compiler = re.compile(pattern)
	
	is_valid = re.match(match_compiler, pincode);

	if is_valid is None:
		return False
	else:
		return True

def get_new_user_data(user_id_pincode_dict, offset_id=None):
	bot_update_url = "https://api.telegram.org/bot{}/getUpdates".format(API_TOKEN)

	if offset_id:
		bot_update_url += "?offset={}".format(offset_id)

	latest_offset_id = offset_id

	response = requests.get(bot_update_url, headers=REQUEST_HEADERS)

	json_data = response.json()

	if len(json_data['result']) == 0:
		print("NO USERS UNTIL NOW.")
		return False

	result_list = json_data['result']

	for data in result_list:
		if 'update_id' in data:
			if latest_offset_id == data['update_id']:
				return
			latest_offset_id = data['update_id']
		if 'message' in data and 'text' in data['message']:
			chat_id = data['message']['chat']['id']
	
			if 'unsub_pincode' in data['message']['text']:
				text_list = data['message']['text'].split(" ")

				if len(text_list) > 1:
					pincode = text_list.pop()

					if is_valid_pincode(pincode) is False:
						continue
					
					if pincode in user_id_pincode_dict and chat_id in user_id_pincode_dict[pincode]:
						user_id_pincode_dict[pincode].remove(chat_id)
			
			elif 'pincode' in data['message']['text']:
				text_list = data['message']['text'].split(" ")

				if len(text_list) > 1:
					pincode = text_list.pop()
					
					if is_valid_pincode(pincode) is False:
						continue
					
					if pincode in user_id_pincode_dict:
						user_id_pincode_dict[pincode].append(chat_id)
					else:
						user_id_pincode_dict[pincode] = [chat_id]

	user_data_without_duplicates = {}

	for key in user_id_pincode_dict:
		if len(user_id_pincode_dict[key]) == 0:
			continue
		user_data_without_duplicates[key] = list(set(user_id_pincode_dict[key]))
	
	# print(user_id_pincode_dict)
	save_user_data(user_data_without_duplicates)

	save_offset_data(latest_offset_id)
	
	return 

def fetch_data(pincode):
	try:

		current_date = datetime.date.today()
		formatted_date = current_date.strftime("%d-%m-%Y")
		
		API_URL = VACCINE_TRACKER_BASE_URL + '?pincode={}&date={}'.format(pincode, formatted_date)
		response = requests.get(API_URL, headers=REQUEST_HEADERS)

		if response.status_code != 200:
			return []

		json_data = response.json()
		centre_list = []

		if 'centers' in json_data:
			centre_list = json_data['centers']
		
		return centre_list

	except Exception as e:
		print("ERROR:{}".format(e))


def identify_available_slots(centre_list):
	free_slots = []

	data_by_group = [
		{
			"group": 18,
			"data_count": 0,
			"has_data": False,
			"grouped_msgs": {},
			"msg": "Vaccine Slots for Age group: 18-44: \n",
		},
		{
			"group": 45,
			"data_count": 0,
			"has_data": False,
			"grouped_msgs": {},
			"msg": "Vaccine Slots for Age group: 45+: \n"
		}
	]

	for centre in centre_list:
		free_centre = {}
		if len(centre['sessions']) == 0:
			continue

		for session in centre['sessions']:

			if session['available_capacity'] > 50:
				free_centre = {
					"name": centre['name'],
					"address": centre['address'], 
					"district": centre['district_name'], 
					"block": centre['block_name'],
					"pincode": centre['pincode'],
					"vaccine": session['vaccine'],
					"date": session['date'],
					"type": centre['fee_type'],
					"group": session['min_age_limit'],
					"quantity": session['available_capacity']
				}
				
				group_idx = 0

				if session['min_age_limit'] == 45:
					group_idx = 1
				
				data_by_group[group_idx]['data_count'] += 1
				data_by_group[group_idx]['has_data'] = True

				age_group = str(session['min_age_limit']) + "+"
		
				formatted_msg = "\n{}".format(prepare_msg(free_centre))

				if session['date'] in data_by_group[group_idx]['grouped_msgs']:
					data_by_group[group_idx]['grouped_msgs'][session['date']]['msg'] += '\n' + formatted_msg
				else: 
					data_by_group[group_idx] = {
						**data_by_group[group_idx],
						'grouped_msgs': {
							**data_by_group[group_idx]['grouped_msgs'],
							session['date']: {
								'msg': "\n\nAge group: {}\nDate: {}\n{}".format(age_group,session['date'],formatted_msg)
							}
						}
					}
				
				free_slots.append(free_centre)

	return data_by_group

def prepare_msg(data_dict):
	msg = "{}, \nAddress: {}, {}, Pincode: {}, \nVaccine Name: {}, \nType: {} \nAvailable Quantity: {}".format(
		data_dict['name'], 
		data_dict['address'], 
		data_dict['district'], 
		str(data_dict['pincode']), 
		data_dict['vaccine'], 
		data_dict['type'], 
		str(data_dict['quantity'])
	)
	return msg

def send_notification(msg_dict, recipient_ids):


	for msg_group in msg_dict:
		if msg_group['has_data']:			
			for recipient_id in recipient_ids:

				for date_key in msg_group['grouped_msgs']:
					msg_data = msg_group['grouped_msgs'][date_key]['msg']
					print("-------------------------------------")
					print('SENDING DATA: {}'.format(msg_data))

					time.sleep(1)

					try:
						response = requests.get("https://api.telegram.org/bot{}/sendMessage?chat_id={}&text={}".format(API_TOKEN, recipient_id, msg_data), headers=REQUEST_HEADERS)


						if response.status_code != 200:
							print("Notification not sent for recipient id: {}, error: {}".format(str(recipient_id), response.text))
					except Exception as e:
						print("CANNOT SEND MSG:{}".format(e))

def main():

	latest_offset = read_offset_data()

	user_data = read_user_data()

	get_new_user_data(user_data, latest_offset)

	print("Available user_data: {}".format(user_data))

	for pincode in user_data:	
		response_data = fetch_data(pincode)

		msg_dict = identify_available_slots(response_data)

		send_notification(msg_dict, user_data[pincode])



if __name__ == "__main__":
	start_time = datetime.datetime.now()
	print("Script started at : {}".format(str(start_time)))
	main()
	run_time = datetime.datetime.now() - start_time
	print("Total Runtime : {}".format(str(run_time)))