start implementation of database framework: create table

This commit is contained in:
Maximilian Keßler 2023-03-02 11:52:15 +01:00
parent a90ff5d451
commit 9a771b0782
Signed by: max
GPG Key ID: BCC5A619923C0BA5
1 changed files with 30 additions and 0 deletions

30
database.py Normal file
View File

@ -0,0 +1,30 @@
import psycopg2
## global connection
conn = psycopg2.connect("dbname=hanab-live user=postgres")
## cursor
cur = conn.cursor()
## check if table exists, else create it
cur.execute("SELECT EXISTS (SELECT FROM pg_tables WHERE schemaname = 'public' AND tablename = 'games');")
a = cur.fetchone()
if a[0] is False:
print("creating table")
cur.execute(
"CREATE TABLE games ("
"id SERIAL PRIMARY KEY,"
"num_players SMALLINT NOT NULL,"
"seed TEXT NOT NULL,"
"score SMALLINT NOT NULL,"
"variant_id SMALLINT NOT NULL,"
"deck_plays BOOLEAN,"
"one_extra_card BOOLEAN,"
"one_less_card BOOLEAN,"
"all_or_nothing BOOLEAN,"
"num_turns SMALLINT"
");")
conn.commit()
else:
print("table already exists")