forked from jesopo/bitbot-old
/
IRCChannel.py
66 lines (60 loc) · 2.29 KB
/
IRCChannel.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import IRCChannelMode
class IRCChannel(object):
def __init__(self, name, server, config):
self.name = name
self.name_lower = name.lower()
self.server = server
self.config = config
self.users = {}
self.modes = {}
self.user_modes = {}
self.log = []
self._destroyed = False
def add_user(self, user):
if not user.id in self.users:
self.users[user.id] = user
if not user.id in self.user_modes:
self.user_modes[user.id] = set([])
def remove_user(self, user):
if user.id in self.users:
del self.users[user.id]
if user.id in self.user_modes:
del self.user_modes[user.id]
def add_mode(self, mode, argument):
if not mode in self.modes:
self.modes[mode] = IRCChannelMode.IRCChannelMode(self, mode)
if argument:
user = self.server.get_user_by_nickname(argument)
if user and user.id in self.users:
self.modes[mode].add_argument(user.id)
self.user_modes[user.id].add(mode)
else:
self.modes[mode].add_argument(argument)
else:
self.modes[mode].enabled = True
def remove_mode(self, mode, argument):
if mode in self.modes:
if argument:
user = self.server.get_user_by_nickname(argument)
if user and user.id in self.users:
self.modes[mode].remove_argument(user.id)
self.user_modes[user.id]
else:
self.modes[mode].remove_argument(argument)
else:
self.modes[mode].enabled = False
def send_who(self):
self.server.send_who(self.name)
def send_message(self, message):
self.server.send_message(self.name, message)
def send_kick(self, user, reason=None):
self.server.send_kick(self.name, user.nickname, reason)
def destroy(self):
for id in list(self.users.keys()):
user = self.users[id]
user.remove_channel(self)
del self.users[id]
self._destroyed = True
self.server.bot.events.on("destroyed").on("channel").call(channel=self)
def is_destroyed(self):
return self._destroyed