fix: indentation

metakgp · Nov 28, 2024 · 6d6fcaa · 6d6fcaa
1 parent 82bd004
commit 6d6fcaa
Showing 1 changed file with 90 additions and 53 deletions.
diff --git a/mftp/mftp.py b/mftp/mftp.py
@@ -13,72 +13,109 @@
 import iitkgp_erp_login.erp as erp
 
 headers = {
-  'timeout': '20',
-  'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Ubuntu Chromium/51.0.2704.79 Chrome/51.0.2704.79 Safari/537.36',
+    "timeout": "20",
+    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Ubuntu Chromium/51.0.2704.79 Chrome/51.0.2704.79 Safari/537.36",
 }
 session = requests.Session()
 
-parser = argparse.ArgumentParser(description='One stop mailing solution for CDC NoticeBoard at IIT KGP')
-parser.add_argument('--smtp', action="store_true", help='Use SMTP for sending the mails', required=False)
-parser.add_argument('--gmail-api', action="store_true", help='Use GMAIL API for sending the mails', required=False)
-parser.add_argument('--ntfy', action="store_true", help='Use NTFY to broadcast notifications', required=False)
-parser.add_argument('--cron', action="store_true", help='Act as cronjob, bypass the continuous loop', required=False)
+parser = argparse.ArgumentParser(
+    description="One stop mailing solution for CDC NoticeBoard at IIT KGP"
+)
+parser.add_argument(
+    "--smtp", action="store_true", help="Use SMTP for sending the mails", required=False
+)
+parser.add_argument(
+    "--gmail-api",
+    action="store_true",
+    help="Use GMAIL API for sending the mails",
+    required=False,
+)
+parser.add_argument(
+    "--ntfy",
+    action="store_true",
+    help="Use NTFY to broadcast notifications",
+    required=False,
+)
+parser.add_argument(
+    "--cron",
+    action="store_true",
+    help="Act as cronjob, bypass the continuous loop",
+    required=False,
+)
 args = parser.parse_args()
 
 while True:
-  now = datetime.now()
-  print(f"================ <<: {now.strftime('%H:%M:%S %d-%m-%Y')} :>> ================", flush=True)
+    now = datetime.now()
+    print(
+        f"================ <<: {now.strftime('%H:%M:%S %d-%m-%Y')} :>> ================",
+        flush=True,
+    )
 
-  print('[ERP LOGIN]', flush=True)
-  _, ssoToken = erp.login(headers, session, ERPCREDS=env, OTP_CHECK_INTERVAL=2, LOGGING=True, SESSION_STORAGE_FILE='.session')
+    print("[ERP LOGIN]", flush=True)
+    _, ssoToken = erp.login(
+        headers,
+        session,
+        ERPCREDS=env,
+        OTP_CHECK_INTERVAL=2,
+        LOGGING=True,
+        SESSION_STORAGE_FILE=".session",
+    )
 
-  if args.gmail_api or args.smtp:
-    _, new, modified = company.fetch(session, headers, ssoToken)
+    if args.gmail_api or args.smtp:
+        _, new, modified = company.fetch(session, headers, ssoToken)
 
-    if new:
-      print('[NEW COMPANIES]', flush=True)
-      for com in new:
-        logging.info(f' {com["Name"]} | {com["Role"]} | {com["CTC"]} | {com["End_Date"]} | {com["Interview_Date"]}')
-    if modified:
-      print('[MODIFIED COMPANIES]', flush=True)
-      for com in modified:
-        logging.info(f' {com["Name"]} | {com["Role"]} | {com["CTC"]} | {com["End_Date"]} | {com["Interview_Date"]}')
+        if new:
+            print("[NEW COMPANIES]", flush=True)
+            for com in new:
+                logging.info(
+                    f" {com['Name']} | {com['Role']} | {com['CTC']} | {com['End_Date']} | {com['Interview_Date']}"
+                )
+        if modified:
+            print("[MODIFIED COMPANIES]", flush=True)
+            for com in modified:
+                logging.info(
+                    f" {com['Name']} | {com['Role']} | {com['CTC']} | {com['End_Date']} | {com['Interview_Date']}"
+                )
 
-    filtered = []
-    if new + modified:
-      filtered = company.filter(new + modified, "OPEN_N")
-      if filtered:
-        for com in filtered:
-          logging.info(f' {com["Name"]} | {com["Role"]} | {com["CTC"]} | {com["End_Date"]} | {com["Interview_Date"]}')
+        filtered = []
+        if new + modified:
+            filtered = company.filter(new + modified, "OPEN_N")
+            if filtered:
+                for com in filtered:
+                    logging.info(
+                        f" {com['Name']} | {com['Role']} | {com['CTC']} | {com['End_Date']} | {com['Interview_Date']}"
+                    )
 
-          latest_ssoToken = session.cookies.get('ssoToken')
-          mail_subject = "APPLY NOW! New companies opened"
-          companies_mail = mail.format_companies(latest_ssoToken, filtered, mail_subject)
-          mail.send_companies(companies_mail, args.gmail_api, args.smtp)
-    else:
-      print("[NO NEW COMPANIES]")
+                latest_ssoToken = session.cookies.get("ssoToken")
+                mail_subject = "APPLY NOW! New companies opened"
+                companies_mail = mail.format_companies(
+                    latest_ssoToken, filtered, mail_subject
+                )
+                mail.send_companies(companies_mail, args.gmail_api, args.smtp)
+        else:
+            print("[NO NEW COMPANIES]")
 
-  notice_db = db.NoticeDB(config={
-    'uri': env.MONGO_URI,
-    'db_name': env.MONGO_DATABASE
-  }, collection_name=env.MONGO_COLLECTION)
-  notice_db.connect()
+    notice_db = db.NoticeDB(
+        config={"uri": env.MONGO_URI, "db_name": env.MONGO_DATABASE},
+        collection_name=env.MONGO_COLLECTION,
+    )
+    notice_db.connect()
 
-  notices = notice.fetch(headers, session, ssoToken, notice_db)
-  if notices:
-    if args.ntfy:
-      notifications = ntfy.format_notices(notices)
-      if notifications:
-          ntfy.send_notices(notifications, notice_db)
+    notices = notice.fetch(headers, session, ssoToken, notice_db)
+    if notices:
+        if args.ntfy:
+            notifications = ntfy.format_notices(notices)
+            if notifications:
+                ntfy.send_notices(notifications, notice_db)
+        else:
+            mails = mail.format_notices(notices)
+            if mails:
+                mail.send_notices(mails, args.smtp, args.gmail_api, notice_db)
     else:
-      mails = mail.format_notices(notices)
-      if mails:
-          mail.send_notices(mails, args.smtp, args.gmail_api, notice_db)
-  else:
-    print('[NO NEW NOTICES]', flush=True)
+        print("[NO NEW NOTICES]", flush=True)
 
-  if args.cron:
-    break
+    if args.cron:
+        break
 
-  print('[PAUSED FOR 2 MINUTES]', flush=True)
-  time.sleep(120)
+    print("[PAUSED FOR 2 MINUTES]", flush=True)
+    time.sleep(120)