124 lines
3.1 KiB
Python
124 lines
3.1 KiB
Python
|
# hearts.py
|
|||
|
|
# Keepin' score is better with <3.
|
|||
|
|
# Eryn Wells <eryn@erynwells.me>
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import os.path
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
HEARTS_FILE = 'hearts.json'
|
|||
|
|
|
|||
|
|
PLUSES = {'prefix': ['++', '<3', ':heart:', '❤️'],
|
|||
|
|
'suffix': ['++']}
|
|||
|
|
MINUSES = {'prefix': ['--', '–', '—', '</3', ':broken_heart:', '💔'],
|
|||
|
|
'suffix': ['--']}
|
|||
|
|
|
|||
|
|
outputs = []
|
|||
|
|
|
|||
|
|
def process_message(data):
|
|||
|
|
try:
|
|||
|
|
text = data['text']
|
|||
|
|
except KeyError:
|
|||
|
|
# TODO: Make this better.
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if text == '!top5':
|
|||
|
|
scores = top5()
|
|||
|
|
outputs.append([data['channel'], scores])
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if text.startswith('!erase'):
|
|||
|
|
name = text[len('!erase'):].strip()
|
|||
|
|
success = erase_score(name)
|
|||
|
|
if success:
|
|||
|
|
outputs.append([data['channel'], "Erased score for _{}_.".format(name)])
|
|||
|
|
else:
|
|||
|
|
outputs.append([data['channel'], "No score for _{}_.".format(name)])
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# TODO: Lots of duplicated code below. Make it better.
|
|||
|
|
operator, prefix, increment = find_operator(text)
|
|||
|
|
if operator:
|
|||
|
|
len_operator = len(operator)
|
|||
|
|
if prefix:
|
|||
|
|
name = text[len_operator:].strip()
|
|||
|
|
else:
|
|||
|
|
name = text[:-len_operator].strip()
|
|||
|
|
score = update_item(name, increment)
|
|||
|
|
outputs.append([data['channel'], '_{}_ now has a score of {}.'.format(name, score)])
|
|||
|
|
|
|||
|
|
def find_operator(text):
|
|||
|
|
prefix_increment = has_prefix(text, PLUSES['prefix'])
|
|||
|
|
if prefix_increment:
|
|||
|
|
return prefix_increment, True, True
|
|||
|
|
|
|||
|
|
suffix_increment = has_suffix(text, PLUSES['suffix'])
|
|||
|
|
if suffix_increment:
|
|||
|
|
return suffix_increment, False, True
|
|||
|
|
|
|||
|
|
prefix_decrement = has_prefix(text, MINUSES['prefix'])
|
|||
|
|
if prefix_decrement:
|
|||
|
|
return prefix_decrement, True, False
|
|||
|
|
|
|||
|
|
suffix_decrement = has_suffix(text, MINUSES['suffix'])
|
|||
|
|
if suffix_decrement:
|
|||
|
|
return suffix_decrement, False, False
|
|||
|
|
|
|||
|
|
return None, None, None
|
|||
|
|
|
|||
|
|
def top5():
|
|||
|
|
data = read_data()
|
|||
|
|
items = [(score, name) for name, score in data.items()]
|
|||
|
|
items.sort()
|
|||
|
|
out = ''
|
|||
|
|
for idx in range(5):
|
|||
|
|
try:
|
|||
|
|
item = items[idx]
|
|||
|
|
out += '{}. _{}_ : {}\n'.format(idx+1, item[1], item[0])
|
|||
|
|
except IndexError:
|
|||
|
|
break
|
|||
|
|
return out
|
|||
|
|
|
|||
|
|
def update_item(name, increment):
|
|||
|
|
data = read_data()
|
|||
|
|
score = data.get(name)
|
|||
|
|
if not score:
|
|||
|
|
score = 0
|
|||
|
|
score += 1 if increment else -1
|
|||
|
|
data[name] = score
|
|||
|
|
write_data(data)
|
|||
|
|
return score
|
|||
|
|
|
|||
|
|
def erase_score(name):
|
|||
|
|
data = read_data()
|
|||
|
|
try:
|
|||
|
|
del data[name]
|
|||
|
|
except KeyError:
|
|||
|
|
return False
|
|||
|
|
else:
|
|||
|
|
write_data(data)
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def read_data():
|
|||
|
|
if not os.path.exists(HEARTS_FILE):
|
|||
|
|
return {}
|
|||
|
|
with open(HEARTS_FILE) as f:
|
|||
|
|
return json.load(f)
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def write_data(obj):
|
|||
|
|
with open(HEARTS_FILE, 'w') as f:
|
|||
|
|
json.dump(obj, f, sort_keys=True, indent=4)
|
|||
|
|
|
|||
|
|
def has_prefix(text, prefixes):
|
|||
|
|
for p in prefixes:
|
|||
|
|
if text.startswith(p):
|
|||
|
|
return p
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def has_suffix(text, suffixes):
|
|||
|
|
for s in suffixes:
|
|||
|
|
if text.endswith(s):
|
|||
|
|
return s
|
|||
|
|
return None
|