78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
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"]],
|
|
"default": False
|
|
}
|
|
|
|
agents[agent["name"]] = agent
|
|
|
|
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"],
|
|
"default": True
|
|
}
|
|
|
|
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"],
|
|
"default": True
|
|
}
|
|
|
|
for agent in agents.values():
|
|
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()
|
|
|
|
cur.close()
|
|
conn.close() |