waitonbot/job_signoff.py

from fuzzywuzzy import process

import channel_util
import google_api
import identifier
import slack_util
import scroll_util

SHEET_ID = "1lPj9GjB00BuIq9GelOWh5GmiGsheLlowPnHLnWBvMOM"

# Note: These ranges use named range feature of google sheets.
# To edit range of jobs, edit the named range in Data -> Named Ranges in the Goole Sheets page
output_range = "JobScoreTracker"

MIN_RATIO = 0.9
SIGNOFF_REWARD = 0.1
SAFETY_DELAY = 5


# Used to track a sufficiently shitty typed name
class BadName(Exception):
    def __init__(self, name, score):
        self.name = name
        self.score = score

    def as_response(self):
        return "Unable to perform operation. Best name match {} had a match score of {}, falling short of match ratio " \
               "{}. Please type name better.".format(self.name, self.score, MIN_RATIO)


def get_curr_points():
    # Get the current stuff
    curr_output = google_api.get_sheet_range(SHEET_ID, output_range)

    # Each element: force length of 2.
    def row_fixer(r):
        if len(r) == 0:
            return ["", 0]
        elif len(r) == 1:
            return [r[0], 0]
        else:
            try:
                v = int(r[1])
            except ValueError:
                v = 0
            return [r[0], v]

    # Fix each row
    curr_output = [row_fixer(x) for x in curr_output]

    # Cut off the chaff - doesn't seem to be necessary.
    return curr_output


def put_points(vals):
    google_api.set_sheet_range(SHEET_ID, output_range, vals)


def signoff_callback(slack, msg, match):
    # Find the index of our person.
    name = match.group(1)

    # Also, who just signed us off?
    signer = identifier.lookup_msg_brother(msg)["name"]

    # Try giving the person a point
    try:
        (bro_name, bro_total), (ass_name, ass_total) = adjust_scores((name, 1), (signer, SIGNOFF_REWARD))
        slack_util.reply(slack, msg, "Gave {} one housejob point.\n"
                                     "They now have {} for this period.\n"
                                     "You ({}) were credited with the signoff".format(bro_name, bro_total, ass_name))
        alert_user(slack, bro_name,
                   "You, who we believe to be {}, just had your house job signed off by {}!".format(bro_name, ass_name))
    except BadName as e:
        # We didn't find a name - no action was performed.
        slack_util.reply(slack, msg, e.as_response())


def alert_user(slack, name, saywhat):
    """
    DM a brother saying something
    """
    brother_dict = scroll_util.find_by_name(name)
    # We do this as a for loop just in case multiple people reg. to same scroll for some reason (e.g. dup accounts)
    for slack_id in identifier.lookup_brother_userids(brother_dict):
        dm_id = slack_util.im_channel_for_id(slack, slack_id)
        if dm_id:
            slack_util.reply(slack, None, saywhat, to_channel=dm_id, in_thread=False)
        else:
            print("Warning: unable to find dm for brother {}".format(brother_dict))


def punish_callback(slack, msg, match):
    # Find the index of our person.
    name = match.group(2)

    # Also, who just signed us off?
    signer = identifier.lookup_msg_brother(msg)["name"]

    # Try giving the person a point
    try:
        (bro_name, bro_total), (ass_name, ass_total) = adjust_scores((name, -1), (signer, -SIGNOFF_REWARD))
        slack_util.reply(slack, msg, "Took one housejob point from {}.\n"
                                     "They now have {} for this period.\n"
                                     "Under the assumption that this was to undo a mistake, we have deducted the "
                                     "usual signoff reward from you, ({}).\n "
                                     "You can easily earn it back by signing off the right person ;).".format(bro_name,
                                                                                                              bro_total,
                                                                                                              ass_name))
        alert_user(slack, bro_name,
                   "You, who we believe to be {}, just had your house job UN-signed off by {}.\n"
                   "Perhaps the asshoman made a mistake when they first signed you off.\n"
                   "If you believe this to be a mistake, talk to them".format(bro_name, signer))

    except BadName as e:
        # We didn't find a name - no action was performed.
        slack_util.reply(slack, msg, e.as_response())


def adjust_scores(*name_delta_tuples):
    # Get the current stuff
    points = get_curr_points()
    names = [p[0] for p in points]
    modified_user_indexes = []

    for name, delta in name_delta_tuples:
        # Find our guy
        target_name, ratio = process.extractOne(name, names)
        ratio = ratio / 100.0

        # If bad ratio, error
        if ratio < MIN_RATIO:
            raise BadName(target_name, ratio)

        # Where is he in the list?
        target_index = names.index(target_name)

        # Get his current score
        curr_score = points[target_index][1]

        # target should be in the form index, (name, score)
        target_new = [target_name, curr_score + delta]

        # Put it back
        points[target_index] = target_new

        # Record where we edited
        modified_user_indexes.append(target_index)

    # Push all to sheets if exit loop without error
    put_points(points)

    # Conver indexes to rows, then return the adjusted name/score_tuples
    return [points[i] for i in modified_user_indexes]


def reset_callback(slack, msg, match):
    pass
    # reset_callback()


signoff_hook = slack_util.Hook(signoff_callback, pattern=r"signoff\s+(.*)",
                               channel_whitelist=[channel_util.HOUSEJOBS])
undosignoff_hook = slack_util.Hook(punish_callback, pattern=r"(unsignoff|undosignoff|undo)\s+(.*)",
                                   channel_whitelist=[channel_util.HOUSEJOBS])
# reset_hook = slack_util.Hook(reset_callback, pattern=r"reset_job_scores")