terminate script on login failed

This commit is contained in:
Jarrian
2023-09-11 16:13:25 +08:00
parent 7575e20953
commit 88774e343c

View File

@@ -114,16 +114,43 @@ class Twitter_Scraper:
def _login(self): def _login(self):
print("Logging in to Twitter...") print("Logging in to Twitter...")
self.driver.get(TWITTER_LOGIN_URL)
self.driver.maximize_window()
sleep(3)
self._input_username() try:
self._input_unusual_activity() self.driver.get(TWITTER_LOGIN_URL)
self._input_password() self.driver.maximize_window()
sleep(3)
self._input_username()
self._input_unusual_activity()
self._input_password()
cookies = self.driver.get_cookies()
auth_token = None
for cookie in cookies:
if cookie["name"] == "auth_token":
auth_token = cookie["value"]
break
if auth_token is None:
raise ValueError(
"""This may be due to the following:
- Internet connection is unstable
- Username is incorrect
- Password is incorrect
"""
)
print()
print("Login Successful")
print()
except Exception as e:
print()
print(f"Login Failed: {e}")
sys.exit(1)
print("Login Successful")
print()
pass pass
def _input_username(self): def _input_username(self):