jarvis/jarvis-tasks
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make Reddit and Twitter optional

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

4
jarvis_tasks/config.py
View file

@ -3,5 +3,5 @@ from jarvis_core.config import Config
class TaskConfig(Config): class TaskConfig(Config):
REQUIRED = ["token", "mongo", "twitter", "reddit"] REQUIRED = ["token", "mongo"]
OPTIONAL = {"log_level": "WARNING"} OPTIONAL = {"log_level": "WARNING", "twitter": None, "reddit": None}

3
jarvis_tasks/tasks/reddit.py
View file

@ -155,6 +155,9 @@ async def stream(bot: Snake) -> None:
bot: Snake instance bot: Snake instance
logger: Global logger logger: Global logger
""" """
if not config.reddit:
logger.warn("Missing Reddit config, not starting")
return
logger.debug("Starting Task-stream") logger.debug("Starting Task-stream")
reddit = Reddit(**config.reddit) reddit = Reddit(**config.reddit)

3
jarvis_tasks/tasks/twitter.py
View file

@ -82,6 +82,9 @@ async def twitter(bot: Snake) -> None:
bot: Snake instance bot: Snake instance
logger: Global logger 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"]) auth = tweepy.AppAuthHandler(config.twitter["consumer_key"], config.twitter["consumer_secret"])
api = tweepy.API(auth) api = tweepy.API(auth)
while True: while True:

2
pyproject.toml
View file

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