start implementation of database framework: create table
This commit is contained in:
parent
a90ff5d451
commit
9a771b0782
1 changed files with 30 additions and 0 deletions
30
database.py
Normal file
30
database.py
Normal 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")
|
Loading…
Reference in a new issue