Fix codacy issues

Author: NamamiShanker

src/arguments/notion.py

@@ -1,8 +1,6 @@
 import os
-import sys
 
-from .utility import Utility
-from .error import SearchError
+from .utility import get_browser_driver
 from .settings import LOGIN_PATH
 from .settings import TOKEN_FILE_PATH
 
@@ -17,11 +15,18 @@ def get_token_from_cookie(cookie, token):
 
 class NotionClient():
     """
+    Implements Login and token retrieval
+
     Handles the entire procedure of connecting to User's Notion account,
-    generating Notion's tokenv2_cookie, storing it locally and uploading content
-    to User's space
+    generating Notion's tokenv2_cookie, storing it locally and uploading
+    content to User's space
     """
     def __init__(self):
+        """
+        No input parameters required for instatiating object.
+        tokenv2_cookie stores the cookie containing user's tokenv2
+        tokenv2_key is used to create environment variable
+        """
         self.tokenv2_cookie = None
         self.tokenv2_key = 'TOKENV2'
 
@@ -47,7 +52,7 @@ def get_cookies_from_login(self):
         Returns the user's cookies which can be used to
         access and transfer content to user's Notion account
         """
-        driver = Utility().get_browser_driver()
+        driver = get_browser_driver()
         try:
             driver.get(LOGIN_PATH)
             WebDriverWait(driver, 300).until(
@@ -64,7 +69,7 @@ def get_tokenv2_cookie(self):
         if not self.tokenv2_cookie:
             try:
                 self.tokenv2_cookie = self.get_token_from_file()
-            except Exception as e:
+            except Exception:
                 try:
                     cookies = self.get_cookies_from_login()
                     self.tokenv2_cookie = get_token_from_cookie(cookies, 'token_v2')

src/arguments/utility.py

@@ -35,6 +35,23 @@
 
 console = Console()
 
+def get_browser_driver():
+    # Try to install web drivers for one of these browsers
+    # Chrome, Firefox, Edge (One of them must be installed)
+    try:
+        return webdriver.Chrome(ChromeDriverManager().install())
+    except ValueError:
+        try:
+            return webdriver.Firefox(executable_path=
+                                        GeckoDriverManager().install())
+        except ValueError:
+            try:
+                return webdriver.Edge(EdgeChromiumDriverManager().\
+                                        install())
+            except ValueError:
+                print("You do not have one of these supported browsers:" +
+                        "Chrome, Firefox, Edge")
+
 class Playbook():
     def __init__(self):
         self.key = 'DYNAMIC'
@@ -350,23 +367,6 @@ def get_ans(self, questions_list):
         stackoverflow_panel.display_panel(questions_list)
         # Support for reddit searching can also be implemented from here
 
-    def get_browser_driver(self):
-        # Try to install web drivers for one of these browsers
-        # Chrome, Firefox, Edge (One of them must be installed)
-        try:
-            return webdriver.Chrome(ChromeDriverManager().install())
-        except ValueError:
-            try:
-                return webdriver.Firefox(executable_path=
-                                           GeckoDriverManager().install())
-            except ValueError:
-                try:
-                    return webdriver.Edge(EdgeChromiumDriverManager().\
-                                            install())
-                except ValueError:
-                    print("You do not have one of these supported browsers:" +
-                          "Chrome, Firefox, Edge")
-
     # Get an access token and extract to a JSON file "access_token.json"
     @classmethod
     def setCustomKey(self):
@@ -391,7 +391,7 @@ def setCustomKey(self):
                                   scope=scopes, redirect_uri=redirect_uri)
         auth_url, state = stackApps.authorization_url(authorization_url)
 
-        driver = self.get_browser_driver()
+        driver = get_browser_driver()
 
         # Open auth_url in one of the supported browsers
         driver.get(auth_url)