RequesterDetails.py

from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update
from telegram.ext import Application, CallbackQueryHandler, CommandHandler, ContextTypes, MessageHandler, filters, ConversationHandler
from datetime import datetime
from decimal import Decimal

import logging
import re
import MainMenu
import DynamoDB
import configparser

####################################### Parameters #######################################

# Create config parser and read config file
config = configparser.ConfigParser()
config.read("config.ini")

# Load bot token
bot_token = config["bot_keys"]["current_bot_token"]

# Enable logging
logging.basicConfig(
    format="%(asctime)s | %(name)s | %(levelname)s | %(message)s", level=logging.INFO
)
logger = logging.getLogger(__name__)

# Create table object
table = DynamoDB.table

END_EDITING = ConversationHandler.END

####################################### Helper Functions #######################################
# Function to put item in a given table
def put_item(table, cols, RequestID, requester_chat_id, requester_user_name, canteen,
             food, tip_amount, fulfiller_chat_id, fulfiller_user_name, request_status, requester_complete, fulfiller_complete):
    data = {
        cols[0]: RequestID,
        cols[1]: requester_chat_id,
        cols[2]: requester_user_name,
        cols[3]: canteen,
        cols[4]: food,
        cols[5]: tip_amount,
        cols[6]: fulfiller_chat_id,
        cols[7]: fulfiller_user_name,
        cols[8]: request_status,
        cols[9]: requester_complete,
        cols[10]: fulfiller_complete
    }
    
    response = table.put_item(Item = data)

    return response

# Get the request from the local python DS.
def getRequest(available_requests, requestID):

    for request in available_requests:
        if requestID == request['requestID']:
            return request


####################################### Main Functions #######################################

