stuff

untitled/screens/pet.py

@@ -0,0 +1,64 @@
+from untitled import rules, ui
+
+
+def pet(cat):
+    if cat.happiness == 100:
+        print("Your cat is already fully happy! They don't want any more pets.")
+        return
+    original_happiness = cat.happiness
+    print(f"Mash keys to pet {cat.name}, press enter when you're done.")
+    count = 0
+    last = None
+    print("\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:
+            rules.excite(cat, -5)
+            print(f"\n{cat.name} ran away to protect your hands")
+            print(
+                f"You lost 5% happiness due to stressing your cat. Before petting, your cat was {round(original_happiness,1)}% happy. You lost {round(original_happiness-cat.happiness,1)}% happiness. Your cat is now {round(cat.happiness,1)}% happy."
+            )
+            return
+    add_happiness = 0
+    if count == 0:
+        print("You didn't pet your cat at all.")
+        return
+    elif count < 50:
+        add_happiness = 5
+        print(f"You didn't pet your cat enough, {cat.name} wants more pets.")
+    elif count <= 200:
+        add_happiness = 15
+        print(f"You pet {cat.name} a lot, {cat.name} is happy.")
+    elif count <= 500:
+        add_happiness = 20
+        print(f"You pet {cat.name} a lot. {cat.name} is very happy.")
+    elif count <= 1000:
+        add_happiness = 20
+        print(f"You pet {cat.name} an absurd amount of times.")
+    elif count <= 10000:
+        add_happiness = 15
+        print(f"{cat.name} has had enough pets.")
+    elif count <= 20000:
+        add_happiness = 10
+        print("What are you even doing at this point?")
+    elif count <= 50000:
+        add_happiness = 5
+        print("You should probably stop now.")
+    elif count <= 75000:
+        add_happiness = 2
+        print("Seriously. Stop.")
+    else:
+        add_happiness = 1
+        print(f"{cat.name} is getting extremely worried about your hands.")
+    rules.excite(cat, add_happiness)
+    print(
+        f"Your cat is now {round(cat.happiness,1)}% percent happy. Before petting, your cat was {round(original_happiness,1)}% happy. You gained {round(cat.happiness-original_happiness,1)}% happiness."
+    )