From d973b547e1ced436ecb5b830301904c12d28c823 Mon Sep 17 00:00:00 2001 From: jesopo Date: Thu, 5 Sep 2019 14:13:58 +0100 Subject: [PATCH] add a `create` flag to IRCChannels.get_id() to optionally not create new ids --- src/IRCChannels.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/IRCChannels.py b/src/IRCChannels.py index 0f36da5d..ebcc8077 100644 --- a/src/IRCChannels.py +++ b/src/IRCChannels.py @@ -25,8 +25,9 @@ class Channels(object): def items(self): return self._channels.items() - def get_id(self, channel_name: str) -> int: - self._bot.database.channels.add(self._server.id, channel_name) + def get_id(self, channel_name: str, create: bool=True) -> int: + if create: + self._bot.database.channels.add(self._server.id, channel_name) return self._bot.database.channels.get_id(self._server.id, channel_name) def _name_lower(self, channel_name: str) -> str: