diff --git a/src/__main__.py b/src/__main__.py index 26d3978..b44e2eb 100644 --- a/src/__main__.py +++ b/src/__main__.py @@ -10,9 +10,9 @@ if __name__ == "__main__": try: parser = argparse.ArgumentParser() # Add long and short arguments - parser.add_argument("--file", "-f", help="The instructions file to use if any.") + parser.add_argument("--file", "-f", help="The instructions file to use.") parser.add_argument("--browser", "-b", default="firefox", help="ie, chrome, firefox (Optional)") - parser.add_argument("--headless", "-hm", default=False, help="Run browser in headless mode.") + parser.add_argument("--headless", "-hm", default=False, help="Run browser in headless mode. (Optional)") parser.add_argument("--persist", "-p", default=False, help="Keep browser open after run. (Optional)") diff --git a/src/core/__init__.py b/src/core/__init__.py index 065d76b..d151b56 100644 --- a/src/core/__init__.py +++ b/src/core/__init__.py @@ -1,3 +1,7 @@ +""" + Core module +""" + from .mixins import ControlerMixin from .utils import Logger diff --git a/src/core/mixins/__init__.py b/src/core/mixins/__init__.py index be71ddc..ebeff20 100644 --- a/src/core/mixins/__init__.py +++ b/src/core/mixins/__init__.py @@ -1 +1,5 @@ +""" + Mixins module +""" + from .ControlerMixin import ControlerMixin diff --git a/src/core/utils/Browser.py b/src/core/utils/Browser.py index f048921..acff046 100644 --- a/src/core/utils/Browser.py +++ b/src/core/utils/Browser.py @@ -35,5 +35,7 @@ class Browser: driver = webdriver.Firefox(options=_options, firefox_profile=profile, log_path=_log_path) + if driver is None: + raise("No driver set! Check your argument or defaults...") return driver diff --git a/src/core/utils/Logger.py b/src/core/utils/Logger.py index 20bf93c..e9a8cc6 100644 --- a/src/core/utils/Logger.py +++ b/src/core/utils/Logger.py @@ -44,6 +44,9 @@ class Logger: folder = "core/logs" file = folder + "/twitter-bot.log" + if not os.path.exists(folder): + os.mkdir(folder) + fh = logging.FileHandler(file) fh.setLevel(level=fhLogLevel) fh.setFormatter(fFormatter) diff --git a/src/core/utils/__init__.py b/src/core/utils/__init__.py index cf7de47..5c72dc4 100644 --- a/src/core/utils/__init__.py +++ b/src/core/utils/__init__.py @@ -1,2 +1,6 @@ +""" + Utils module +""" + from .Logger import Logger from .Browser import Browser