|
|
import random
|
|
|
from db_utils import get_or_create_user, add_pull, get_db_connection
|
|
|
from add_character import add_character
|
|
|
|
|
|
def get_character():
|
|
|
''' Gets a random character from the database'''
|
|
|
conn = get_db_connection()
|
|
|
cur = conn.cursor()
|
|
|
cur.execute('SELECT * FROM characters')
|
|
|
characters = cur.fetchall()
|
|
|
conn.close()
|
|
|
|
|
|
if not characters:
|
|
|
return None, None, None, None
|
|
|
|
|
|
weights = [c['weight'] for c in characters]
|
|
|
chosen = random.choices(characters, weights=weights, k=1)[0]
|
|
|
|
|
|
return chosen['id'], chosen['name'], chosen['file_id'], chosen['rarity']
|
|
|
|
|
|
def is_float(val):
|
|
|
try:
|
|
|
float(val)
|
|
|
return True
|
|
|
except ValueError:
|
|
|
return False
|
|
|
|
|
|
|
|
|
def generate_response(parsed_command):
|
|
|
|
|
|
'''Given a command with arguments, processes the game state and
|
|
|
returns a response'''
|
|
|
|
|
|
command, full_user, arguments, note_obj = parsed_command
|
|
|
|
|
|
if command == "roll":
|
|
|
user_id = get_or_create_user(full_user)
|
|
|
character_id, character_name, file_id, rarity = get_character()
|
|
|
|
|
|
if not character_id:
|
|
|
#TODO: Can't have tuples of a single element
|
|
|
# Return these as a dict or object instead.
|
|
|
return(f"@{full_user} Uwaaa... something went wrong! No characters found. 😿")
|
|
|
|
|
|
add_pull(user_id,character_id)
|
|
|
stars = '⭐️' * rarity
|
|
|
return([f"@{full_user} 🎲 Congrats! You rolled {stars} **{character_name}**\nShe's all yours now~ 💖✨",[file_id]])
|
|
|
|
|
|
if command == "create":
|
|
|
# Example call from bot logic
|
|
|
image_url = note_obj.get("files", [{}])[0].get("url") if note_obj.get("files") else None
|
|
|
if not image_url:
|
|
|
return "You need an image to create a character, dumbass."
|
|
|
|
|
|
if len(arguments) != 3:
|
|
|
return "Please specify the following attributes in order: name, rarity, drop weighting"
|
|
|
|
|
|
if not (arguments[1].isnumeric() and 1 <= int(arguments[1]) <= 5):
|
|
|
return f"Invalid rarity: '{arguments[1]}' must be a number between 1 and 5"
|
|
|
|
|
|
if not (is_float(arguments[2]) and 0.0 < float(arguments[2]) <= 1.0):
|
|
|
return f"Invalid drop weight: '{arguments[2]}' must be a decimal value between 0.0 and 1.0"
|
|
|
|
|
|
character_id, file_id = add_character(
|
|
|
name=arguments[0],
|
|
|
rarity=int(arguments[1]),
|
|
|
weight=float(arguments[2]),
|
|
|
image_url=image_url
|
|
|
)
|
|
|
return([f"Added {arguments[0]}, ID {character_id}.",[file_id]])
|
|
|
return None
|
|
|
|
