Added modules

This commit is contained in:
Matthias Schiffer 2010-05-02 20:28:17 +02:00
parent 41fec54c30
commit 0277a779c9
8 changed files with 77 additions and 21 deletions

1
.gitignore vendored
View file

@ -1,2 +1,3 @@
*~
*.pyo
config.py

View file

@ -1,5 +0,0 @@
config = {
'jid': '',
'password': '',
'cacertfile': ''
}

10
config.py.default Normal file
View file

@ -0,0 +1,10 @@
config = {
'jid': '',
'password': '',
'cacertfile': '',
'rooms': [],
'nick': '',
'modules': []
}

View file

@ -4,6 +4,7 @@ from pyxmpp.client import Client
from pyxmpp.jabber.muc import *
from pyxmpp.interface import implements
from pyxmpp.interfaces import *
from functools import partial
class MessageHandler:
implements(IMessageHandlersProvider)
@ -17,26 +18,24 @@ class MessageHandler:
]
def message(self, stanza):
return Message(
to_jid=stanza.get_from(),
from_jid=stanza.get_to(),
stanza_type=stanza.get_type(),
subject=stanza.get_subject(),
body=stanza.get_body())
pass
class MucHandler(MucRoomHandler):
def __init__(self, client):
MucRoomHandler.__init__(self)
self.client = client
def message_received(self, user, stanza):
if(user.same_as(self.room_state.me)):
return
self.room_state.send_message(stanza.get_body()[::-1])
for mod in self.client.module_manager.modules:
mod.groupchat(stanza.get_from().bare().as_unicode(), stanza.get_from().resource, stanza.get_body(), self.room_state.send_message)
def user_joined(self, user, stanza):
if(not self.room_state.joined or user.same_as(self.room_state.me)):
return
self.room_state.send_message('Willkommen, ' + user.nick)
class PresenceHandler(object):
implements(IPresenceHandlersProvider)
@ -112,8 +111,11 @@ class VersionHandler(object):
class XMPPConnection(Client):
def __init__(self, config):
Client.__init__(self, JID(config['jid']), config['password'], tls_settings = TLSSettings(require = True, cacert_file = config['cacertfile']))
def __init__(self, config, modman):
self.config = config
self.module_manager = modman
Client.__init__(self, JID(config['jid']), config['password'], tls_settings = TLSSettings(require = True, cacert_file = config['cacertfile']), keepalive = 60)
self.interface_providers = [
MessageHandler(self),
@ -125,4 +127,6 @@ class XMPPConnection(Client):
Client.authorized(self)
self.room_manager = MucRoomManager(self.stream)
self.room_manager.set_handlers()
self.room_manager.join(JID('c3hl@conference.jabber.ccc.de'), 'curunir', MucHandler(), history_maxstanzas=0)
for room in self.config['rooms']:
self.room_manager.join(JID(room), self.config['nick'], MucHandler(self), history_maxstanzas=0)

View file

@ -3,6 +3,18 @@ from connection.xmpp import XMPPConnection
import signal, sys
from config import config
class ModuleManager:
def __init__(self, config):
self.config = config
self.modules = []
for mod in config['modules']:
self.load(mod)
def load(self, name):
mod = __import__('modules.' + name, globals(), locals(), ['Module'])
self.modules.append(mod.Module(self))
run = True
def exithandler(signum, frame):
@ -13,10 +25,14 @@ signal.signal(signal.SIGINT, exithandler)
signal.signal(signal.SIGTERM, exithandler)
signal.signal(signal.SIGQUIT, exithandler)
connection = XMPPConnection(config)
modman = ModuleManager(config)
connection = XMPPConnection(config, modman)
connection.connect()
while run:
connection.stream.loop_iter(1)
act = connection.stream.loop_iter(1)
if not act:
connection.idle()
connection.disconnect()

0
modules/__init__.py Normal file
View file

12
modules/base.py Normal file
View file

@ -0,0 +1,12 @@
class ModuleBase:
def __init__(self, manager):
self.manager = manager
def commands(self):
return []
def message(self, message_type, message_from, message_subject, message_body, reply):
pass
def groupchat(self, room, message_from, message_body, reply):
pass

18
modules/help.py Normal file
View file

@ -0,0 +1,18 @@
from base import ModuleBase
import re
class Module(ModuleBase):
def __init__(self, manager):
ModuleBase.__init__(self, manager)
def commands(self):
return [('!help', 'Zeigt diese Hilfe an...')]
def groupchat(self, room, message_from, message_body, reply):
if not re.match(r'!help(?:\W|\Z)', message_body):
return
commands = reduce(lambda l, mod: l + mod.commands(), self.manager.modules, [])
helpstring = reduce(lambda s, (c, h): s + c + ': ' + h + '\n', commands, '')
reply('Befehle:\n\n' + helpstring)