]>
jfr.im git - erebus.git/blob - erebus.py
3 # Erebus IRC bot - Author: John Runyon
8 import os
, sys
, select
, MySQLdb
, MySQLdb
.cursors
9 import bot
, config
, ctlmod
22 def __init__(self
, nick
, auth
=None):
28 return self
.auth
is not None
30 def authed(self
, auth
):
31 if auth
== '0': auth
= None
40 c
.execute("SELECT level FROM users WHERE auth = %s", (self
.auth
,))
43 self
.glevel
= row
['level']
48 def __str__(self
): return self
.nick
49 def __repr__(self
): return "<User %r (%d)>" % (self
.nick
,self
.glevel
)
51 class Channel(object):
56 def __init__(self
, name
):
59 def userjoin(self
, user
, level
=None):
60 if user
not in self
.users
: self
.users
.append(user
)
61 if level
== 'op' and user
not in self
.ops
: self
.ops
.append(user
)
62 if level
== 'voice' and user
not in self
.voices
: self
.voices
.append(user
)
63 def userpart(self
, user
):
64 if user
in self
.ops
: self
.ops
.remove(user
)
65 if user
in self
.voices
: self
.voices
.remove(user
)
66 if user
in self
.users
: self
.users
.remove(user
)
68 def userop(self
, user
):
69 if user
in self
.users
and user
not in self
.ops
: self
.ops
.append(user
)
70 def uservoice(self
, user
):
71 if user
in self
.users
and user
not in self
.voices
: self
.voices
.append(user
)
72 def userdeop(self
, user
):
73 if user
in self
.ops
: self
.ops
.remove(user
)
74 def userdevoice(self
, user
):
75 if user
in self
.voices
: self
.voices
.remove(user
)
77 def __str__(self
): return self
.name
78 def __repr__(self
): return "<Channel %r>" % (self
.name
)
80 def __init__(self
, trigger
):
81 self
.trigger
= trigger
82 if os
.name
== "posix":
84 self
.po
= select
.poll()
85 else: # f.e. os.name == "nt" (Windows)
86 self
.potype
= "select"
89 def newbot(self
, nick
, user
, bind
, server
, port
, realname
, chans
):
90 if bind
is None: bind
= ''
91 obj
= bot
.Bot(self
, nick
, user
, bind
, server
, port
, realname
, chans
)
92 self
.bots
[nick
.lower()] = obj
94 def newfd(self
, obj
, fileno
):
95 self
.fds
[fileno
] = obj
96 if self
.potype
== "poll":
97 self
.po
.register(fileno
, select
.POLLIN
)
98 elif self
.potype
== "select":
99 self
.fdlist
.append(fileno
)
101 def bot(self
, name
): #get Bot() by name (nick)
102 return self
.bots
[name
.lower()]
103 def fd(self
, fileno
): #get Bot() by fd/fileno
104 return self
.fds
[fileno
]
106 def user(self
, nick
):
108 if nick
in self
.users
:
109 return self
.users
[nick
]
111 user
= self
.User(nick
)
112 self
.users
[nick
] = user
114 def channel(self
, name
): #TODO #get Channel() by name
115 return self
.Channel(name
.lower())
118 if self
.potype
== "poll":
119 return [fd
for (fd
, ev
) in self
.po
.poll()]
120 elif self
.potype
== "select":
121 return select
.select(self
.fdlist
, [], [])[0]
123 def connectall(self
):
124 for bot
in self
.bots
.itervalues():
125 if bot
.conn
.state
== 0:
129 def hook(self
, word
, handler
):
130 self
.msghandlers
[word
] = handler
131 def unhook(self
, word
):
132 del self
.msghandlers
[word
]
133 def hashook(self
, word
):
134 return word
in self
.msghandlers
135 def gethook(self
, word
):
136 return self
.msghandlers
[word
]
141 cfg
= config
.Config('bot.config')
142 main
= Erebus(cfg
.trigger
)
144 autoloads
= [mod
for mod
, yes
in cfg
.items('autoloads') if int(yes
) == 1]
145 for mod
in autoloads
:
146 ctlmod
.load(main
, mod
)
148 main
.db
= MySQLdb
.connect(host
=cfg
.dbhost
, user
=cfg
.dbuser
, passwd
=cfg
.dbpass
, db
=cfg
.dbname
, cursorclass
=MySQLdb
.cursors
.DictCursor
)
150 c
.execute("SELECT nick, user, bind FROM bots WHERE active = 1")
154 c2
= main
.db
.cursor()
155 c2
.execute("SELECT chname FROM chans WHERE bot = %s AND active = 1", (row
['nick'],))
156 chans
= [chdic
['chname'] for chdic
in c2
.fetchall()]
158 main
.newbot(row
['nick'], row
['user'], row
['bind'], cfg
.host
, cfg
.port
, cfg
.realname
, chans
)
162 poready
= main
.poll()
163 for fileno
in poready
:
164 for line
in main
.fd(fileno
).getdata():
165 main
.fd(fileno
).parse(line
)
167 if __name__
== '__main__':