From 3b66d1e379a537b3d7d3621d54c3f294e1c2f39b Mon Sep 17 00:00:00 2001
From: SavageCore <talk@savagecore.uk>
Date: Thu, 13 Mar 2025 00:51:41 +0000
Subject: [PATCH] Aither 2FA support

---
 .gitignore     | 2 ++
 screenshots.py | 7 +++++++
 2 files changed, 9 insertions(+)

diff --git a/.gitignore b/.gitignore
index ba35e2a..f796c36 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,3 +2,5 @@ config.ini
 screenshots
 downloaded_files
 __pycache__
+
+secrets.ini
diff --git a/screenshots.py b/screenshots.py
index c81ac0b..328b3ef 100644
--- a/screenshots.py
+++ b/screenshots.py
@@ -79,6 +79,13 @@ def take_screenshot(tracker_name, driver=driver, is_load_at_runtime=False):
 
     # Login and save screenshot
     driver.find_element(By.CLASS_NAME, "auth-form__primary-button").click()
+
+    # 2FA is enabled, ask for the code
+    if driver.find_elements(By.ID, "code"):
+        code = input("Please enter the 2FA code: ")
+        code_field = driver.find_element(By.ID, "code")
+        code_field.send_keys(code)
+
     take_screenshot("Aither")
     print("Aither Screenshoted")