X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=turbot%2Finteraction.py;h=8f94a064071674d75dc1a039ac2d273b2a72e33b;hb=99dfffda6befec46a1ecee6cae093224ee7cc6d0;hp=ab35a395bbdc6fe1020b3d84611865a6bf6db69b;hpb=a22c3fcd19986ee27219381716cbfcd105267581;p=turbot diff --git a/turbot/interaction.py b/turbot/interaction.py index ab35a39..8f94a06 100644 --- a/turbot/interaction.py +++ b/turbot/interaction.py @@ -7,6 +7,7 @@ import json import re import requests from botocore.exceptions import ClientError +from boto3.dynamodb.conditions import Key from turbot.slack import slack_send_message actions = {} @@ -95,32 +96,63 @@ def new_hunt_submission(turb, payload, metadata): "Hunt ID can only contain lowercase letters, " + "numbers, and underscores") - # Check to see if the hunts table exists - hunts_table = turb.db.Table("hunts") - + # Check to see if the turbot table exists try: - exists = hunts_table.table_status in ("CREATING", "UPDATING", - "ACTIVE") + exists = turb.table.table_status in ("CREATING", "UPDATING", + "ACTIVE") except ClientError: exists = False - # Create the hunts table if necessary. + # Create the turbot table if necessary. if not exists: - hunts_table = turb.db.create_table( - TableName='hunts', + turb.table = turb.db.create_table( + TableName='turbot', KeySchema=[ - {'AttributeName': 'channel_id', 'KeyType': 'HASH'}, + {'AttributeName': 'PK', 'KeyType': 'HASH'}, + {'AttributeName': 'SK', 'KeyType': 'RANGE'}, ], AttributeDefinitions=[ + {'AttributeName': 'PK', 'AttributeType': 'S'}, + {'AttributeName': 'SK', 'AttributeType': 'S'}, {'AttributeName': 'channel_id', 'AttributeType': 'S'}, + {'AttributeName': 'hunt_id', 'AttributeType': 'S'}, ], ProvisionedThroughput={ 'ReadCapacityUnits': 5, 'WriteCapacityUnits': 5 - } + }, + GlobalSecondaryIndexes=[ + { + 'IndexName': 'channel_id_index', + 'KeySchema': [ + {'AttributeName': 'channel_id', 'KeyType': 'HASH'} + ], + 'Projection': { + 'ProjectionType': 'ALL' + }, + 'ProvisionedThroughput': { + 'ReadCapacityUnits': 5, + 'WriteCapacityUnits': 5 + } + }, + { + 'IndexName': 'hunt_id_index', + 'KeySchema': [ + {'AttributeName': 'hunt_id', 'KeyType': 'HASH'} + ], + 'Projection': { + 'ProjectionType': 'ALL' + }, + 'ProvisionedThroughput': { + 'ReadCapacityUnits': 5, + 'WriteCapacityUnits': 5 + } + } + ] ) - return submission_error("hunt_id", - "Still bootstrapping hunts table. Try again.") + return submission_error( + "hunt_id", + "Still bootstrapping turbot table. Try again in a minute, please.") # Create a channel for the hunt try: @@ -135,12 +167,14 @@ def new_hunt_submission(turb, payload, metadata): # Insert the newly-created hunt into the database # (leaving it as non-active for now until the channel-created handler # finishes fixing it up with a sheet and a companion table) - hunts_table.put_item( + turb.table.put_item( Item={ - 'channel_id': channel_id, + "PK": "hunt-{}".format(hunt_id), + "SK": "hunt-{}".format(hunt_id), + "hunt_id": hunt_id, + "channel_id": channel_id, "active": False, "name": name, - "hunt_id": hunt_id, "url": url } ) @@ -242,7 +276,32 @@ def channel_is_hunt(turb, channel_id): Returns a dict (filled with database entries) if there is a hunt for this channel, otherwise returns None.""" - return get_table_item(turb, "hunts", 'channel_id', channel_id) + response = turb.table.query( + IndexName = "channel_id_index", + KeyConditionExpression=Key("channel_id").eq(channel_id) + ) + + if 'Items' not in response: + return None + + return response['Items'][0] + +def find_hunt_for_hunt_id(turb, hunt_id): + """Given a hunt ID find the database for for that hunt + + Returns None if hunt ID is not found, otherwise a + dictionary with all fields from the hunt's row in the table, + (channel_id, active, hunt_id, name, url, sheet_url, etc.). + + """ + turbot_table = turb.db.Table("turbot") + + response = turbot_table.get_item(Key={'PK': 'hunt-{}'.format(hunt_id)}) + + if 'Item' in response: + return response['Item'] + else: + return None def find_hunt_for_channel(turb, channel_id, channel_name): """Given a channel ID/name find the id/name of the hunt for this channel @@ -250,29 +309,22 @@ def find_hunt_for_channel(turb, channel_id, channel_name): This works whether the original channel is a primary hunt channel, or if it is one of the channels of a puzzle belonging to the hunt. - Returns a tuple of (hunt_name, hunt_id) or (None, None).""" + Returns None if channel does not belong to a hunt, otherwise a + dictionary with all fields from the hunt's row in the table, + (channel_id, active, hunt_id, name, url, sheet_url, etc.). + + """ - (hunt, _) = channel_is_hunt(turb, channel_id) + hunt = channel_is_hunt(turb, channel_id) if hunt: - return (hunt['hunt_id'], hunt['name']) + return hunt # So we're not a hunt channel, let's look to see if we are a # puzzle channel with a hunt-id prefix. hunt_id = channel_name.split('-')[0] - hunts_table = turb.db.Table("hunts") - - response = hunts_table.scan( - FilterExpression='hunt_id = :hunt_id', - ExpressionAttributeValues={':hunt_id': hunt_id} - ) - - if 'Items' in response and len(response['Items']): - item = response['Items'][0] - return (item['hunt_id'], item['name']) - - return (None, None) + return find_hunt_for_hunt_id(turb, hunt_id) def puzzle(turb, body, args): """Implementation of the /puzzle command @@ -284,23 +336,23 @@ def puzzle(turb, body, args): channel_name = body['channel_name'][0] trigger_id = body['trigger_id'][0] - (hunt_id, hunt_name) = find_hunt_for_channel(turb, - channel_id, - channel_name) + hunt = find_hunt_for_channel(turb, + channel_id, + channel_name) - if not hunt_id: + if not hunt: return bot_reply("Sorry, this channel doesn't appear to " + "be a hunt or puzzle channel") view = { "type": "modal", "private_metadata": json.dumps({ - "hunt_id": hunt_id, + "hunt_id": hunt['hunt_id'], }), "title": {"type": "plain_text", "text": "New Puzzle"}, "submit": { "type": "plain_text", "text": "Create" }, "blocks": [ - section_block(text_block("*For {}*".format(hunt_name))), + section_block(text_block("*For {}*".format(hunt['name']))), input_block("Puzzle name", "name", "Name of the puzzle"), input_block("Puzzle URL", "url", "External URL of puzzle", optional=True) @@ -345,17 +397,18 @@ def puzzle_submission(turb, payload, metadata): "Error creating Slack channel {}: {}" .format(hunt_dash_channel, e.response['error'])) - puzzle_channel_id = response['channel']['id'] + channel_id = response['channel']['id'] # Insert the newly-created puzzle into the database - table = turb.db.Table(hunt_id) - table.put_item( + turb.table.put_item( Item={ - "channel_id": puzzle_channel_id, + "PK": "hunt-{}".format(hunt_id), + "SK": "puzzle-{}".format(puzzle_id), + "puzzle_id": puzzle_id, + "channel_id": channel_id, "solution": [], "status": 'unsolved', "name": name, - "puzzle_id": puzzle_id, "url": url, } ) @@ -374,7 +427,7 @@ def set_channel_topic(turb, puzzle): description = '' if status == 'solved': - description += "Solved: `{}` ".format('`, `'.join(puzzle['solution'])) + description += "SOLVED: `{}` ".format('`, `'.join(puzzle['solution'])) description += name @@ -441,9 +494,30 @@ def solved(turb, body, args): turb.slack_client, channel_id, "Puzzle mark solved by {}: `{}`".format(user_name, args)) + # Also report the solution to the hunt channel + hunt = find_hunt_for_hunt_id(turb, puzzle['hunt_id']) + slack_send_message( + turb.slack_client, hunt['channel_id'], + "Puzzle <{}|{}> has been solved!".format( + puzzle['channel_url'], + puzzle['name']) + ) + # And update the puzzle's description set_channel_topic(turb, puzzle) + # And rename the sheet to prefix with "SOLVED: " + turbot.sheets.renameSheet(turb, puzzle['sheet_url'], + 'SOLVED: ' + puzzle['name']) + + # Finally, rename the Slack channel to add the suffix '-solved' + channel_name = "{}-{}-solved".format( + puzzle['hunt_id'], + puzzle['puzzle_id']) + turb.slack_client.conversations_rename( + channel=puzzle['channel_id'], + name=channel_name) + return lambda_ok commands["/solved"] = solved