125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
from data.cat import Cat
|
|
import systems.ui as ui
|
|
import re
|
|
import data.text
|
|
import random
|
|
|
|
|
|
def check_cat_name(name):
|
|
name = name.strip()
|
|
if not name:
|
|
return "Please choose a name:"
|
|
if len(name) < 4 or len(name) > 20:
|
|
return "Please choose a name greater than 4 characters and less than 20:"
|
|
if re.search(r"[^a-zA-Z \-']", name):
|
|
return "Please only use letters, spaces, hyphens and apostrophes:"
|
|
return "Ok"
|
|
|
|
|
|
def generate_cat_choice(personality=None):
|
|
size = random.choice(data.text.CAT_SIZES)
|
|
color = random.choice(data.text.CAT_COLORS)
|
|
eyes = random.choice(data.text.CAT_EYE_COLORS)
|
|
personality = personality or random.choice(data.text.CAT_PERSONALITIES)
|
|
article = "An" if size[0] in "aeiou" else "A"
|
|
description = f"{article} {size} {color} kitten with {eyes} eyes who {personality}"
|
|
return description, {
|
|
"size": size,
|
|
"color": color,
|
|
"eyes": eyes,
|
|
"personality": personality,
|
|
}
|
|
|
|
|
|
def generate_cat_choices(n=5):
|
|
personalities = random.sample(data.text.CAT_PERSONALITIES, n)
|
|
return [generate_cat_choice(p) for p in personalities]
|
|
|
|
|
|
def shelter(include_welcome=True):
|
|
while True:
|
|
cat_choices = generate_cat_choices()
|
|
descriptions = []
|
|
for desc, _ in cat_choices:
|
|
descriptions.append(desc)
|
|
descriptions.append("Reroll")
|
|
kitten_description = ui.select(
|
|
f"{"Welcome to the shelter! " if include_welcome else ""}Please choose a kitten to adopt.",
|
|
descriptions,
|
|
)
|
|
if kitten_description != "Reroll":
|
|
break
|
|
for desc, traits in cat_choices:
|
|
if desc == kitten_description:
|
|
break
|
|
name = ui.text("Please choose a name for your kitten:")
|
|
while True:
|
|
check_result = check_cat_name(name)
|
|
if check_result != "Ok":
|
|
name = ui.text(check_result)
|
|
else:
|
|
break
|
|
if not ui.confirm(
|
|
f"Are you sure you want to adopt {name}, a{kitten_description[1:]}?"
|
|
):
|
|
return shelter(include_welcome=False)
|
|
print(f"You pick up {name}, {name} is ready to leave the shelter.")
|
|
return Cat(name, traits)
|
|
|
|
|
|
def pet(cat: Cat):
|
|
print(f"Mash keys to pet {cat.name}, press enter when you're done.")
|
|
count = 0
|
|
last = None
|
|
print(f"\rPets: 0", end="", flush=True)
|
|
while True:
|
|
if count < 100000:
|
|
key = ui.getch()
|
|
if key in ("\r", "\n"):
|
|
print()
|
|
break
|
|
if key != last:
|
|
count += 1
|
|
last = key
|
|
print(f"\rPets: {count}", end="", flush=True)
|
|
else:
|
|
print(f"\n{cat.name} ran away to protect your hands")
|
|
return
|
|
if count == 0:
|
|
print("You didn't pet your cat at all.")
|
|
elif count < 50:
|
|
print(f"You didn't pet your cat enough, {cat.name} wants more pets.")
|
|
elif count <= 200:
|
|
print(f"You pet {cat.name} a lot, {cat.name} is happy.")
|
|
elif count > 75000:
|
|
print(f"{cat.name} is getting extremely worried about your hands.")
|
|
elif count > 50000:
|
|
print("Seriously. Stop.")
|
|
elif count > 20000:
|
|
print("You should probably stop now.")
|
|
elif count > 10000:
|
|
print(f"What are you even doing at this point?")
|
|
elif count > 1000:
|
|
print(f"{cat.name} has had enough pets.")
|
|
elif count > 500:
|
|
print(f"You pet {cat.name} an absurd amount of times.")
|
|
else:
|
|
print(f"You pet {cat.name} a lot. {cat.name} is very happy.")
|
|
|
|
|
|
def house(cat: Cat):
|
|
print("Welcome to your house!")
|
|
while True:
|
|
match ui.select(
|
|
"Please choose an option",
|
|
["Check on your cat", "Pet your cat", "Leave your house"],
|
|
):
|
|
case "Check on your cat":
|
|
print(
|
|
f"{cat.name} - a {cat.traits["size"]} {cat.traits["color"]} with {cat.traits["eyes"]} eyes"
|
|
)
|
|
case "Pet your cat":
|
|
pet(cat)
|
|
case "Leave your house":
|
|
break
|