From b24ea5b6101048ff67024f21afc2d7512bf3478c Mon Sep 17 00:00:00 2001
From: Kwarde <kevin@kwarde.com>
Date: Mon, 15 Jan 2024 13:22:19 +0100
Subject: [PATCH] Create wands

---
 spells.py |  2 +-
 wands.py  | 69 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 70 insertions(+), 1 deletion(-)
 create mode 100644 wands.py

diff --git a/spells.py b/spells.py
index 69207b3..ebd9506 100644
--- a/spells.py
+++ b/spells.py
@@ -92,7 +92,7 @@ spell_none = Spell(__INVALID_SPELL, 0, 0, 0, "(internal) invalid spell", SPELL_T
 ##
 ## Standalone spell functions
 ##
-def find_spell_by_name(input):
+def find_spell_by_name(input: str):
     for i in Spell.spellList:
         if input.title() == i.name.title():
             return i
diff --git a/wands.py b/wands.py
new file mode 100644
index 0000000..1f3d1a5
--- /dev/null
+++ b/wands.py
@@ -0,0 +1,69 @@
+WAND_CORE_UNICORN   = 0 #Succes +3%
+WAND_CORE_PHOENIX   = 1 #Speed  +4%
+WAND_CORE_DRAGON    = 2 #Damage +5%
+
+WAND_WOOD_ASH       = 0 #Succes -3%
+WAND_WOOD_ELDER     = 1 #Speed  -4%
+WAND_WOOD_APPLE     = 2 #Damage -5%
+
+class Wand:
+    _next_wand_id = 1
+
+    def __init__(self, wand_core: int, wand_wood: int):
+        self.id = Wand._next_wand_id
+
+        self.core = wand_core
+        self.wood = wand_wood
+
+        self.succes_rate = 1
+        self.speed = 1
+        self.damage = 1
+
+        if self.core == WAND_CORE_UNICORN:
+            self.succes_rate *= 1.03
+        elif self.core == WAND_CORE_PHOENIX:
+            self.speed *= 1.04
+        elif self.core == WAND_CORE_DRAGON:
+            self.speed *= 1.05
+
+        if self.wood == WAND_WOOD_ASH:
+            self.succes_rate *= 0.97
+        elif self.wood == WAND_WOOD_ELDER:
+            self.succes_rate *= 0.96
+        elif self.wood == WAND_WOOD_APPLE:
+            self.damage *= 0.95
+
+        Wand._next_wand_id += 1
+
+    def __repr__(self):
+        succes_rate = round(self.succes_rate * 100, 3)
+        speed = round(self.speed * 100, 3)
+        damage = round(self.dmg * 100, 3)
+        return "Modifiers: SUCCES RATE: {srate}%\tSPEED: {speed}\tDAMAGE: {dmg}".format(srate=succes_rate, speed=speed, dmg=damage)
+    
+    def get_wand_core(self):
+        if self.core == WAND_CORE_UNICORN: return "Unicorn hair"
+        elif self.core == WAND_CORE_PHOENIX: return "Phoenix feather"
+        elif self.core == WAND_CORE_DRAGON: return "Dragon heartstring"
+        else: return "Muggle's electric wire"
+    
+    def get_wand_wood(self):
+        if self.wood == WAND_WOOD_ASH: return "Ash"
+        elif self.wood == WAND_WOOD_ELDER: return "Elder"
+        elif self.wood == WAND_WOOD_APPLE: return "Apple"
+        else: return "Common wood"
+
+##
+## Wands
+##
+wand_1 = Wand(WAND_CORE_UNICORN, WAND_WOOD_ASH)
+wand_2 = Wand(WAND_CORE_UNICORN, WAND_WOOD_ELDER)
+wand_3 = Wand(WAND_CORE_UNICORN, WAND_WOOD_APPLE)
+
+wand_4 = Wand(WAND_CORE_PHOENIX, WAND_WOOD_ASH)
+wand_5 = Wand(WAND_CORE_PHOENIX, WAND_WOOD_ELDER)
+wand_6 = Wand(WAND_CORE_PHOENIX, WAND_WOOD_APPLE)
+
+wand_7 = Wand(WAND_CORE_DRAGON, WAND_WOOD_ASH)
+wand_8 = Wand(WAND_CORE_DRAGON, WAND_WOOD_ELDER)
+wand_9 = Wand(WAND_CORE_DRAGON, WAND_WOOD_APPLE)        
\ No newline at end of file