Adding database creation with GitHub Actions

.github/workflows/database.yml

@@ -10,6 +10,7 @@ jobs:
         image: postgres
         env:
           POSTGRES_PASSWORD: postgres
+          POSTGRES_DB: big-brother
         # Set health checks to wait until postgres has started
         options: >-
           --health-cmd pg_isready
@@ -25,10 +26,10 @@ jobs:
         uses: actions/checkout@v4
 
       - name: Install dependencies
-        run: pip3 install mariadb
+        run: pip3 install psycopg2
 
       - name: Connect to PostgreSQL
         run: python3 database/db-creation.py
         env:
           POSTGRES_HOST: postgres
-          POSTGRES_PORT: 5432
+          POSTGRES_PORT: 5432

database/db-creation.py

@@ -1,35 +1,18 @@
-import os
-import mariadb
+import psycopg2
 
-env_user = os.getenv('MARIADB_USER')
-env_password = os.getenv('MARIADB_PASSWORD')
-env_database = os.getenv('MARIADB_DATABASE')
+conn = psycopg2.connect(
+   database="big-brother", user='postgres', password='postgres', host='127.0.0.1', port= '5432'
+)
+cursor = conn.cursor()
 
-def executeDBQuery(query):
-    conn = mariadb.connect(
-        user=env_user, 
-        password=env_password, 
-        host="felixmielcarek-bigbrotherdb",
-        database=env_database
-    )
-    conn.autocommit = True
-    cur = conn.cursor()
-    cur.execute(query)
-    cur.close()
-    conn.close()
+cursor.execute("DROP TABLE IF EXISTS USERS")
 
-grant_privileges_query = '''
-GRANT ALL PRIVILEGES ON {database}.* TO '{user}'@'%' IDENTIFIED BY '{password}';
-FLUSH PRIVILEGES;
-'''.format(database=env_database, user=env_user, password=env_password)
-
-create_table_query = '''
-CREATE TABLE IF NOT EXISTS users (
-    spotifyid VARCHAR(255) NOT NULL,
-    accesstoken VARCHAR(255),
-    refreshtoken VARCHAR(255),
-    PRIMARY KEY (spotifyid)
-);
-'''
-
-executeDBQuery(create_table_query)
+sql ='''CREATE TABLE USERS(
+   SPOTIFY_ID       VARCHAR(255) PRIMARY KEY,
+   ACCESS_TOKEN     VARCHAR(255),
+   REFRESH_TOKEN    VARCHAR(255)
+)'''
+cursor.execute(sql)
+print("Table created successfully........")
+conn.commit()
+conn.close()