From e7f30ecea4c3a893986854375f024ff7f16ad501 Mon Sep 17 00:00:00 2001
From: AINDUSTRIES <contact.aindustries@gmail.com>
Date: Thu, 3 Sep 2026 02:36:21 +0200
Subject: [PATCH] Added agents

---
 agents.py | 57 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 agents.py

diff --git a/agents.py b/agents.py
new file mode 100644
index 0000000..2c0ae6e
--- /dev/null
+++ b/agents.py
@@ -0,0 +1,57 @@
+import psycopg2, json
+
+"""
+CREATE TABLE IF NOT EXISTS agents (
+    id uuid DEFAULT gen_random_uuid() PRIMARY KEY,
+    rarity uuid REFERENCES rarity(id) NOT NULL,
+    name varchar NOT NULL,
+    display varchar NOT NULL,
+    def_index integer NOT NULL,
+    team integer NOT NULL, -- 0 = CT, 1 = T, 2 = BOTH
+    model_player varchar NOT NULl
+);
+"""
+
+conn = psycopg2.connect("dbname=porning user=aindustries")
+cur = conn.cursor()
+
+cur.execute("SELECT * FROM rarity")
+raw_rarity = cur.fetchall()
+rarity = {rarity[1]: rarity[0] for rarity in raw_rarity}
+
+with open("agents.json", "r") as f:
+    content = json.load(f)
+
+agents = {}
+
+for raw in content:
+    if raw["id"] in agents:
+        continue
+
+    team = raw["team"]["id"]
+
+    if team == "counter-terrorists":
+        team = 0
+    elif team == "terrorists":
+        team = 1
+    else:
+        team = 2
+
+    agent = {
+        "name": raw["id"],
+        "display": raw["name"],
+        "def_index": raw["def_index"],
+        "team": team,
+        "model_player": raw["model_player"],
+        "rarity": rarity[raw["rarity"]["name"]]
+    }
+
+    agents[agent["name"]] = agent
+
+for agent in agents.values():
+    cur.execute("INSERT INTO agents (name, display, def_index, team, model_player, rarity) VALUES (%s, %s, %s, %s, %s, %s)", (agent["name"], agent["display"], agent["def_index"], agent["team"], agent["model_player"], agent["rarity"]))
+
+conn.commit()
+
+cur.close()
+conn.close()
\ No newline at end of file