module 7 ex3 WIP

07/ex3/FantasyCardFactory.py

@@ -0,0 +1,46 @@
+from .CardFactory import CardFactory
+from random import choice
+from ex0 import Card, CreatureCard
+from ex1 import SpellCard, ArtifactCard
+
+creature_cards = [
+    CreatureCard("Fire Dragon", 20, "Rare", 7, 20),
+    CreatureCard("Goblin Warrior", 12, "Common", 4, 10),
+]
+
+spell_cards = [
+    SpellCard("Fire Ball", 5, "Rare", "Decrase health by 1 for 3 round"),
+    SpellCard("Ice Spike", 3, "Common", "Reduce damage by 2 for 1 round"),
+    SpellCard(
+        "Lightning bolt", 10, "Legendary", "Card can't play for 3 round"
+    ),
+]
+
+artifact_cards = [
+    ArtifactCard(
+        "Mana Ring", 2, "Common", 3, "Increase mana by 1 for each round"
+    ),
+    ArtifactCard("Witch Staff", 6, "Legendary", 5, ""),
+]
+
+
+class FantasyCardFactory(CardFactory):
+    def create_creature(self, name_or_power: str | int | None = None) -> Card:
+        card = choice(creature_cards)
+        if isinstance(name_or_power, str):
+            card.name = name_or_power
+        elif isinstance(name_or_power, int):
+            card.attack = name_or_power
+        return card
+
+    def create_spell(self, name_or_power: str | int | None = None) -> Card:
+        return super().create_spell(name_or_power)
+
+    def create_artifact(self, name_or_power: str | int | None = None) -> Card:
+        return super().create_artifact(name_or_power)
+
+    def create_themed_deck(self, size: int) -> dict:
+        return super().create_themed_deck(size)
+
+    def get_supported_types(self) -> dict:
+        return super().get_supported_types()