Download_appointment.py

import requests
from fake_useragent import UserAgent
import hashlib
import json


class CoWinCancel():

    def line_break(self): 
        print("-"*25)
    def __init__(self,mobile_no):
        self.mobile_no = str(mobile_no)

        # All users data with active booking
        self.user_data = []

        # Request Session
        self.session =  requests.Session() 

        # Data for sending request
        self.data = {} 
        self.appointmentID = self.getAppointmentID()

        # Token Received from CoWIN
        self.bearerToken = None  # Session Token

      
        # Login and Save Token in file( filename same as mobile no)
        self.getSession()



    # Set Header in self.session = requests.Session()
    def set_headers(self):
        temp_user_agent = UserAgent()
        self.session.headers.update({
            'User-Agent': temp_user_agent.random,
            'Accept': 'application/json, text/plain, */*',
            'Accept-Language': 'en-US,en;q=0.5',
            'Content-Type': 'application/json',
            'Origin': 'https://selfregistration.cowin.gov.in',
            'Connection': 'keep-alive',
            'Referer': 'https://selfregistration.cowin.gov.in/',
            'TE': 'Trailers',
        })
    
    # returning self.data 
    def get_data(self):
        return json.dumps(self.data).encode('utf-8')

    # Save Token after login to CoWIN
    def putSession(self):
        with open(f"{self.mobile_no}.log", "w") as f:
            f.write(self.bearerToken)

    # Get Token saved in file for re-login and use
    def getSession(self):
        self.set_headers()
        try:
            with open(f"{self.mobile_no}.log", "r") as f:
                self.bearerToken = f.read()
            self.session.headers.update({
                    'Authorization': 'Bearer {}'.format(self.bearerToken)
                })
            self.session.get('https://cdn-api.co-vin.in/api/v2/appointment/beneficiaries').json()
        except (FileNotFoundError,json.decoder.JSONDecodeError):
            self.login_cowin()
            

    # Login to selfregistration.cowin.gov.in/
    def login_cowin(self):

        self.data = {
        "secret":"U2FsdGVkX1+gGN13ULaCVtLSWmsyZwAdXXTIAvLQp2HOXrIBCcq0yyOZQqzzfiFiEYs7KoAOTK2j4qPF/sEVww==",
        "mobile": self.mobile_no
            }

        response = self.session.post('https://cdn-api.co-vin.in/api/v2/auth/generateMobileOTP',data=self.get_data())

        otpSha265 = self.get_otp()

        txn_id = response.json()['txnId']

        self.data = {
                        "otp":otpSha265,
                        "txnId": txn_id
                                    }
        
        response = self.session.post('https://cdn-api.co-vin.in/api/v2/auth/validateMobileOtp',data=self.get_data())
        
        self.bearerToken = response.json()['token']

        self.session.headers.update({
            'Authorization': 'Bearer {}'.format(self.bearerToken)
        })
        self.putSession() 

    # Request for OTP 
    def get_otp(self):
         
        print("OTP Sent 📲 ... \n")

        otp = input("Enter Otp Received on phone: ")

        return hashlib.sha256(otp.encode('utf-8')).hexdigest()

    # Show Registered Beneficiary Details
    def getAppointmentID(self):
        return input("Appointment ID: ")
        # response = self.session.get('https://cdn-api.co-vin.in/api/v2/appointment/beneficiaries').json()

        # USERS = []

        # if not response.get('beneficiaries',[]):
        #     print("No user added in beneficiaries")
        #     return
        # # print(json.dumps(response, indent=3))
        # counter = 1
        # for user in response.get('beneficiaries'):
        #     for appointments in user.get('appointments'):
        #         if appointments.get(f'appointment_id'):
        #             data = {
        #                     'index' : counter, 
        #                     'user_name' : f'{user.get("name")}',
        #                     'reference_id' : f'{user.get("beneficiary_reference_id")}',
        #                     'appointment_id' : f'{appointments.get("appointment_id")}',
        #                     'dose' : f'{appointments.get("dose")}',
        #                     'center':f'{appointments.get("name")}'
        #                     }
        #             USERS.append(data)
        #             counter +=1
         
        # self.user_data = USERS
   
    # Cancel booking
    def downloadBookings(self):
        response = self.session.get(f'https://cdn-api.co-vin.in/api/v2/appointment/appointmentslip/download?appointment_id={self.appointmentID}')
        status = response.status_code
        if status == 200:
            with open(f"{self.mobile_no}_appoint_slip_{self.appointmentID[:5]}.pdf", "wb") as f:
                f.write(response.content)
            exit()
        if status == 400:
            print(f'{status} : {response.json()}')

       

#  Driver Program
def main(mobile_no):
    newUser = CoWinCancel(mobile_no)
    result = newUser.downloadBookings()
    if result == True:
        print("\n**********************************************************************\nDo you want to download more appointment?")
        ans = input("Press Y to restart or press enter/return key to exit: ")
        if ans.lower()== 'y':
            main(mobile_no)
        else:
            pass
    elif result == -1:
        print("No active bookings found")


if __name__ == '__main__':
    mobile_no = input("Enter Your Mobile No: ")
    if len(mobile_no)==10:  
        main(mobile_no)
    else:
        print("Wrong Mobile Number")