Fix missing config for rook

This commit is contained in:
Zeva Rose 2022-04-29 17:02:26 -06:00
parent b8aa191750
commit fe8bd90854

View file

@ -22,6 +22,7 @@ class JarvisConfig(CConfig):
"max_messages": 1000, "max_messages": 1000,
"twitter": None, "twitter": None,
"reddit": None, "reddit": None,
"rook_token": None,
} }
@ -50,6 +51,7 @@ class Config(object):
max_messages: int = 1000, max_messages: int = 1000,
twitter: dict = None, twitter: dict = None,
reddit: dict = None, reddit: dict = None,
rook_token: str = None,
) -> None: ) -> None:
"""Initialize the config object.""" """Initialize the config object."""
self.token = token self.token = token
@ -63,6 +65,7 @@ class Config(object):
self.twitter = twitter self.twitter = twitter
self.reddit = reddit self.reddit = reddit
self.sync = sync or os.environ.get("SYNC_COMMANDS", False) self.sync = sync or os.environ.get("SYNC_COMMANDS", False)
self.rook_token = rook_token
self.__db_loaded = False self.__db_loaded = False
self.__mongo = MongoClient(**self.mongo["connect"]) self.__mongo = MongoClient(**self.mongo["connect"])