def main():
parser = argparse.ArgumentParser()
parser.add_argument('--token', help='Pass token directly')
args = parser.parse_args()
filename = 'vaccine-booking-details.json'
mobile = None
print('Running Script')
beep(500, 150)
try:
base_request_header = {
'User-Agent':
'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36',
}
if args.token:
token = args.token
else:
mobile = input("Enter the registered mobile number: ")
token = generate_token_OTP(mobile, base_request_header)
request_header = copy.deepcopy(base_request_header)
request_header["Authorization"] = f"Bearer {token}"
if os.path.exists(filename):
print(
"\n=================================== Note ===================================\n"
)
print(
f"Info from perhaps a previous run already exists in {filename} in this directory."
)
print(
f"IMPORTANT: If this is your first time running this version of the application, DO NOT USE THE FILE!"
)
try_file = input(
"Would you like to see the details and confirm to proceed? (y/n Default y): "
)
try_file = try_file if try_file else 'y'
if try_file == 'y':
collected_details = get_saved_user_info(filename)
print(
"\n================================= Info =================================\n"
)
display_info_dict(collected_details)
file_acceptable = input(
"\nProceed with above info? (y/n Default n): ")
file_acceptable = file_acceptable if file_acceptable else 'n'
if file_acceptable != 'y':
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
else:
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
else:
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
confirm_and_proceed(collected_details)
info = SimpleNamespace(**collected_details)
token_valid = True
while token_valid:
request_header = copy.deepcopy(base_request_header)
request_header["Authorization"] = f"Bearer {token}"
# call function to check and book slots
token_valid = check_and_book(request_header,
info.beneficiary_dtls,
info.location_dtls,
info.search_option,
min_slots=info.minimum_slots,
ref_freq=info.refresh_freq,
auto_book=info.auto_book,
start_date=info.start_date,
vaccine_type=info.vaccine_type,
fee_type=info.fee_type)
# check if token is still valid
beneficiaries_list = requests.get(BENEFICIARIES_URL,
headers=request_header)
if beneficiaries_list.status_code == 200:
token_valid = True
else:
# if token invalid, regenerate OTP and new token
beep(WARNING_BEEP_DURATION[0], WARNING_BEEP_DURATION[1])
print('Token is INVALID.')
token_valid = False
tryOTP = input('Try for a new Token? (y/n Default y): ')
if tryOTP.lower() == 'y' or not tryOTP:
if not mobile:
mobile = input("Enter the registered mobile number: ")
token = generate_token_OTP(mobile, base_request_header)
token_valid = True
else:
print("Exiting")
os.system("pause")
except Exception as e:
print(str(e))
print('Exiting Script')
os.system("pause")
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--token", help="Pass token directly")
parser.add_argument("--mobile", help="Pass mobile directly")
parser.add_argument("--kvdb-bucket", help="Pass kvdb.io bucket directly")
parser.add_argument("--config", help="Config file name")
parser.add_argument(
"--no-tty",
help="Do not ask any terminal inputs. Proceed with smart choices",
action="store_false",
)
args = parser.parse_args()
if args.config:
filename = args.config
else:
filename = "vaccine-booking-details-"
if args.mobile:
mobile = args.mobile
else:
mobile = None
if args.kvdb_bucket:
kvdb_bucket = args.kvdb_bucket
else:
kvdb_bucket = KVDB_BUCKET
print("Running Script")
beep(500, 150)
try:
base_request_header = {
"User-Agent":
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36",
"origin": "https://selfregistration.cowin.gov.in/",
"referer": "https://selfregistration.cowin.gov.in/",
}
token = None
otp_pref = "n"
if args.token:
token = args.token
else:
if mobile is None:
mobile = input("Enter the registered mobile number: ")
if not args.config:
filename = filename + mobile + ".json"
if not kvdb_bucket:
otp_pref = (input(
"\nDo you want to enter OTP manually, instead of auto-read? \nRemember selecting n would require some setup described in README (y/n Default n): "
) if args.no_tty else "n")
otp_pref = otp_pref if otp_pref else "n"
if otp_pref == "n":
kvdb_bucket = (input(
"Please refer KVDB setup in ReadMe to setup your own KVDB bucket. Please enter your KVDB bucket value here: "
) if args.no_tty and kvdb_bucket is None else kvdb_bucket)
if not kvdb_bucket:
print(
"Sorry, having your private KVDB bucket is mandatory. Please refer ReadMe and create your own private KVBD bucket."
)
sys.exit()
if kvdb_bucket:
print(
"\n### Note ### Please make sure the URL configured in the IFTTT/Shortcuts app on your phone is: "
+ "https://kvdb.io/" + kvdb_bucket + "/" + mobile + "\n")
while token is None:
if otp_pref == "n":
try:
token = generate_token_OTP(mobile, base_request_header,
kvdb_bucket)
except Exception as e:
print(str(e))
print("OTP Retrying in 5 seconds")
time.sleep(5)
elif otp_pref == "y":
token = generate_token_OTP_manual(mobile,
base_request_header)
request_header = copy.deepcopy(base_request_header)
request_header["Authorization"] = f"Bearer {token}"
if os.path.exists(filename):
print(
"\n=================================== Note ===================================\n"
)
print(
f"Info from perhaps a previous run already exists in {filename} in this directory."
)
print(
f"IMPORTANT: If this is your first time running this version of the application, DO NOT USE THE FILE!"
)
try_file = (input(
"Would you like to see the details and confirm to proceed? (y/n Default y): "
) if args.no_tty else "y")
try_file = try_file if try_file else "y"
if try_file == "y":
collected_details = get_saved_user_info(filename)
print(
"\n================================= Info =================================\n"
)
display_info_dict(collected_details)
file_acceptable = (
input("\nProceed with above info? (y/n Default y): ")
if args.no_tty else "y")
file_acceptable = file_acceptable if file_acceptable else "y"
if file_acceptable != "y":
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
else:
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
else:
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
confirm_and_proceed(collected_details, args.no_tty)
# HACK: Temporary workaround for not supporting reschedule appointments
beneficiary_ref_ids = [
beneficiary["bref_id"]
for beneficiary in collected_details["beneficiary_dtls"]
]
beneficiary_dtls = fetch_beneficiaries(request_header)
if beneficiary_dtls.status_code == 200:
beneficiary_dtls = [
beneficiary
for beneficiary in beneficiary_dtls.json()["beneficiaries"] if
beneficiary["beneficiary_reference_id"] in beneficiary_ref_ids
]
active_appointments = []
for beneficiary in beneficiary_dtls:
expected_appointments = (1 if beneficiary["vaccination_status"]
== "Partially Vaccinated" else 0)
if len(beneficiary["appointments"]) > expected_appointments:
data = beneficiary["appointments"][expected_appointments]
beneficiary_data = {
"name": data["name"],
"state_name": data["state_name"],
"dose": data["dose"],
"date": data["date"],
"slot": data["slot"],
}
active_appointments.append({
"beneficiary":
beneficiary["name"],
**beneficiary_data
})
if active_appointments:
print(
"The following appointments are active! Please cancel them manually first to continue"
)
display_table(active_appointments)
beep(WARNING_BEEP_DURATION[0], WARNING_BEEP_DURATION[1])
return
else:
print(
"WARNING: Failed to check if any beneficiary has active appointments. Please cancel before using this script"
)
if args.no_tty:
input("Press any key to continue execution...")
info = SimpleNamespace(**collected_details)
if info.find_option == 1:
disable_re_assignment_feature()
while True: # infinite-loop
# create new request_header
request_header = copy.deepcopy(base_request_header)
request_header["Authorization"] = f"Bearer {token}"
# call function to check and book slots
try:
token_valid = is_token_valid(token)
# token is invalid ?
# If yes, generate new one
if not token_valid:
print("Token is INVALID.")
token = None
while token is None:
if otp_pref == "n":
try:
token = generate_token_OTP(
mobile, base_request_header, kvdb_bucket)
except Exception as e:
print(str(e))
print("OTP Retrying in 5 seconds")
time.sleep(5)
elif otp_pref == "y":
token = generate_token_OTP_manual(
mobile, base_request_header)
check_and_book(
request_header,
info.beneficiary_dtls,
info.location_dtls,
info.pin_code_location_dtls,
info.find_option,
info.search_option,
min_slots=info.minimum_slots,
ref_freq=info.refresh_freq,
# auto_book=info.auto_book,
start_date=info.start_date,
vaccine_type=info.vaccine_type,
fee_type=info.fee_type,
mobile=mobile,
# captcha_automation=info.captcha_automation,
dose_num=get_dose_num(collected_details),
)
except Exception as e:
print(str(e))
print("Retryin in 5 seconds")
time.sleep(5)
except Exception as e:
print(str(e))
print("Exiting Script")
os.system("pause")
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--token', help='Pass token directly')
args = parser.parse_args()
filename = 'vaccine-booking-details-'
mobile = None
print('Running Script')
beep(500, 150)
try:
base_request_header = {
'User-Agent':
'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36',
}
token = None
if args.token:
token = args.token
else:
mobile = input("Enter the registered mobile number: ")
filename = filename + mobile + ".json"
otp_pref = input(
"\nDo you want to enter OTP manually, instead of auto-read? \nRemember selecting n would require some setup described in README (y/n Default n): "
)
otp_pref = otp_pref if otp_pref else "n"
while token is None:
if otp_pref == "n":
try:
token = generate_token_OTP(mobile, base_request_header)
except Exception as e:
print(str(e))
print('OTP Retrying in 5 seconds')
time.sleep(5)
elif otp_pref == "y":
token = generate_token_OTP_manual(mobile,
base_request_header)
request_header = copy.deepcopy(base_request_header)
request_header["Authorization"] = f"Bearer {token}"
if os.path.exists(filename):
print(
"\n=================================== Note ===================================\n"
)
print(
f"Info from perhaps a previous run already exists in {filename} in this directory."
)
print(
f"IMPORTANT: If this is your first time running this version of the application, DO NOT USE THE FILE!"
)
try_file = input(
"Would you like to see the details and confirm to proceed? (y/n Default y): "
)
try_file = try_file if try_file else 'y'
if try_file == 'y':
collected_details = get_saved_user_info(filename)
print(
"\n================================= Info =================================\n"
)
display_info_dict(collected_details)
file_acceptable = input(
"\nProceed with above info? (y/n Default n): ")
file_acceptable = file_acceptable if file_acceptable else 'n'
if file_acceptable != 'y':
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
else:
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
else:
collected_details = collect_user_details(request_header)
save_user_info(filename, collected_details)
confirm_and_proceed(collected_details)
# HACK: Temporary workaround for not supporting reschedule appointments
beneficiary_ref_ids = [
beneficiary["bref_id"]
for beneficiary in collected_details["beneficiary_dtls"]
]
beneficiary_dtls = fetch_beneficiaries(request_header)
if beneficiary_dtls.status_code == 200:
beneficiary_dtls = [
beneficiary
for beneficiary in beneficiary_dtls.json()['beneficiaries'] if
beneficiary['beneficiary_reference_id'] in beneficiary_ref_ids
]
active_appointments = []
for beneficiary in beneficiary_dtls:
expected_appointments = (1 if beneficiary['vaccination_status']
== "Partially Vaccinated" else 0)
if len(beneficiary['appointments']) > expected_appointments:
data = beneficiary['appointments'][expected_appointments]
beneficiary_data = {
'name': data['name'],
'state_name': data['state_name'],
'dose': data['dose'],
'date': data['date'],
'slot': data['slot']
}
active_appointments.append({
"beneficiary":
beneficiary['name'],
**beneficiary_data
})
if active_appointments:
print(
"The following appointments are active! Please cancel them manually first to continue"
)
display_table(active_appointments)
beep(WARNING_BEEP_DURATION[0], WARNING_BEEP_DURATION[1])
return
else:
print(
"WARNING: Failed to check if any beneficiary has active appointments. Please cancel before using this script"
)
input("Press any key to continue execution...")
info = SimpleNamespace(**collected_details)
token_valid = True
while token_valid:
request_header = copy.deepcopy(base_request_header)
request_header["Authorization"] = f"Bearer {token}"
# call function to check and book slots
try:
token_valid = check_and_book(
request_header,
info.beneficiary_dtls,
info.location_dtls,
info.search_option,
min_slots=info.minimum_slots,
ref_freq=info.refresh_freq,
auto_book=info.auto_book,
start_date=info.start_date,
vaccine_type=info.vaccine_type,
fee_type=info.fee_type,
mobile=mobile,
captcha_automation=info.captcha_automation,
captcha_automation_api_key=info.captcha_automation_api_key,
dose_num=get_dose_num(collected_details))
# check if token is still valid
beneficiaries_list = requests.get(BENEFICIARIES_URL,
headers=request_header)
if beneficiaries_list.status_code == 200:
token_valid = True
else:
# if token invalid, regenerate OTP and new token
# beep(WARNING_BEEP_DURATION[0], WARNING_BEEP_DURATION[1])
print('Token is INVALID.')
token_valid = False
token = None
while token is None:
if otp_pref == "n":
try:
token = generate_token_OTP(
mobile, base_request_header)
except Exception as e:
print(str(e))
print('OTP Retrying in 5 seconds')
time.sleep(5)
elif otp_pref == "y":
token = generate_token_OTP_manual(
mobile, base_request_header)
token_valid = True
except Exception as e:
print(str(e))
print('Retryin in 5 seconds')
time.sleep(5)
except Exception as e:
print(str(e))
print('Exiting Script')
os.system("pause")