diff --git a/agents.py b/agents.py index 4639a9d..cb0b5ae 100644 --- a/agents.py +++ b/agents.py @@ -43,31 +43,34 @@ for raw in content: "def_index": raw["def_index"], "team": team, "model_player": raw["model_player"], - "rarity": rarity[raw["rarity"]["name"]] + "rarity": rarity[raw["rarity"]["name"]], + "default": False } agents[agent["name"]] = agent -agent["agent-0"] = { +agents["agent-0"] = { "name": "agent-0", "display": "Default CT Agent", "def_index": 0, "team": 0, "model_player": "agents/models/ctm_sas/ctm_sas.vmdl", - "rarity": rarity["Stock"] + "rarity": rarity["Stock"], + "default": True } -agent["agent-1"] = { +agents["agent-1"] = { "name": "agent-1", "display": "Default T Agent", "def_index": 0, "team": 1, "model_player": "agents/models/tm_phoenix/tm_phoenix.vmdl", - "rarity": rarity["Stock"] + "rarity": rarity["Stock"], + "default": True } 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"])) + cur.execute("INSERT INTO agents (name, display, def_index, team, model_player, rarity, \"default\") VALUES (%s, %s, %s, %s, %s, %s, %s)", (agent["name"], agent["display"], agent["def_index"], agent["team"], agent["model_player"], agent["rarity"], agent["default"])) conn.commit()