]>
jfr.im git - erebus.git/blob - erebus.py
5 import os
, sys
, select
, MySQLdb
, MySQLdb
.cursors
17 def __init__(self
, nick
, auth
=None):
24 def authed(self
, auth
):
28 def checklevel(self
): self
.level
= 9999 #TODO get level from db
30 def __str__(self
): return self
.nick
31 def __repr__(self
): return "<User %r>" % (self
.nick
)
33 class Channel(object):
38 def __init__(self
, name
):
41 def userjoin(self
, user
, level
=None):
42 if user
not in self
.users
: self
.users
.append(user
)
43 if level
== 'op' and user
not in self
.ops
: self
.ops
.append(user
)
44 if level
== 'voice' and user
not in self
.voices
: self
.voices
.append(user
)
45 def userpart(self
, user
):
46 if user
in self
.ops
: self
.ops
.remove(user
)
47 if user
in self
.voices
: self
.voices
.remove(user
)
48 if user
in self
.users
: self
.users
.remove(user
)
50 def userop(self
, user
):
51 if user
in self
.users
and user
not in self
.ops
: self
.ops
.append(user
)
52 def uservoice(self
, user
):
53 if user
in self
.users
and user
not in self
.voices
: self
.voices
.append(user
)
54 def userdeop(self
, user
):
55 if user
in self
.ops
: self
.ops
.remove(user
)
56 def userdevoice(self
, user
):
57 if user
in self
.voices
: self
.voices
.remove(user
)
59 def __str__(self
): return self
.name
60 def __repr__(self
): return "<Channel %r>" % (self
.name
)
63 if os
.name
== "posix":
65 self
.po
= select
.poll()
66 else: # f.e. os.name == "nt" (Windows)
67 self
.potype
= "select"
70 def newbot(self
, nick
, user
, bind
, server
, port
, realname
, chans
):
71 if bind
is None: bind
= ''
72 obj
= bot
.Bot(self
, nick
, user
, bind
, server
, port
, realname
, chans
)
73 self
.bots
[nick
.lower()] = obj
75 def newfd(self
, obj
, fileno
):
76 self
.fds
[fileno
] = obj
77 if self
.potype
== "poll":
78 self
.po
.register(fileno
, select
.POLLIN
)
79 elif self
.potype
== "select":
80 self
.fdlist
.append(fileno
)
82 def bot(self
, name
): #get Bot() by name (nick)
83 return self
.bots
[name
.lower()]
84 def fd(self
, fileno
): #get Bot() by fd/fileno
85 return self
.fds
[fileno
]
87 def user(self
, nick
): #TODO #get User() by nick
88 return self
.User(nick
.lower())
89 def channel(self
, name
): #TODO #get Channel() by name
90 return self
.Channel(name
.lower())
93 if self
.potype
== "poll":
94 return [fd
for (fd
, ev
) in self
.po
.poll()]
95 elif self
.potype
== "select":
96 return select
.select(self
.fdlist
, [], [])[0]
99 for bot
in self
.bots
.itervalues():
100 if bot
.conn
.state
== 0:
104 def modlist(self
): pass
105 def hasmod(self
, name
): pass
106 def loadmod(self
, name
): pass
107 def unloadmod(self
, name
): pass
108 def reloadmod(self
, name
): pass
111 def bind(self
, word
, handler
): pass
112 def addbind(self
, word
, handler
): pass
113 def rmbind(self
, word
, handler
): pass
114 def getbind(self
, word
, handler
): pass
116 cfg
= config
.Config('bot.config')
120 main
.db
= MySQLdb
.connect(host
=cfg
.dbhost
, user
=cfg
.dbuser
, passwd
=cfg
.dbpass
, db
=cfg
.dbname
, cursorclass
=MySQLdb
.cursors
.DictCursor
)
122 c
.execute("SELECT nick, user, bind FROM bots WHERE active = 1")
126 c2
= main
.db
.cursor()
127 c2
.execute("SELECT chname FROM chans WHERE bot = %s AND active = 1", (row
['nick'],))
128 chans
= [chdic
['chname'] for chdic
in c2
.fetchall()]
130 main
.newbot(row
['nick'], row
['user'], row
['bind'], cfg
.host
, cfg
.port
, cfg
.realname
, chans
)
134 poready
= main
.poll()
135 for fileno
in poready
:
136 main
.fd(fileno
).getdata()
138 if __name__
== '__main__':