parent
b5b1753d7d
commit
4a38ce4e7b
@ -1,35 +1,18 @@
|
|||||||
import os
|
import psycopg2
|
||||||
import mariadb
|
|
||||||
|
|
||||||
env_user = os.getenv('MARIADB_USER')
|
conn = psycopg2.connect(
|
||||||
env_password = os.getenv('MARIADB_PASSWORD')
|
database="big-brother", user='postgres', password='postgres', host='127.0.0.1', port= '5432'
|
||||||
env_database = os.getenv('MARIADB_DATABASE')
|
|
||||||
|
|
||||||
def executeDBQuery(query):
|
|
||||||
conn = mariadb.connect(
|
|
||||||
user=env_user,
|
|
||||||
password=env_password,
|
|
||||||
host="felixmielcarek-bigbrotherdb",
|
|
||||||
database=env_database
|
|
||||||
)
|
)
|
||||||
conn.autocommit = True
|
cursor = conn.cursor()
|
||||||
cur = conn.cursor()
|
|
||||||
cur.execute(query)
|
|
||||||
cur.close()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
grant_privileges_query = '''
|
cursor.execute("DROP TABLE IF EXISTS USERS")
|
||||||
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 = '''
|
sql ='''CREATE TABLE USERS(
|
||||||
CREATE TABLE IF NOT EXISTS users (
|
SPOTIFY_ID VARCHAR(255) PRIMARY KEY,
|
||||||
spotifyid VARCHAR(255) NOT NULL,
|
ACCESS_TOKEN VARCHAR(255),
|
||||||
accesstoken VARCHAR(255),
|
REFRESH_TOKEN VARCHAR(255)
|
||||||
refreshtoken VARCHAR(255),
|
)'''
|
||||||
PRIMARY KEY (spotifyid)
|
cursor.execute(sql)
|
||||||
);
|
print("Table created successfully........")
|
||||||
'''
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
executeDBQuery(create_table_query)
|
|
Loading…
Reference in new issue