diff --git a/jarvis_tasks/config.py b/jarvis_tasks/config.py index 64bf9a4..2944561 100644 --- a/jarvis_tasks/config.py +++ b/jarvis_tasks/config.py @@ -3,5 +3,5 @@ from jarvis_core.config import Config class TaskConfig(Config): - REQUIRED = ["token", "mongo", "twitter", "reddit"] - OPTIONAL = {"log_level": "WARNING"} + REQUIRED = ["token", "mongo"] + OPTIONAL = {"log_level": "WARNING", "twitter": None, "reddit": None} diff --git a/jarvis_tasks/tasks/reddit.py b/jarvis_tasks/tasks/reddit.py index 3819c88..129de86 100644 --- a/jarvis_tasks/tasks/reddit.py +++ b/jarvis_tasks/tasks/reddit.py @@ -155,6 +155,9 @@ async def stream(bot: Snake) -> None: bot: Snake instance logger: Global logger """ + if not config.reddit: + logger.warn("Missing Reddit config, not starting") + return logger.debug("Starting Task-stream") reddit = Reddit(**config.reddit) diff --git a/jarvis_tasks/tasks/twitter.py b/jarvis_tasks/tasks/twitter.py index ab26e6d..ff3c295 100644 --- a/jarvis_tasks/tasks/twitter.py +++ b/jarvis_tasks/tasks/twitter.py @@ -82,6 +82,9 @@ async def twitter(bot: Snake) -> None: bot: Snake instance logger: Global logger """ + if not config.twitter: + logger.warn("Missing Twitter config, not starting") + return auth = tweepy.AppAuthHandler(config.twitter["consumer_key"], config.twitter["consumer_secret"]) api = tweepy.API(auth) while True: diff --git a/pyproject.toml b/pyproject.toml index 11d8892..b73a863 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "jarvis-tasks" -version = "0.4.0" +version = "0.4.1" description = "" authors = ["Your Name "]