async def promptCanteen(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    # Get the role selected from the user and store into user_data
    context.user_data[MainMenu.ROLE] = update.callback_query.data

    # Once the user clicks a button, we need to "answer" the CallbackQuery.
    await update.callback_query.answer()

    # Store information about their role.
    user = update.callback_query.from_user
    logger.info("Role of '%s' (chat_id: '%s'): '%s'", update.effective_user.name, update.effective_user.id, update.callback_query.data)

    await update.callback_query.message.reply_text(text=f"You have chosen to be a {context.user_data[MainMenu.ROLE]}.")

    # Define the canteens using a 2D array.
    inlineCanteen = [
        [InlineKeyboardButton("The Deck", callback_data="deck")],
        [InlineKeyboardButton("Frontier", callback_data="frontier")],
        [InlineKeyboardButton("Fine Foods", callback_data="fine_foods")],
        [InlineKeyboardButton("Flavours @ Utown", callback_data="flavours")],
        [InlineKeyboardButton("TechnoEdge", callback_data="technoedge")],
        [InlineKeyboardButton("PGPR", callback_data="pgpr")],
    ]

    # Transform the 2D array into an actual inline keyboard that can be interpreted by Telegram.
    inlineCanteenTG = InlineKeyboardMarkup(inlineCanteen)

    await update.callback_query.message.reply_text("Now, please select from the list of canteens.", reply_markup=inlineCanteenTG)

    return MainMenu.CANTEEN

async def selectCanteen(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    # Get the canteen selected from the requester and store the input into user_data
    context.user_data[MainMenu.CANTEEN] = update.callback_query.data

    # Once the user clicks a button, we need to "answer" the CallbackQuery.
    await update.callback_query.answer()

    # Let user know their selected canteen.
    await update.callback_query.message.reply_text(text=f"You have chosen {MainMenu.canteenDict[update.callback_query.data]} as your canteen.")

    # Store information about their canteen.
    logger.info("Requester '%s' (chat_id: '%s') selected '%s' as their canteen.", update.effective_user.name, update.effective_user.id, context.user_data[MainMenu.CANTEEN])

    await update.callback_query.message.reply_text("Great! Now, please state the food you'd like to order.")

    return MainMenu.FOOD

async def requesterFood(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:    
    # Get the food requested from the requester and store it in user_data.
    context.user_data[MainMenu.FOOD] = update.message.text

    # Store information about their name.
    user = update.message.from_user
    logger.info("Requester '%s' (chat_id: '%s') entered '%s' as their food at '%s' canteen.", update.effective_user.name, update.effective_user.id,
                context.user_data[MainMenu.FOOD], context.user_data[MainMenu.CANTEEN])

    await update.message.reply_text("Finally, how much would you like to tip the fulfiller for your request? (excluding food prices, to the nearest cent)")

    return MainMenu.OFFER_PRICE

async def requesterPrice(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    tip_pattern = r'^\d+(\.\d{1,2})?$'

    if not re.match(tip_pattern, update.message.text):
        await update.message.reply_text("Sorry, you have entered an invalid input. Please try again.")
        return MainMenu.OFFER_PRICE
    
    # Get the tipping amount from the requester and store it in user_data.
    context.user_data[MainMenu.OFFER_PRICE] = Decimal(update.message.text)

    # Store information about their name.
    user = update.message.from_user
    logger.info("Tip amount set by requester '%s': '%0.2f'", update.effective_user.name, context.user_data[MainMenu.OFFER_PRICE])

    # RequestID which consists of time + telegram username
    RequestID = "{}#{}".format(datetime.now().timestamp(), update.effective_user.id)

    # Columns in the request table in DynamoDB
    columns = ["RequestID", "requester_chat_id", "requester_user_name", "canteen", "food", "tip_amount", "fulfiller_chat_id", "fulfiller_user_name", "request_status", "requester_complete", "fulfiller_complete"]

    request = {
        "RequestID" : RequestID,
        "requester_chat_id" : update.effective_user.id,
        "requester_user_name" : update.effective_user.name,
        "canteen" : context.user_data[MainMenu.CANTEEN],
        "food" : context.user_data[MainMenu.FOOD],
        "tip_amount" : context.user_data[MainMenu.OFFER_PRICE],
        "fulfiller_chat_id" : "",
        "fulfiller_user_name" : "",
        "request_status" : "Available",
        "requester_complete" : "false",
        "fulfiller_complete" : "false",
    }

    # Store the input of the requester's request into user_data.
    context.user_data[MainMenu.REQUEST_MADE] = request
    
    # Put details of request into local python DS.
    MainMenu.available_requests.append(request)
    
    # Put item in table
    response = put_item(table,
                        columns,
                        RequestID,
                        str(update.effective_user.id),
                        update.effective_user.name,
                        context.user_data[MainMenu.CANTEEN], 
                        context.user_data[MainMenu.FOOD], 
                        context.user_data[MainMenu.OFFER_PRICE],
                        "",
                        "",
                        "Available",
                        "false",
                        "false")

    await update.message.reply_text(parse_mode="HTML", 
                                    text="Request placed! \n<b><u>Summary</u></b>" + 
                                    "\nCanteen: " + MainMenu.canteenDict[context.user_data[MainMenu.CANTEEN]] + 
                                    "\nFood: " + context.user_data[MainMenu.FOOD] +
                                    "\nTip Amount: $" + str(context.user_data[MainMenu.OFFER_PRICE]))
    
    await update.message.reply_text("We will notify and connect you with a fulfiller when found. \n\nTo cancel and delete your current request, use the /cancel command. \n\nTo edit your current request, the use /edit command.")

    return MainMenu.AWAIT_FULFILLER


async def editCanteenPrompt(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    # Once the user clicks a button, we need to "answer" the CallbackQuery.
    await update.callback_query.answer()

    # Store information about their role.
    logger.info("Requester '%s' (chat_id: '%s') is modifying their request (canteen)", update.effective_user.name, update.effective_user.id)

    # Define the canteens using a 2D array.
    inlineCanteen = [
        [InlineKeyboardButton("The Deck", callback_data="deck")],
        [InlineKeyboardButton("Frontier", callback_data="frontier")],
        [InlineKeyboardButton("Fine Foods", callback_data="fine_foods")],
        [InlineKeyboardButton("Flavours @ Utown", callback_data="flavours")],
        [InlineKeyboardButton("TechnoEdge", callback_data="technoedge")],
        [InlineKeyboardButton("PGPR", callback_data="pgpr")],
    ]

    # Transform the 2D array into an actual inline keyboard that can be interpreted by Telegram.
    inlineCanteenTG = InlineKeyboardMarkup(inlineCanteen)

    await update.callback_query.message.reply_text("Which canteen would you like to change to?", reply_markup=inlineCanteenTG)

    return MainMenu.EDIT_CANTEEN

async def editCanteen(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    # Get the canteen selected from the requester and update the user_data
    context.user_data[MainMenu.CANTEEN] = update.callback_query.data

    # Once the user clicks a button, we need to "answer" the CallbackQuery.
    await update.callback_query.answer()

    # Let user know their selected canteen.
    await update.callback_query.message.reply_text(text=f"You have changed your canteen to '{MainMenu.canteenDict[update.callback_query.data]}'")

    # Store information about their canteen.
    logger.info("Requester '%s' (chat_id: '%s') their canteen to '%s'.", update.effective_user.name, update.effective_user.id, context.user_data[MainMenu.CANTEEN])

    # Get the specific request from the list of requests of the selected canteen via the requestIndex.
    request = context.user_data[MainMenu.REQUEST_MADE]

    response = DynamoDB.table.update_item(
        Key = {
            "RequestID": request["RequestID"]
        },
        UpdateExpression = "SET canteen = :new_canteen",
        ExpressionAttributeValues = {
            ":new_canteen" : str(context.user_data[MainMenu.CANTEEN])
        }
    )

    logger.info("DynamoDB update_item response for RequestID '%s': '%s'", request["RequestID"], response["ResponseMetadata"]["HTTPStatusCode"])

    await update.callback_query.message.reply_text(parse_mode="HTML", 
                                    text="Canteen changed! Here is your updated request: \n<b><u>Summary</u></b>" + 
                                    "\nCanteen: " + MainMenu.canteenDict[context.user_data[MainMenu.CANTEEN]] + 
                                    "\nFood: " + context.user_data[MainMenu.FOOD] +
                                    "\nTip Amount: $" + str(context.user_data[MainMenu.OFFER_PRICE]))

    await update.callback_query.message.reply_text("We will notify and connect you with a fulfiller when found. \n\nTo cancel and delete your current request, use the /cancel command. \n\nTo edit your current request, the use /edit command.")

    return END_EDITING

async def editFoodPrompt(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    # Once the user clicks a button, we need to "answer" the CallbackQuery.
    await update.callback_query.answer()

    # Store information about their role.
    logger.info("Requester '%s' (chat_id: '%s') is modifying their request (food)", update.effective_user.name, update.effective_user.id)

    await update.callback_query.message.reply_text("What food would you like to change to?")

    return MainMenu.EDIT_FOOD


async def editFood(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    # Get the new requesetd food from the requester and store it in user_data.
    context.user_data[MainMenu.FOOD] = update.message.text

    # Let user know their selected canteen.
    await update.message.reply_text(text=f"You have changed your requested food to '{context.user_data[MainMenu.FOOD]}'")

   # Store information about the new tip amount.
    logger.info("New food requested by requester '%s': '%s'", update.effective_user.name, context.user_data[MainMenu.FOOD])

    # Get the request of the requester stored in user_data
    request = context.user_data[MainMenu.REQUEST_MADE]

    response = DynamoDB.table.update_item(
        Key = {
            "RequestID": request["RequestID"]
        },
        UpdateExpression = "SET food = :new_food",
        ExpressionAttributeValues = {
            ":new_food" : str(context.user_data[MainMenu.FOOD])
        }
    )

    logger.info("DynamoDB update_item response for RequestID '%s': '%s'", request["RequestID"], response["ResponseMetadata"]["HTTPStatusCode"])

    await update.message.reply_text(parse_mode="HTML", 
                                    text="Requested food changed! Here is your updated request: \n<b><u>Summary</u></b>" + 
                                    "\nCanteen: " + MainMenu.canteenDict[context.user_data[MainMenu.CANTEEN]] + 
                                    "\nFood: " + context.user_data[MainMenu.FOOD] +
                                    "\nTip Amount: $" + str(context.user_data[MainMenu.OFFER_PRICE]))

    await update.message.reply_text("We will notify and connect you with a fulfiller when found. \n\n To cancel and delete your current request, use the /cancel command. \n\n To edit your current request, the use /edit command.")

    return END_EDITING

async def editTipPrompt(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    # Once the user clicks a button, we need to "answer" the CallbackQuery.
    await update.callback_query.answer()

    # Store information about their role.
    logger.info("Requester '%s' (chat_id: '%s') is modifying their request (tip amount)", update.effective_user.name, update.effective_user.id)

    await update.callback_query.message.reply_text("What tip amount would you like to change to?")

    return MainMenu.EDIT_TIP

async def editTip(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
    tip_pattern = r'^\d+(\.\d{1,2})?$'

    if not re.match(tip_pattern, update.message.text):
        await update.message.reply_text("Sorry, you have entered an invalid input. Please try again.")
        return MainMenu.EDIT_TIP
    
    # Get the tipping amount from the requester and store it in user_data.
    context.user_data[MainMenu.OFFER_PRICE] = Decimal(update.message.text)

    # Let user know their selected canteen.
    await update.message.reply_text(text=f"You have changed your tip amount to ${context.user_data[MainMenu.OFFER_PRICE]}")

   # Store information about the new tip amount.
    logger.info("New tip amount set by requester '%s': '%0.2f'", update.effective_user.name, context.user_data[MainMenu.OFFER_PRICE])

    # Get the request of the requester stored in user_data
    request = context.user_data[MainMenu.REQUEST_MADE]

    response = DynamoDB.table.update_item(
        Key = {
            "RequestID": request["RequestID"]
        },
        UpdateExpression = "SET tip_amount = :new_tip",
        ExpressionAttributeValues = {
            ":new_tip" : str(context.user_data[MainMenu.OFFER_PRICE])
        }
    )

    logger.info("DynamoDB update_item response for RequestID '%s': '%s'", request["RequestID"], response["ResponseMetadata"]["HTTPStatusCode"])

    await update.message.reply_text(parse_mode="HTML", 
                                    text="Tip amount changed! Here is your updated request: \n<b><u>Summary</u></b>" + 
                                    "\nCanteen: " + MainMenu.canteenDict[context.user_data[MainMenu.CANTEEN]] + 
                                    "\nFood: " + context.user_data[MainMenu.FOOD] +
                                    "\nTip Amount: $" + str(context.user_data[MainMenu.OFFER_PRICE]))

    await update.message.reply_text("We will notify and connect you with a fulfiller when found. \n\n To cancel and delete your current request, use the /cancel command. \n\n To edit your current request, the use /edit command.")

    return END_EDITING