cli and multi language support

app/browser.py

@@ -1,7 +1,5 @@
-import pickle
 import selenium
 import selenium.webdriver
-import time
 import logging
 
 settings = {}
@@ -10,7 +8,16 @@ def init(settings_obj):
     global settings
     settings = settings_obj
 
-def create_webdriver():
+def create_webdriver(driver='chrome'):
+    if driver == 'chrome':
+        return create_webdriver_chrome()
+    elif driver == 'firefox':
+        return create_webdriver_firefox()
+
+def create_webdriver_firefox():
+    pass
+
+def create_webdriver_chrome():
     opt = selenium.webdriver.chrome.options.Options()
     opt.add_argument('--user-agent=' + settings.WEBDRIVER_USER_AGENT)
     opt.add_argument('--kiosk-printing')

app/cli.py

@@ -31,6 +31,7 @@ def main():
 class InteractiveInterface:
     def __init__(self):
         self.model = model.Model()
+        x = self.model.webdriver # Request the browser open immediately
 
     def run(self, args, main=True):
             try:

app/model.py

@@ -25,7 +25,9 @@ class Model:
     @property
     def webdriver(self):
         if not self._webdriver:
-            self._webdriver = browser.create_webdriver()
+            self._webdriver = browser.create_webdriver(settings.WEBDRIVER_BROWSER)
+            page_api = pages.LandingPage(self.webdriver)
+            page_api.goto_landing_page()
         return self._webdriver
 
     def do_random_page(self):
@@ -39,7 +41,6 @@ class Model:
         page_api.goto_random_article()
 
         # Article page
-        
         pages_visited = []
         while True:
             page_api = pages.ArticlePage(self.webdriver)

app/pages.py

@@ -2,6 +2,7 @@ import logging
 import re
 import selenium
 import selenium.webdriver
+import time
 
 settings = {}
 
@@ -22,6 +23,7 @@ class PageRootObject:
 
     def click(self, el):
         self.highlight(el, 'red')
+        time.sleep(settings.PAGE_DELAY)
         el.click()
 
     def highlight(self, el, color):
@@ -47,7 +49,7 @@ class MainPage(PageRootObject):
         super().__init__(driver)
 
     def goto_random_article(self):
-        link = self.driver.find_element_by_partial_link_text('Random article')
+        link = self.driver.find_element_by_xpath('//li[contains(@id, "n-randompage")]/a')
         self.click(link)
 
 class ArticlePage(PageRootObject):