send only parameters as "argv"

This commit is contained in:
Thorsten
2016-01-01 20:33:22 +01:00
parent f46acc007d
commit 5fe0db35c0
2 changed files with 132 additions and 140 deletions

View File

@@ -19,17 +19,67 @@ log = logging.getLogger(__name__)
@pluginfunction('version', 'prints version', ptypes_COMMAND) @pluginfunction('version', 'prints version', ptypes_COMMAND)
def command_version(argv, **args): def command_version(argv, **args):
log.info('sent version string') log.info('sent version string')
return { return {
'msg': args['reply_user'] + (''': I'm running ''' + VERSION) 'msg': args['reply_user'] + (''': I'm running ''' + VERSION)
} }
@pluginfunction('uptime', 'prints uptime', ptypes_COMMAND)
def command_uptime(argv, **args):
u = int(config.runtimeconf_get('start_time') + time.time())
plural_uptime = 's'
plural_request = 's'
if 1 == u:
plural_uptime = ''
if 1 == int(config.runtimeconf_get('request_counter')):
plural_request = ''
log.info('sent statistics')
return {
'msg': args['reply_user'] + (''': happily serving for %d second%s, %d request%s so far.''' % (
u, plural_uptime, int(config.runtimeconf_get('request_counter')), plural_request))
}
@pluginfunction('info', 'prints info message', ptypes_COMMAND)
def command_info(argv, **args):
log.info('sent long info')
return {
'msg': args['reply_user'] + (
''': I'm a bot, my job is to extract <title> tags from posted URLs. In case I'm annoying or for further
questions, please talk to my master %s. I'm rate limited.
To make me exit immediately, highlight me with 'hangup' in the message
(emergency only, please). For other commands, highlight me with 'help'.''' % (
config.conf_get('bot_owner')))
}
@pluginfunction('ping', 'sends pong', ptypes_COMMAND, ratelimit_class=RATE_INTERACTIVE)
def command_ping(argv, **args):
rnd = random.randint(0, 3) # 1:4
if 0 == rnd:
msg = args['reply_user'] + ''': peng (You're dead now.)'''
log.info('sent pong (variant)')
elif 1 == rnd:
msg = args['reply_user'] + ''': I don't like you, leave me alone.'''
log.info('sent pong (dontlike)')
else:
msg = args['reply_user'] + ''': pong'''
log.info('sent pong')
return {
'msg': msg
}
@pluginfunction('klammer', 'prints an anoying paper clip aka. Karl Klammer', ptypes_COMMAND, @pluginfunction('klammer', 'prints an anoying paper clip aka. Karl Klammer', ptypes_COMMAND,
ratelimit_class=RATE_FUN | RATE_GLOBAL) ratelimit_class=RATE_FUN | RATE_GLOBAL)
def command_klammer(argv, **args): def command_klammer(argv, **args):
log.info('sent karl klammer') log.info('sent karl klammer')
return { return {
'msg': ( 'msg': (
@@ -44,9 +94,34 @@ def command_klammer(argv, **args):
} }
@pluginfunction('excuse', 'prints BOFH style excuses', ptypes_COMMAND)
def command_excuse(argv, **args):
log.info('BOFH plugin called')
excuse = random.sample(excuses, 1)[0]
return {
'msg': args['reply_user'] + ': ' + excuse
}
@pluginfunction('terminate', 'hidden prototype', ptypes_COMMAND, ratelimit_class=RATE_FUN | RATE_GLOBAL)
def command_terminate(argv, **args):
return {
'msg': 'insufficient power supply, please connect fission module'
}
@pluginfunction('source', 'prints git URL', ptypes_COMMAND)
def command_source(argv, **_):
log.info('sent source URL')
return {
'msg': 'My source code can be found at %s' % config.conf_get('src-url')
}
@pluginfunction('unikot', 'prints an unicode string', ptypes_COMMAND, ratelimit_class=RATE_FUN | RATE_GLOBAL) @pluginfunction('unikot', 'prints an unicode string', ptypes_COMMAND, ratelimit_class=RATE_FUN | RATE_GLOBAL)
def command_unicode(argv, **args): def command_unicode(argv, **args):
log.info('sent some unicode') log.info('sent some unicode')
return { return {
'msg': ( 'msg': (
@@ -58,23 +133,14 @@ def command_unicode(argv, **args):
} }
@pluginfunction('source', 'prints git URL', ptypes_COMMAND)
def command_source(argv, **_):
log.info('sent source URL')
return {
'msg': 'My source code can be found at %s' % config.conf_get('src-url')
}
@pluginfunction('dice', 'rolls a dice, optional N times', ptypes_COMMAND, ratelimit_class=RATE_INTERACTIVE) @pluginfunction('dice', 'rolls a dice, optional N times', ptypes_COMMAND, ratelimit_class=RATE_INTERACTIVE)
def command_dice(argv, **args): def command_dice(argv, **args):
try: try:
count = 1 if len(argv) < 2 else int(argv[1]) count = 1 if argv else int(argv[0])
except ValueError as e: except ValueError as e:
return { return {
'msg': '%s: dice: error when parsing int(%s): %s' % ( 'msg': '%s: dice: error when parsing int(%s): %s' % (
args['reply_user'], argv[1], str(e) args['reply_user'], argv[0], str(e)
) )
} }
@@ -108,87 +174,31 @@ def command_dice(argv, **args):
@pluginfunction('choose', 'chooses randomly between arguments', ptypes_COMMAND, ratelimit_class=RATE_INTERACTIVE) @pluginfunction('choose', 'chooses randomly between arguments', ptypes_COMMAND, ratelimit_class=RATE_INTERACTIVE)
def command_choose(argv, **args): def command_choose(argv, **args):
alternatives = argv
alternatives = argv[1:]
if 2 > len(alternatives): if 2 > len(alternatives):
return { return {
'msg': '%s: choosing between one or less things is pointless' % args['reply_user'] 'msg': '%s: choosing between one or less things is pointless' % args['reply_user']
} }
choice = random.choice(alternatives) choice = random.choice(alternatives)
log.info('sent random choice') log.info('sent random choice')
return { return {
'msg': '%s: I prefer %s!' % (args['reply_user'], choice) 'msg': '%s: I prefer %s!' % (args['reply_user'], choice)
} }
@pluginfunction('uptime', 'prints uptime', ptypes_COMMAND)
def command_uptime(argv, **args):
u = int(config.runtimeconf_get('start_time') + time.time())
plural_uptime = 's'
plural_request = 's'
if 1 == u:
plural_uptime = ''
if 1 == int(config.runtimeconf_get('request_counter')):
plural_request = ''
log.info('sent statistics')
return {
'msg': args['reply_user'] + (''': happily serving for %d second%s, %d request%s so far.''' % (
u, plural_uptime, int(config.runtimeconf_get('request_counter')), plural_request))
}
@pluginfunction('ping', 'sends pong', ptypes_COMMAND, ratelimit_class=RATE_INTERACTIVE)
def command_ping(argv, **args):
rnd = random.randint(0, 3) # 1:4
if 0 == rnd:
msg = args['reply_user'] + ''': peng (You're dead now.)'''
log.info('sent pong (variant)')
elif 1 == rnd:
msg = args['reply_user'] + ''': I don't like you, leave me alone.'''
log.info('sent pong (dontlike)')
else:
msg = args['reply_user'] + ''': pong'''
log.info('sent pong')
return {
'msg': msg
}
@pluginfunction('info', 'prints info message', ptypes_COMMAND)
def command_info(argv, **args):
log.info('sent long info')
return {
'msg': args['reply_user'] + (
''': I'm a bot, my job is to extract <title> tags from posted URLs. In case I'm annoying or for further
questions, please talk to my master %s. I'm rate limited.
To make me exit immediately, highlight me with 'hangup' in the message
(emergency only, please). For other commands, highlight me with 'help'.''' % (
config.conf_get('bot_owner')))
}
@pluginfunction('teatimer', 'sets a tea timer to $1 or currently %d seconds' % config.conf_get('tea_steep_time'), @pluginfunction('teatimer', 'sets a tea timer to $1 or currently %d seconds' % config.conf_get('tea_steep_time'),
ptypes_COMMAND) ptypes_COMMAND)
def command_teatimer(argv, **args): def command_teatimer(argv, **args):
steep = config.conf_get('tea_steep_time') steep = config.conf_get('tea_steep_time')
if len(argv) > 1: if argv:
try: try:
steep = int(argv[1]) steep = int(argv[0])
except Exception as e: except ValueError as e:
return { return {
'msg': args['reply_user'] + ': error when parsing int(%s): %s' % ( 'msg': args['reply_user'] + ': error when parsing int(%s): %s' % (
argv[1], str(e) argv[0], str(e)
) )
} }
@@ -215,11 +225,11 @@ def command_teatimer(argv, **args):
@pluginfunction('unicode-lookup', 'search unicode characters', ptypes_COMMAND, @pluginfunction('unicode-lookup', 'search unicode characters', ptypes_COMMAND,
ratelimit_class=RATE_INTERACTIVE) ratelimit_class=RATE_INTERACTIVE)
def command_unicode_lookup(argv, **args): def command_unicode_lookup(argv, **args):
if len(argv) <= 1: if not argv:
return { return {
'msg': args['reply_user'] + ': usage: decode {single character}' 'msg': args['reply_user'] + ': usage: decode {single character}'
} }
search_words = argv[1:] search_words = argv
import unicode import unicode
@@ -252,15 +262,13 @@ def command_unicode_lookup(argv, **args):
@pluginfunction('decode', 'prints the long description of an unicode character', ptypes_COMMAND, @pluginfunction('decode', 'prints the long description of an unicode character', ptypes_COMMAND,
ratelimit_class=RATE_INTERACTIVE) ratelimit_class=RATE_INTERACTIVE)
def command_decode(argv, **args): def command_decode(argv, **args):
if len(argv) <= 1: if not argv:
return { return {
'msg': args['reply_user'] + ': usage: decode {single character}' 'msg': args['reply_user'] + ': usage: decode {single character}'
} }
log.info('decode called for %s' % argv[0])
log.info('decode called for %s' % argv[1])
out = [] out = []
for i, char in enumerate(argv[1]): for i, char in enumerate(argv[0]):
if i > 9: if i > 9:
out.append('... limit reached.') out.append('... limit reached.')
break break
@@ -287,31 +295,32 @@ def command_decode(argv, **args):
} }
else: else:
return { return {
'msg': [args['reply_user'] + ': decoding %s:' % argv[1]] + out 'msg': [args['reply_user'] + ': decoding %s:' % argv[0]] + out
} }
@pluginfunction('show-blacklist', 'show the current URL blacklist, optionally filtered', ptypes_COMMAND) @pluginfunction('show-blacklist', 'show the current URL blacklist, optionally filtered', ptypes_COMMAND)
def command_show_blacklist(argv, **args): def command_show_blacklist(argv, **args):
log.info('sent URL blacklist') log.info('sent URL blacklist')
if argv:
argv1 = None if len(argv) < 2 else argv[1] urlpart = argv[0]
else:
urlpart = None
return { return {
'msg': [ 'msg': [
args['reply_user'] + ': URL blacklist%s: ' % ( args['reply_user'] + ': URL blacklist%s: ' % (
'' if not argv1 else ' (limited to %s)' % argv1 '' if not urlpart else ' (limited to %s)' % urlpart
) )
] + [ ] + [
b for b in config.runtime_config_store['url_blacklist'].values() if not argv1 or argv1 in b b for b in config.runtime_config_store['url_blacklist'].values() if not urlpart or urlpart in b
] ]
} }
def usersetting_get(argv, args): def usersetting_get(argv, args):
arg_user = args['reply_user'] arg_user = args['reply_user']
arg_key = argv[1] arg_key = argv[0]
if arg_user not in config.runtime_config_store['user_pref']: if arg_user not in config.runtime_config_store['user_pref']:
return { return {
@@ -331,8 +340,8 @@ def command_usersetting(argv, **args):
settings = ['spoiler'] settings = ['spoiler']
arg_user = args['reply_user'] arg_user = args['reply_user']
arg_key = argv[1] if len(argv) > 1 else None arg_key = argv[0] if len(argv) > 0 else None
arg_val = argv[2] if len(argv) > 2 else None arg_val = argv[1] if len(argv) > 1 else None
if arg_key not in settings: if arg_key not in settings:
return { return {
@@ -391,24 +400,14 @@ def command_cookie(argv, **args):
} }
@pluginfunction('terminate', 'hidden prototype', ptypes_COMMAND, ratelimit_class=RATE_FUN | RATE_GLOBAL)
def command_terminate(argv, **args):
return {
'msg': 'insufficient power supply, please connect fission module'
}
@pluginfunction('wp-en', 'crawl the english Wikipedia', ptypes_COMMAND) @pluginfunction('wp-en', 'crawl the english Wikipedia', ptypes_COMMAND)
def command_wp_en(argv, **args): def command_wp_en(argv, **args):
if argv[0]:
argv[0] = 'wp'
return command_wp(argv, lang='en', **args) return command_wp(argv, lang='en', **args)
@pluginfunction('wp', 'crawl the german Wikipedia', ptypes_COMMAND) @pluginfunction('wp', 'crawl the german Wikipedia', ptypes_COMMAND)
def command_wp(argv, lang='de', **args): def command_wp(argv, lang='de', **args):
query = ' '.join(argv[1:]) query = ' '.join(argv)
if query == '': if query == '':
return { return {
@@ -464,31 +463,20 @@ def command_wp(argv, lang='de', **args):
} }
@pluginfunction('excuse', 'prints BOFH style excuses', ptypes_COMMAND)
def command_excuse(argv, **args):
log.info('BOFH plugin called')
excuse = random.sample(excuses, 1)[0]
return {
'msg': args['reply_user'] + ': ' + excuse
}
@pluginfunction('show-moinlist', 'show the current moin reply list, optionally filtered', ptypes_COMMAND) @pluginfunction('show-moinlist', 'show the current moin reply list, optionally filtered', ptypes_COMMAND)
def command_show_moinlist(argv, **args): def command_show_moinlist(argv, **args):
log.info('sent moin reply list') log.info('sent moin reply list')
argv1 = None if len(argv) < 2 else argv[1] user = None if not argv else argv[0]
return { return {
'msg': 'msg':
'%s: moin reply list%s: %s' % ( '%s: moin reply list%s: %s' % (
args['reply_user'], args['reply_user'],
'' if not argv1 else ' (limited to %s)' % argv1, '' if not user else ' (limited to %s)' % user,
', '.join([ ', '.join([
b for b in moin_strings_hi + moin_strings_bye b for b in moin_strings_hi + moin_strings_bye
if not argv1 or argv1.lower() in b.lower() if not user or user.lower() in b.lower()
]) ])
) )
} }
@@ -497,14 +485,14 @@ def command_show_moinlist(argv, **args):
@pluginfunction( @pluginfunction(
'record', 'record a message for a now offline user (usage: record {user} {some message})', ptypes_COMMAND) 'record', 'record a message for a now offline user (usage: record {user} {some message})', ptypes_COMMAND)
def command_record(argv, **args): def command_record(argv, **args):
if 3 > len(argv): if len(argv) < 2:
return { return {
'msg': '%s: usage: record {user} {some message}' % args['reply_user'] 'msg': '%s: usage: record {user} {some message}' % args['reply_user']
} }
target_user = argv[1].lower() target_user = argv[0].lower()
message = '{} ({}): '.format(args['reply_user'], time.strftime('%Y-%m-%d %H:%M')) message = '{} ({}): '.format(args['reply_user'], time.strftime('%Y-%m-%d %H:%M'))
message += ' '.join(argv[2:]) message += ' '.join(argv[1:])
if config.conf_get('persistent_locked'): if config.conf_get('persistent_locked'):
return { return {
@@ -530,17 +518,17 @@ def command_record(argv, **args):
def command_show_recordlist(argv, **args): def command_show_recordlist(argv, **args):
log.info('sent offline records list') log.info('sent offline records list')
argv1 = None if len(argv) < 2 else argv[1] user = None if not argv else argv[0]
return { return {
'msg': 'msg':
'%s: offline records%s: %s' % ( '%s: offline records%s: %s' % (
args['reply_user'], args['reply_user'],
'' if not argv1 else ' (limited to %s)' % argv1, '' if not user else ' (limited to %s)' % user,
', '.join( ', '.join(
[ [
'%s (%d)' % (key, len(val)) for key, val in config.runtime_config_store['user_records'].items() '%s (%d)' % (key, len(val)) for key, val in config.runtime_config_store['user_records'].items()
if not argv1 or argv1.lower() in key.lower() if not user or user.lower() in key.lower()
] ]
) )
) )
@@ -641,49 +629,53 @@ def provoke_bots(argv, **args):
@pluginfunction("remove-from-botlist", "remove a user from the botlist", ptypes_COMMAND) @pluginfunction("remove-from-botlist", "remove a user from the botlist", ptypes_COMMAND)
def remove_from_botlist(argv, **args): def remove_from_botlist(argv, **args):
if len(argv) != 2: if not argv:
return {'msg': "wrong number of arguments!"} return {'msg': "wrong number of arguments!"}
suspect = argv[0]
if args['reply_user'] != config.conf_get('bot_owner') and args['reply_user'] != argv[1]: if args['reply_user'] != config.conf_get('bot_owner') and args['reply_user'] != suspect:
return {'msg': "only %s or the bot may do this!" % config.conf_get('bot_owner')} return {'msg': "only %s or the bot may do this!" % config.conf_get('bot_owner')}
if argv[1] in config.runtime_config_store['other_bots']: if suspect in config.runtime_config_store['other_bots']:
config.runtime_config_store['other_bots'].remove(argv[1]) config.runtime_config_store['other_bots'].remove(suspect)
config.runtimeconf_persist() config.runtimeconf_persist()
return {'msg': '%s was removed from the botlist.' % argv[1]} return {'msg': '%s was removed from the botlist.' % suspect}
else: else:
return False return False
@pluginfunction("add-to-botlist", "add a user to the botlist", ptypes_COMMAND) @pluginfunction("add-to-botlist", "add a user to the botlist", ptypes_COMMAND)
def add_to_botlist(argv, **args): def add_to_botlist(argv, **args):
if len(argv) != 2: if not argv:
return {'msg': "wrong number of arguments!"} return {'msg': "wrong number of arguments!"}
suspect = argv[0]
if args['reply_user'] != config.conf_get('bot_owner'): if args['reply_user'] != config.conf_get('bot_owner'):
return {'msg': "only %s may do this!" % config.conf_get('bot_owner')} return {'msg': "only %s may do this!" % config.conf_get('bot_owner')}
if argv[1] not in config.runtime_config_store['other_bots']: if suspect not in config.runtime_config_store['other_bots']:
config.runtime_config_store['other_bots'].append(argv[1]) config.runtime_config_store['other_bots'].append(suspect)
config.runtimeconf_persist() config.runtimeconf_persist()
return {'msg': '%s was added to the botlist.' % argv[1]} return {'msg': '%s was added to the botlist.' % suspect}
else: else:
return {'msg': '%s is already in the botlist.' % argv[1]} return {'msg': '%s is already in the botlist.' % suspect}
@pluginfunction("set-status", "set bot status", ptypes_COMMAND) @pluginfunction("set-status", "set bot status", ptypes_COMMAND)
def set_status(argv, **args): def set_status(argv, **args):
if len(argv) != 2: if not argv:
return return
else:
command = argv[0]
if argv[1] == 'mute' and args['reply_user'] == config.conf_get('bot_owner'): if command == 'mute' and args['reply_user'] == config.conf_get('bot_owner'):
return { return {
'presence': { 'presence': {
'status': 'xa', 'status': 'xa',
'msg': 'I\'m muted now. You can unmute me with "%s: set_status unmute"' % config.conf_get("bot_nickname") 'msg': 'I\'m muted now. You can unmute me with "%s: set_status unmute"' % config.conf_get("bot_nickname")
} }
} }
elif argv[1] == 'unmute' and args['reply_user'] == config.conf_get('bot_owner'): elif command == 'unmute' and args['reply_user'] == config.conf_get('bot_owner'):
return { return {
'presence': { 'presence': {
'status': None, 'status': None,
@@ -703,10 +695,10 @@ def save_config(argv, **args):
@pluginfunction('flausch', "make people flauschig", ptypes_COMMAND, ratelimit_class=RATE_FUN) @pluginfunction('flausch', "make people flauschig", ptypes_COMMAND, ratelimit_class=RATE_FUN)
def flausch(argv, **args): def flausch(argv, **args):
if len(argv) != 2: if not argv:
return return
return { return {
'msg': '{}: *flausch*'.format(argv[1]) 'msg': '{}: *flausch*'.format(argv[0])
} }
@@ -730,11 +722,11 @@ def reload_runtimeconfig(argv, **args):
@pluginfunction('snitch', "tell on a spammy user", ptypes_COMMAND) @pluginfunction('snitch', "tell on a spammy user", ptypes_COMMAND)
def ignore_user(argv, **args): def ignore_user(argv, **args):
if len(argv) != 2: if not argv:
return {'msg': 'syntax: "{}: snitch username"'.format(config.conf_get("bot_nickname"))} return {'msg': 'syntax: "{}: snitch username"'.format(config.conf_get("bot_nickname"))}
then = time.time() + 15*60 then = time.time() + 15*60
spammer = argv[1] spammer = argv[0]
if spammer == config.conf_get("bot_owner"): if spammer == config.conf_get("bot_owner"):
return { return {

View File

@@ -275,7 +275,7 @@ class UrlBot(IdleBot):
parser_list=[pl.plugin_name for pl in plugin_storage[ptypes_PARSE]], parser_list=[pl.plugin_name for pl in plugin_storage[ptypes_PARSE]],
reply_user=reply_user, reply_user=reply_user,
msg_obj=msg_obj, msg_obj=msg_obj,
argv=words[1:] argv=words[2:]
) )
if ret: if ret: