Skip to content

Commit 6cb3b4d

Browse files
committed
Update notion.py error.py
1 parent fc3cff4 commit 6cb3b4d

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

src/arguments/error.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,13 @@ def evoke_search_error(self, error_statement):
2121

2222
class LoginError():
2323
def __init__(self, error_statement, success=False):
24-
# the error statement
25-
self.error_statement = error_statement
24+
"""Implements error printing for User Login
2625
27-
# success determines the color of message
26+
Error statement is printed in 'Green' or 'Red'
27+
Color depending upon the value of success variable
28+
"""
29+
self.error_statement = error_statement
2830
self.success = success
29-
3031
self.evoke_search_error()
3132

3233
def evoke_search_error(self):

src/arguments/notion.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,6 @@
99
from selenium.webdriver.support.ui import WebDriverWait
1010
from selenium.webdriver.support import expected_conditions as EC
1111

12-
from rich import console
13-
1412
def get_token_from_cookie(cookie, token):
1513
for el in cookie:
1614
if el['name'] == token:
@@ -80,7 +78,7 @@ def get_tokenv2_cookie(self):
8078
try:
8179
cookies = get_cookies_from_login()
8280
self.tokenv2_cookie = get_token_from_cookie(cookies, 'token_v2')
83-
except Exception as e:
81+
except Exception:
8482
self.tokenv2_cookie = None
8583
finally:
8684
if self.tokenv2_cookie:

0 commit comments

Comments
 (0)