Make Reddit and Twitter optional

This commit is contained in:
Zeva Rose 2022-04-20 00:25:06 -06:00
parent 01111811da
commit 3a6190247a
4 changed files with 9 additions and 3 deletions

View file

@ -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}

View file

@ -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)

View file

@ -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:

View file

@ -1,6 +1,6 @@
[tool.poetry]
name = "jarvis-tasks"
version = "0.4.0"
version = "0.4.1"
description = ""
authors = ["Your Name <you@example.com>"]