From 5f17e441b981de388268c250547d4f6b59ba3d1c Mon Sep 17 00:00:00 2001
From: roShan <45506201+truroshan@users.noreply.github.com>
Date: Thu, 13 May 2021 11:30:25 +0530
Subject: [PATCH] Update

---
 cowin.py | 87 +++++++++++++++++++++++++++++++++++---------------------
 1 file changed, 55 insertions(+), 32 deletions(-)

diff --git a/cowin.py b/cowin.py
index 9a50bb5..3624c73 100644
--- a/cowin.py
+++ b/cowin.py
@@ -9,37 +9,49 @@ import re
 cowin = CoWinAPI()
 scheduler = BlockingScheduler()
 
+def line_break():
+    print("-"*25)
+
+def clear_screen():
+    os.system("clear")
+
 def notify():
     available_centers = cowin.get_availability_by_pincode(PINCODE,min_age_limt=AGE)
 
-    AVAILABLE = False
     for center in available_centers.get('centers',[]):
-        for session in center.get('sessions'):
-            capacity = session.get('available_capacity')
+
+        for session in center.get('sessions')[1:]:  # Starting from Next Day
+
             center_name = center.get('name')
+            center_id = center.get('center_id')
+
+            capacity = session.get('available_capacity')
             session_date = session.get('date')
-            if capacity != 0 and center.get('center_id') in  CENTER_ID:
-                
-                MSG = f'{capacity} slots left [{session_date}] @ {center_name} ({PINCODE})'
+            vaccine_name = session.get('vaccine')
+
+            if capacity != 0 and center_id in  CENTER_ID:
+
+                MSG = f'💉 {capacity} #{vaccine_name} / {session_date} / {center_name} 📍{PINCODE}'
 
                 # Send Notification via Termux:API App
                 os.system(f"termux-notification --content '{MSG}'")
-                AVAILABLE =  True
-    
+
+                CENTER_ID.remove(center_id)
+
     # When last Checked
-    print("Last Checked  ⌛️ : " + datetime.now().strftime("%H:%M:%S"))
+    print("Last Checked  ✅ : " + datetime.now().strftime("%H:%M:%S") + " 🕐")
     sys.stdout.write("\033[F")
-    
-    # Stop Scheduler
-    if AVAILABLE:
+
+    # when CENTER_ID list is empty Stop Scheduler
+    if not CENTER_ID:
         print("Shutting Down CoWin Script 👩‍💻 ")
         scheduler.shutdown(wait=False)
-        
 
-def main(pincode, age = 45,time = 1):
+
+def main(pincode, age = 18,time = 1):
 
     if age < 18 :
-        print("Age is less than 18.") 
+        print("Age is less than 18.")
         return
     else:
         age = 18 if 18 <= age < 45 else 45
@@ -47,46 +59,57 @@ def main(pincode, age = 45,time = 1):
     available_centers = cowin.get_availability_by_pincode(str(pincode),min_age_limt=age)
 
     CENTERS = {}
+    INDEX_S = []
 
-    print("Select Vaccination Center 🏥 :")
+    print(f"Select Vaccination Center ({pincode}) 💉 \n")
     for index,center in enumerate(available_centers.get('centers',[]),start=1):
         print(f'{index} : {center.get("name")}')
         CENTERS[index] = center.get('center_id')
-    
+
+        INDEX_S.append(index)
+
+    print()
 
     global CENTER_ID, AGE, PINCODE
 
-    print("-"*20)
-    print("Enter Multiple Value with Space. \n(Example : 1 2 3 4)")
-    print("-"*20)
+    line_break()
+    print("""
+* Select One Center
+    input : 1
+* Select Mutiple with Space
+    input : 1 2 3 4
+* Select All Center
+    Hit Enter without Input\n""")
 
-    INDEX_S = input("Enter Index's : ")
+    line_break()
 
-    INDEX_S = re.findall("(\d)",INDEX_S)
-    
+    input_index = input("Enter Index's : ")
+
+    if input_index != '':
+        INDEX_S = re.findall("(\d)",input_index)
+
+    clear_screen()
 
     CENTER_ID = []
     for  index in INDEX_S:
         if CENTERS.get(int(index)):
             CENTER_ID.append(CENTERS.get(int(index)))
 
-    if not INDEX_S or not CENTER_ID:
-        print("No Index Selected\n Program exited.")
-        exit()
-
     AGE, PINCODE = age,str(pincode)
 
-    os.system("clear")
-
-    scheduler.add_job(notify, 'cron',hour = "10-22", minute = f'0-59/{time}')
+    clear_screen()
 
+    scheduler.add_job(notify, 'cron',hour = "8-22", minute = f'0-59/{time}')
+    print(f" 📍 {PINCODE} 💉 {AGE}+ ⌛️ {time} Minute")
 
 
 
 if __name__ == '__main__':
-    
+    clear_screen()
+
     fire.Fire(main)
 
-    print("CoWin Slot Checker 💉 ..")
+    print("CoWin Slot Checking 🔃\nfor Tomorrow and Day After 📆 ...")
+    line_break()
 
     scheduler.start()