import typing from src import EventManager, IRCBot, IRCChannel, IRCServer, utils class Channels(object): def __init__(self, server: "IRCServer.Server", bot: "IRCBot.Bot", events: EventManager.Events): self._server = server self._bot = bot self._events = events self._channels = {} # type: typing.Dict[str, IRCChannel.Channel] def __iter__(self) -> typing.Iterable[IRCChannel.Channel]: return (channel for channel in self._channels.values()) def __contains__(self, name: str) -> bool: return self.contains(name) def __len__(self) -> int: return len(self._channels) def __getitem__(self, name: str): return self.get(name) def keys(self): return self._channels.keys() def values(self): return self._channels.values() 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) return self._bot.database.channels.get_id(self._server.id, channel_name) def _name_lower(self, channel_name: str) -> str: return self._server.irc_lower(channel_name) def contains(self, name: str) -> bool: lower = self._name_lower(name) return name[0] in self._server.channel_types and lower in self._channels def add(self, name: str) -> IRCChannel.Channel: id = self.get_id(name) lower = self._name_lower(name) new_channel = IRCChannel.Channel(lower, id, self._server, self._bot) self._channels[lower] = new_channel self._events.on("new.channel").call(channel=new_channel, server=self) return new_channel def remove(self, channel: IRCChannel.Channel): lower = self._name_lower(channel.name) del self._channels[lower] def get(self, name: str): return self._channels[self._name_lower(name)] def rename(self, old_name, new_name): old_lower = self._name_lower(old_name) new_lower = self._name_lower(new_name) channel = self.channels.pop(old_lower) channel.name = new_name self._channels[new_name] = channel self._bot.database.channels.rename(channel.id, new_lower)