✨ Add Notion.so login feature

Author: NamamiShanker

.gitignore

@@ -31,4 +31,7 @@ cython_debug/
 
 # User files related to API testing and authentication
 access_token.json
-response_data.json
+response_data.json
+
+# Selenium data
+geckodriver.log

main.py

@@ -55,19 +55,24 @@
                     help="View and organise the playbook",
                     action='store_true')
 
+parser.add_argument("-no",
+                    "--notion",
+                    help="Login to your Notion account to save playbook",
+                    action='store_true')
+
 ARGV = parser.parse_args()
 
 search_flag = Search(ARGV)
 
 if __name__ == "__main__":
     if ARGV.start:
         print('''
-        \U0001F604 Hello and Welcome to Dynamic CLI 
-        \U0001F917 Use the following commands to get started 
-        \U0001F50E Search on StackOverflow with '-s' 
-        \U0001F4C4 Open browser to create new Stack Overflow question with '-n [title(optional)]' 
+        \U0001F604 Hello and Welcome to Dynamic CLI
+        \U0001F917 Use the following commands to get started
+        \U0001F50E Search on StackOverflow with '-s'
+        \U0001F4C4 Open browser to create new Stack Overflow question with '-n [title(optional)]'
         \U0001F4C2 Save answer to a file with '-file'
-        \U00002728 Know the version of Dynamic CLI with '-V' 
+        \U00002728 Know the version of Dynamic CLI with '-V'
         \U0001F609 See this message again with '-st'
         \U00002755 Get help with '-h'
         ''')

src/arguments/notion.py

@@ -0,0 +1,79 @@
+import os
+import json
+
+from .utility import Utility
+from .error import SearchError
+
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+
+
+from webdriver_manager.chrome import ChromeDriverManager
+from webdriver_manager.firefox import GeckoDriverManager
+from webdriver_manager.microsoft import EdgeChromiumDriverManager
+
+def get_token_from_cookie(cookie, token):
+    for el in cookie:
+        if el['name'] == token:
+            return el
+    pass
+
+class NotionClient():
+    """
+    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
+    """
+    def __init__(self):
+        self.tokenv2_cookie = None
+        self.base_url = "https://www.notion.so/"
+        self.login_path = "login/"
+        self.linux_path = "/home/{}/Documents/dynamic".format(os.getenv('USER'))
+        self.mac_path = "/Users/{}/Documents/dynamic".format(os.getenv('USER'))
+        self.file_name = 'tokenv2_cookie.key'
+
+    def get_token_from_file(self):
+        raise FileNotFoundError("File not found")
+        return None
+
+    def get_login_path(self):
+        return self.base_url + self.login_path
+
+    def save_token_file(self):
+        pass
+
+    def get_cookies_from_login(self):
+        """
+        Provides the user browser window to login to Notion
+        Returns the user's cookies which can be used to
+        access and transfer content to user's Notion account
+        """
+        driver = Utility().get_browser_driver()
+        try:
+            driver.get(self.get_login_path())
+            WebDriverWait(driver, 300).until(
+                EC.presence_of_element_located((By.CLASS_NAME,
+                                                "notion-sidebar-container")))
+            return driver.get_cookies()
+        except Exception as e:
+            print(e)
+        finally:
+            driver.quit()
+
+    def set_tokenv2_cookie(self):
+        # Sets 'tokenv2_cookie equal to the particular cookie containing token_v2
+        if not self.tokenv2_cookie:
+            try:
+                self.tokenv2_cookie = self.get_token_from_file()
+            except:
+                try:
+                    cookies = self.get_cookies_from_login()
+                    self.tokenv2_cookie = get_token_from_cookie(cookies, 'token_v2')
+                except Exception as e:
+                    print(e)
+
+    def print_token_v2(self):
+        self.set_tokenv2_cookie()
+        print(self.tokenv2_cookie)

src/arguments/search.py

@@ -10,6 +10,7 @@
 from .save import SaveSearchResults
 from .update import UpdateApplication
 from .api_test import ApiTesting
+from .notion import NotionClient
 
 version = "0.1.0"
 class Prompt():
@@ -53,6 +54,8 @@ def search_args(self):
             update.check_for_updates()
         elif self.arguments.GET:
             self.api_test_object.get_request()
+        elif self.arguments.notion:
+            NotionClient().print_token_v2()
 
     def search_for_results(self, save=False):
         queries = ["What do you want to search", "Tags"]
@@ -78,12 +81,12 @@ def search_for_results(self, save=False):
         else:
             data = self.utility_object.get_ans(questions)
             print('''
-            \U0001F604 Hopefully you found what you were looking for! 
+            \U0001F604 Hopefully you found what you were looking for!
             \U0001F4C2 You can save an answer to a file with '-file'
 
             Not found what you were looking for \U00002754
             \U0001F4C4 Open browser and post your question on StackOverflow with '-n [title (optional)]'
-            
+
             \U0001F50E To search more use '-s'
             ''')
 

src/arguments/utility.py

@@ -310,6 +310,18 @@ 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:
+            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:
+                    raise Exception("Browser/Browser driver now found.")
+
     # Get an access token and extract to a JSON file "access_token.json"
     @classmethod
     def setCustomKey(self):
@@ -332,15 +344,9 @@ def setCustomKey(self):
         # Try to install web drivers for one of these browsers
         # Chrome, Firefox, Edge (One of them must be installed)
         try:
-            driver = webdriver.Chrome(ChromeDriverManager().install())
-        except ValueError:
-            try:
-                driver = webdriver.Firefox(executable_path=GeckoDriverManager().install())
-            except ValueError:
-                try:
-                    driver = webdriver.Edge(EdgeChromiumDriverManager().install())
-                except ValueError:
-                    print("You do not have one of these supported browsers: Chrome, Firefox, Edge")
+            driver = Utility().get_browser_driver()
+        except Exception as e:
+            print(e.message)
 
         # Open auth_url in one of the supported browsers
         driver.get(auth_url)