Merge pull request #65 from Python-World/STOOL2

added default chrome driver if browser and driver not provided

Author: chavarera

Diff for: README.rst

@@ -44,19 +44,19 @@ Usage
 
 .. code-block:: python
 
-      """Example code with context manager"""
+    """Example code with context manager"""
 
-      from s_tool.core import SeleniumDriver as SBot
+    from s_tool.core import SeleniumTools as SBot
 
-      with SBot("firefox", headless=True) as self:
-          self.get("https://google.com")
-          sessionid = self.session()
-          url = self.url()
-          cookies = self.cookies()
-
-          # print sessionid,url,cookies
-          print(f"\nurl     :   {url} \nsession :   {sessionid}\ncookies :   {cookies}\n")
+    with SBot(browser="firefox", headless=True) as self:
+        self.get("https://example.com")
+        sessionid = self.sessionid()
+        url = self.url()
+        cookies = self.cookies()
 
+        # print sessionid,url,cookies
+        print(f"\nurl     :   {url} \nsession :   {sessionid}\ncookies :   {cookies}\n")
+     
 
 * Example Using class
 
@@ -84,8 +84,7 @@ Usage
 
     bot = SBot(browser ="firefox", headless=True)  # change headless=False to run with gui mode
     bot.run()
-    bot.close()
-
+    bot._close()
 
 Methods
 ^^^^^^^

Diff for: s_tool/core.py

@@ -74,6 +74,9 @@ def __enter__(self):
 
     def _load_driver(self):
         """Create Selenium webdriver object"""
+        if self.browser is None:
+            self.browser='chrome'
+
         obj = SeleniumDriver(browser=self.browser,
                              headless=self.headless,
                              executable_path=self.executable_path)
@@ -204,6 +207,9 @@ def _validate_driver(self) -> None:
         err = "Selenium WebDriver validation failed."
         supported_browsers = self._get_supported_browsers()
 
+        if self.driver is None:
+            return False
+
         if hasattr(
                 self.driver,
                 'name') and self.driver.name not in supported_browsers: