90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
import logging
|
|
import yaml
|
|
import time
|
|
import re
|
|
import redis
|
|
|
|
from matrix_bot_api.matrix_bot_api import MatrixBotAPI
|
|
from matrix_bot_api.mregex_handler import MRegexHandler
|
|
from matrix_bot_api.mcommand_handler import MCommandHandler
|
|
|
|
|
|
def show_karma_cb(room, event):
|
|
allkkeys = rs.keys("partybot:k_" + event['room_id'] + "@*")
|
|
|
|
reply = ""
|
|
for k in allkkeys:
|
|
handle = k.decode().split('@')[1]
|
|
karma = rs.get(k)
|
|
reply += handle + " has " + str(karma.decode()) + " karma.\n"
|
|
|
|
room.send_notice(reply)
|
|
|
|
|
|
def karma_cb(room, event):
|
|
members = room.get_joined_members()
|
|
msender = re.search('@([\w]+_)?([^:]+)', event['sender'])
|
|
sender = msender.group(2)
|
|
tokens = re.search("([\w]+)(\-\-|\+\+)", event['content']['body'])
|
|
handle = tokens.group(1)
|
|
sign = tokens.group(2)
|
|
reply = ""
|
|
|
|
if sender.lower() == handle.lower():
|
|
reply = "@" + sender + ": Silly human, your karma must be decided by others!"
|
|
room.send_notice(reply)
|
|
return
|
|
|
|
for mxid in members:
|
|
alias = re.search('@(pnut_)?([^:]+)', mxid).group(2)
|
|
if alias.lower() == handle.lower():
|
|
print('match')
|
|
|
|
if sign == '++':
|
|
rs.incr("partybot:k_" + event['room_id'] + "@" + handle)
|
|
elif sign == '--':
|
|
rs.decr("partybot:k_" + event['room_id'] + "@" + handle)
|
|
|
|
_show_karma(room, event['room_id'], handle)
|
|
|
|
break
|
|
|
|
|
|
def _show_karma(room, room_id, handle):
|
|
|
|
k = int(rs.get("partybot:k_" + room_id + "@" + handle))
|
|
reply = handle + " has " + str(k) + " karma in this channel."
|
|
room.send_notice(reply)
|
|
|
|
|
|
def recv_event_q(queue, post):
|
|
if queue == "MNDP":
|
|
room = bot.client.rooms[config['queue']['MNDP']]
|
|
room.send_notice(post)
|
|
elif queue == "JUKEBOX":
|
|
room = bot.client.rooms[config['queue']['JUKEBOX']]
|
|
room.send_notice(post)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
logging.basicConfig(level=logging.DEBUG,filename='partybot.log')
|
|
|
|
with open("partybot-config.yaml", "rb") as config_file:
|
|
config = yaml.load(config_file)
|
|
|
|
rs = redis.StrictRedis()
|
|
bot = MatrixBotAPI(config['username'], config['password'], config['url'])
|
|
|
|
bot.add_handler(MRegexHandler("([\w]+)(\-\-|\+\+)", karma_cb))
|
|
bot.add_handler(MCommandHandler("karma", show_karma_cb))
|
|
|
|
bot.start_polling()
|
|
while True:
|
|
time.sleep(1)
|
|
item = rs.rpop('MNDP')
|
|
if item:
|
|
recv_event_q('MNDP', item.decode())
|
|
item = rs.rpop('JUKEBOX')
|
|
if item:
|
|
recv_event_q('JUKEBOX', item.decode())
|