breeding data fixed

static/data/remove.py

@@ -0,0 +1,75 @@
+import xml.etree.ElementTree as ET
+import sqlite3
+
+DATABASE = "monsters.db"
+
+
+def connect_db():
+    return sqlite3.connect(DATABASE)
+
+
+def create_tables():
+    with connect_db() as conn:
+        cursor = conn.cursor()
+        cursor.execute(
+            """
+            CREATE TABLE IF NOT EXISTS breeds (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                target TEXT NOT NULL
+            );
+        """
+        )
+        cursor.execute(
+            """
+            CREATE TABLE IF NOT EXISTS breed_requirements (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                breed_id INTEGER NOT NULL,
+                requirement_type TEXT NOT NULL,
+                requirement_value TEXT NOT NULL,
+                FOREIGN KEY (breed_id) REFERENCES breeds (id)
+            );
+        """
+        )
+        conn.commit()
+
+
+def insert_breeding_data(xml_file):
+    tree = ET.parse(xml_file)
+    root = tree.getroot()
+
+    with connect_db() as conn:
+        cursor = conn.cursor()
+
+        for breed_element in root.findall(".//breeds/breed"):
+            target = breed_element.get("target")
+
+            cursor.execute("INSERT INTO breeds (target) VALUES (?)", (target,))
+            breed_id = cursor.lastrowid
+
+            base_requirements = breed_element.find("./base")
+            insert_breed_requirements(cursor, breed_id, base_requirements, "base")
+
+            mate_requirements = breed_element.find("./mate")
+            insert_breed_requirements(cursor, breed_id, mate_requirements, "mate")
+
+        conn.commit()
+
+
+def insert_breed_requirements(cursor, breed_id, requirements_element, requirement_type):
+    for requirement_element in requirements_element.findall("./breed-requirement"):
+        family = requirement_element.get("family")
+        monster = requirement_element.get("monster")
+
+        cursor.execute(
+            """
+            INSERT INTO breed_requirements (breed_id, requirement_type, requirement_value)
+            VALUES (?, ?, ?)
+        """,
+            (breed_id, requirement_type, family or monster),
+        )
+
+
+if __name__ == "__main__":
+    create_tables()
+    insert_breeding_data("data.xml")
+    print("Breeding data inserted successfully.")