Hanabi/Py-Hanabi
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

ensure correct ordering of suits in variant. add some indices

Browse source
This commit is contained in:
Maximilian Keßler 2023-05-12 18:26:27 +02:00
parent 9f37572876
commit 406348f07e
Signed by: max
GPG key ID: BCC5A619923C0BA5
1 changed files with 18 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
database.py
View file

@ -77,6 +77,9 @@ def create_static_tables():
"reversed BOOLEAN NOT NULL" "reversed BOOLEAN NOT NULL"
")" ")"
) )
cur.execute(
"CREATE INDEX suits_name_idx ON suits (name)"
)
cur.execute("DROP TABLE IF EXISTS colors CASCADE;") cur.execute("DROP TABLE IF EXISTS colors CASCADE;")
cur.execute( cur.execute(
@ -85,6 +88,9 @@ def create_static_tables():
"name TEXT NOT NULL UNIQUE" "name TEXT NOT NULL UNIQUE"
")" ")"
) )
cur.execute(
"CREATE INDEX colors_name_idx ON colors (name)"
)
cur.execute("DROP TABLE IF EXISTS suit_colors CASCADE;") cur.execute("DROP TABLE IF EXISTS suit_colors CASCADE;")
cur.execute( cur.execute(
@ -92,7 +98,8 @@ def create_static_tables():
"suit_id INTEGER NOT NULL," "suit_id INTEGER NOT NULL,"
"color_id INTEGER NOT NULL," "color_id INTEGER NOT NULL,"
"FOREIGN KEY (suit_id) REFERENCES suits (id) ON DELETE CASCADE," "FOREIGN KEY (suit_id) REFERENCES suits (id) ON DELETE CASCADE,"
"FOREIGN KEY (color_id) REFERENCES colors (id) ON DELETE CASCADE" "FOREIGN KEY (color_id) REFERENCES colors (id) ON DELETE CASCADE,"
"UNIQUE (suit_id, color_id)"
")" ")"
) )
@ -118,14 +125,20 @@ def create_static_tables():
"special_rank SMALLINT" "special_rank SMALLINT"
")" ")"
) )
cur.execute(
"CREATE INDEX variants_name_idx ON variants (name)"
)
cur.execute("DROP TABLE IF EXISTS variant_suits CASCADE") cur.execute("DROP TABLE IF EXISTS variant_suits CASCADE")
cur.execute( cur.execute(
"CREATE TABLE variant_suits (" "CREATE TABLE variant_suits ("
"variant_id INT NOT NULL," "variant_id INT NOT NULL,"
"suit_id INT NOT NULL," "suit_id INT NOT NULL,"
"index SMALLINT NOT NULL,"
"FOREIGN KEY (variant_id) REFERENCES variants (id) ON DELETE CASCADE," "FOREIGN KEY (variant_id) REFERENCES variants (id) ON DELETE CASCADE,"
"FOREIGN KEY (suit_id) REFERENCES suits (id) ON DELETE CASCADE" "FOREIGN KEY (suit_id) REFERENCES suits (id) ON DELETE CASCADE,"
"UNIQUE (variant_id, suit_id),"
"UNIQUE (variant_id, index)"
")" ")"
) )
conn.commit() conn.commit()
@ -261,7 +274,7 @@ def init_static_tables():
) )
) )
for suit in suits: for index, suit in enumerate(suits):
cur.execute( cur.execute(
"SELECT id FROM suits WHERE name = %s", "SELECT id FROM suits WHERE name = %s",
(suit,) (suit,)
@ -271,8 +284,8 @@ def init_static_tables():
print(suit) print(suit)
cur.execute( cur.execute(
"INSERT INTO variant_suits (variant_id, suit_id) VALUES (%s, %s)", "INSERT INTO variant_suits (variant_id, suit_id, index) VALUES (%s, %s, %s)",
(var_id, suit_id) (var_id, suit_id, index)
) )
conn.commit() conn.commit()