mirror of
http://aero2k.de/t/repos/urlbot-native.git
synced 2017-09-06 15:25:38 +02:00
all death to the tab character
This commit is contained in:
170
idlebot.py
170
idlebot.py
@@ -6,18 +6,18 @@ import sys
|
||||
from common import VERSION, EVENTLOOP_DELAY, conf_load
|
||||
|
||||
try:
|
||||
from local_config import conf, set_conf
|
||||
from local_config import conf, set_conf
|
||||
except ImportError:
|
||||
sys.stderr.write('''
|
||||
sys.stderr.write('''
|
||||
%s: E: local_config.py isn't tracked because of included secrets and
|
||||
%s site specific configurations. Rename local_config.py.skel and
|
||||
%s adjust to you needs.
|
||||
'''[1:] % (
|
||||
sys.argv[0],
|
||||
' ' * len(sys.argv[0]),
|
||||
' ' * len(sys.argv[0])
|
||||
))
|
||||
sys.exit(1)
|
||||
sys.argv[0],
|
||||
' ' * len(sys.argv[0]),
|
||||
' ' * len(sys.argv[0])
|
||||
))
|
||||
sys.exit(1)
|
||||
|
||||
from sleekxmpp import ClientXMPP
|
||||
|
||||
@@ -25,98 +25,98 @@ got_hangup = False
|
||||
|
||||
|
||||
class IdleBot(ClientXMPP):
|
||||
def __init__(self, jid, password, rooms, nick):
|
||||
ClientXMPP.__init__(self, jid, password)
|
||||
def __init__(self, jid, password, rooms, nick):
|
||||
ClientXMPP.__init__(self, jid, password)
|
||||
|
||||
self.rooms = rooms
|
||||
self.nick = nick
|
||||
self.rooms = rooms
|
||||
self.nick = nick
|
||||
|
||||
self.add_event_handler('session_start', self.session_start)
|
||||
self.add_event_handler('groupchat_message', self.muc_message)
|
||||
self.priority = 0
|
||||
self.status = None
|
||||
self.show = None
|
||||
self.add_event_handler('session_start', self.session_start)
|
||||
self.add_event_handler('groupchat_message', self.muc_message)
|
||||
self.priority = 0
|
||||
self.status = None
|
||||
self.show = None
|
||||
|
||||
self.logger = logging.getLogger(__name__)
|
||||
self.logger = logging.getLogger(__name__)
|
||||
|
||||
def session_start(self, _):
|
||||
self.get_roster()
|
||||
self.send_presence(ppriority=self.priority, pstatus=self.status, pshow=self.show)
|
||||
def session_start(self, _):
|
||||
self.get_roster()
|
||||
self.send_presence(ppriority=self.priority, pstatus=self.status, pshow=self.show)
|
||||
|
||||
for room in self.rooms:
|
||||
self.logger.info('%s: joining' % room)
|
||||
ret = self.plugin['xep_0045'].joinMUC(
|
||||
room,
|
||||
self.nick,
|
||||
wait=True
|
||||
)
|
||||
self.logger.info('%s: joined with code %s' % (room, ret))
|
||||
for room in self.rooms:
|
||||
self.logger.info('%s: joining' % room)
|
||||
ret = self.plugin['xep_0045'].joinMUC(
|
||||
room,
|
||||
self.nick,
|
||||
wait=True
|
||||
)
|
||||
self.logger.info('%s: joined with code %s' % (room, ret))
|
||||
|
||||
def muc_message(self, msg_obj):
|
||||
"""
|
||||
Handle muc messages, return if irrelevant content or die by hangup.
|
||||
:param msg_obj:
|
||||
:return:
|
||||
"""
|
||||
# don't talk to yourself
|
||||
if msg_obj['mucnick'] == self.nick or 'groupchat' != msg_obj['type']:
|
||||
return False
|
||||
elif msg_obj['body'].startswith(conf('bot_user')) and 'hangup' in msg_obj['body']:
|
||||
self.logger.warn("got 'hangup' from '%s': '%s'" % (
|
||||
msg_obj['mucnick'], msg_obj['body']
|
||||
))
|
||||
global got_hangup
|
||||
got_hangup = True
|
||||
return False
|
||||
elif msg_obj['mucnick'] in conf_load().get("other_bots", ()):
|
||||
# not talking to the other bot.
|
||||
return False
|
||||
else:
|
||||
return True
|
||||
def muc_message(self, msg_obj):
|
||||
"""
|
||||
Handle muc messages, return if irrelevant content or die by hangup.
|
||||
:param msg_obj:
|
||||
:return:
|
||||
"""
|
||||
# don't talk to yourself
|
||||
if msg_obj['mucnick'] == self.nick or 'groupchat' != msg_obj['type']:
|
||||
return False
|
||||
elif msg_obj['body'].startswith(conf('bot_user')) and 'hangup' in msg_obj['body']:
|
||||
self.logger.warn("got 'hangup' from '%s': '%s'" % (
|
||||
msg_obj['mucnick'], msg_obj['body']
|
||||
))
|
||||
global got_hangup
|
||||
got_hangup = True
|
||||
return False
|
||||
elif msg_obj['mucnick'] in conf_load().get("other_bots", ()):
|
||||
# not talking to the other bot.
|
||||
return False
|
||||
else:
|
||||
return True
|
||||
|
||||
|
||||
def start(botclass, active=False):
|
||||
logging.basicConfig(
|
||||
level=conf('loglevel', logging.INFO),
|
||||
format=sys.argv[0] + ' %(asctime)s %(levelname).1s %(funcName)-15s %(message)s'
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
logger.info(VERSION)
|
||||
logging.basicConfig(
|
||||
level=conf('loglevel', logging.INFO),
|
||||
format=sys.argv[0] + ' %(asctime)s %(levelname).1s %(funcName)-15s %(message)s'
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
logger.info(VERSION)
|
||||
|
||||
jid = conf('jid')
|
||||
if '/' not in jid:
|
||||
jid = '%s/%s' % (jid, botclass.__name__)
|
||||
bot = botclass(
|
||||
jid=jid,
|
||||
password=conf('password'),
|
||||
rooms=conf('rooms'),
|
||||
nick=conf('bot_user')
|
||||
)
|
||||
import plugins
|
||||
jid = conf('jid')
|
||||
if '/' not in jid:
|
||||
jid = '%s/%s' % (jid, botclass.__name__)
|
||||
bot = botclass(
|
||||
jid=jid,
|
||||
password=conf('password'),
|
||||
rooms=conf('rooms'),
|
||||
nick=conf('bot_user')
|
||||
)
|
||||
import plugins
|
||||
|
||||
if active:
|
||||
plugins.register_all()
|
||||
if plugins.plugin_enabled_get(plugins.command_dsa_watcher):
|
||||
# first result is lost.
|
||||
plugins.command_dsa_watcher(['dsa-watcher', 'crawl'])
|
||||
if active:
|
||||
plugins.register_all()
|
||||
if plugins.plugin_enabled_get(plugins.command_dsa_watcher):
|
||||
# first result is lost.
|
||||
plugins.command_dsa_watcher(['dsa-watcher', 'crawl'])
|
||||
|
||||
bot.connect()
|
||||
bot.register_plugin('xep_0045')
|
||||
bot.process()
|
||||
global got_hangup
|
||||
bot.connect()
|
||||
bot.register_plugin('xep_0045')
|
||||
bot.process()
|
||||
global got_hangup
|
||||
|
||||
while 1:
|
||||
try:
|
||||
# print("hangup: %s" % got_hangup)
|
||||
if got_hangup or not plugins.event_trigger():
|
||||
bot.disconnect()
|
||||
sys.exit(1)
|
||||
while 1:
|
||||
try:
|
||||
# print("hangup: %s" % got_hangup)
|
||||
if got_hangup or not plugins.event_trigger():
|
||||
bot.disconnect()
|
||||
sys.exit(1)
|
||||
|
||||
time.sleep(EVENTLOOP_DELAY)
|
||||
except KeyboardInterrupt:
|
||||
print('')
|
||||
exit(130)
|
||||
time.sleep(EVENTLOOP_DELAY)
|
||||
except KeyboardInterrupt:
|
||||
print('')
|
||||
exit(130)
|
||||
|
||||
|
||||
if '__main__' == __name__:
|
||||
start(IdleBot)
|
||||
start(IdleBot)
|
||||
|
||||
Reference in New Issue
Block a user