This commit is contained in:
2026-06-25 14:23:33 -04:00
parent 71de577f83
commit 32ead8417c
2 changed files with 8 additions and 6 deletions

View File

@@ -1,6 +1,6 @@
import random import random
from untitled import content from untitled import PACKAGE_ROOT, content, namegen, rules
def generate_trait_sentence(traits): def generate_trait_sentence(traits):
@@ -26,3 +26,8 @@ def generate_cat_choice(personality=None):
def generate_cat_choices(n=5): def generate_cat_choices(n=5):
personalities = random.sample(content.CAT_PERSONALITIES, n) personalities = random.sample(content.CAT_PERSONALITIES, n)
return [generate_cat_choice(p) for p in personalities] return [generate_cat_choice(p) for p in personalities]
with open(PACKAGE_ROOT / "assets" / "cat_names.txt") as f:
names = f.read().splitlines()
cat_name_generator = namegen.NameGenerator(names, rules.validate_auto_gen_cat_name)

View File

@@ -1,10 +1,7 @@
from untitled import PACKAGE_ROOT, generation, model, namegen, rules, ui from untitled import generation, model, rules, ui
def adoption(): def adoption():
with open(PACKAGE_ROOT / "assets" / "cat_names.txt") as f:
names = f.read().splitlines()
name_generator = namegen.NameGenerator(names, rules.validate_auto_gen_cat_name)
print("Welcome to the shelter!") print("Welcome to the shelter!")
while True: while True:
auto_name = "" auto_name = ""
@@ -28,7 +25,7 @@ def adoption():
break break
print(error) print(error)
else: else:
auto_name = name_generator.generate_name().capitalize() auto_name = generation.cat_name_generator.generate_name().capitalize()
if ui.confirm( if ui.confirm(
f"Do you want to adopt {name}, {generation.generate_trait_sentence(choice).lower()}?" f"Do you want to adopt {name}, {generation.generate_trait_sentence(choice).lower()}?"