iritmaximus/timetrials
1
0
Fork 0
mirror of https://github.com/iritmaximus/timetrials.git synced 2026-10-12 19:13:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix: added database reset function

Browse Source
This commit is contained in:
iritmaximus
2023-02-20 17:10:53 +02:00
parent 2c662ceff3
commit 581bc54077
1 changed files with 29 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
flaskr/database.py
+29 -26
View File
@@ -10,42 +10,45 @@ db = SQLAlchemy(app)
def create_db():
print("Initializing the db...")
sqllist = [
"CREATE TABLE IF NOT EXISTS users (id int primary key, name varchar(80), password varchar(80));",
"CREATE TABLE IF NOT EXISTS games (id int primary key, name varchar(80));",
"CREATE TABLE IF NOT EXISTS courses (id int primary key, name varchar(80), game_id int references games(id));",
"CREATE TABLE IF NOT EXISTS standards (id int primary key, name varchar(50), tier varchar(3), points numeric, game_id int references games(id), course_id int references courses(id));",
"CREATE TABLE IF NOT EXISTS times (id int primary key, game_id int references games(id), course_id int references courses(id), timems varchar(80), date date);",
]
with app.app_context():
for sql in sqllist:
db.session.execute(sql)
db.session.commit()
db.session.execute(open("sql/schema.sql", "r").read())
db.session.execute(open("sql/games.sql", "r").read())
db.session.execute(open("sql/cups.sql", "r").read())
db.session.execute(open("sql/courses.sql", "r").read())
db.session.commit()
print("Db initialization completed.")
return "Success"
def reset_db():
print("Resetting the db...")
db.session.execute(open("sql/resettables.sql", "r").read())
db.session.execute(open("sql/schema.sql", "r").read())
db.session.execute(open("sql/games.sql", "r").read())
db.session.execute(open("sql/cups.sql", "r").read())
db.session.execute(open("sql/courses.sql", "r").read())
db.session.commit()
def test_db():
print("TESTING THE DB ***********************")
print("*********************** TESTING THE DB ***********************")
with app.app_context():
sql = "DROP TABLE IF EXISTS test;"
db.session.execute(sql)
db.session.commit()
sql = "DROP TABLE IF EXISTS test;"
db.session.execute(sql)
db.session.commit()
sql = "CREATE TABLE IF NOT EXISTS test (id int primary key, greeting text);"
db.session.execute(sql)
db.session.commit()
sql = "CREATE TABLE IF NOT EXISTS test (id int primary key, greeting text);"
db.session.execute(sql)
db.session.commit()
for x in range(10):
sql = f"INSERT INTO test (id, greeting) VALUES ({randint(1,100)}, 'hello');"
db.session.execute(sql)
db.session.commit()
result = db.session.execute("SELECT * FROM test;").fetchall()
print(result)
db.session.commit()
result = db.session.execute("SELECT * FROM test;").fetchall()
print(result)
return "Working lol"