FumeNodes/SCNewsBot
1
0
Fork 0
mirror of https://github.com/hencaric/SCNewsBot.git synced 2026-08-17 06:56:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

add warning for legacy config

Browse Source
This commit is contained in:
mudkipdev
2023-05-07 21:25:44 -06:00
parent 24aa35dc03
commit 122fc79d01
3 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scnewsbot/__main__.py
View File

@@ -9,12 +9,21 @@ class InvalidTokenException(Exception):
pass pass
class InvalidConfigException(Exception):
pass
def main() -> None: def main() -> None:
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN") DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
if not DISCORD_TOKEN: if not DISCORD_TOKEN:
raise InvalidTokenException("A Discord token was not set.") raise InvalidTokenException("A Discord token was not set.")
with open("config.toml", "rb") as config_file: with open("config.toml", "rb") as config_file:
if "[permissions]" in config_file.read():
raise InvalidConfigException(
"The format for permissions has changed; please check the new example config."
)
config = Config(tomllib.load(config_file)) config = Config(tomllib.load(config_file))
bot = Bot(config) bot = Bot(config)

8
scnewsbot/utils.py
View File

@@ -34,9 +34,13 @@ class Config:
return self._get_allowed_objects("allowed_users") return self._get_allowed_objects("allowed_users")
def _get_allowed_objects(self, object_name, /) -> list: def _get_allowed_objects(self, object_name, /) -> list:
allowed_objects = self.config["permissions"]["announcements"].get(object_name, []) allowed_objects = self.config["permissions"]["announcements"].get(
object_name, []
)
if self.debug: if self.debug:
allowed_objects += self.config["permissions"]["announcements"]["debug"].get(object_name, []) allowed_objects += self.config["permissions"]["announcements"]["debug"].get(
object_name, []
)
return allowed_objects return allowed_objects