Add some database import export functions

This commit is contained in:
2023-11-03 16:51:20 +01:00
parent fa485472c8
commit fbe4e8cbee

36
database_utils.py Normal file
View File

@ -0,0 +1,36 @@
import csv
from model import *
def load_csv(filename):
with open("data/" + filename + ".csv", newline="") as file:
reader = csv.reader(file, delimiter=",")
next(reader, None) # skip header
return list(reader)
# @todo CSV-Writer
def write_csv(filename):
with open("data/" + filename + ".csv", newline="") as file:
writer = csv.writer(file, delimiter=",")
# @todo Complete CSV Files
# @todo Reload Static Data
# Reload static database data, this has to be called from the app context
def reload_static_data(db):
print("Initializing DataBase with Static Values...")
# Create it (if it doesn't exist!)
db.create_all()
# Clear static data
Team.query.delete()
Driver.query.delete()
# Reload static data
for row in load_csv("teams"):
db.session.add(Team().from_csv(row))
for row in load_csv("drivers"):
db.session.add(Driver().from_csv(row))
db.session.commit()
# @todo Export Dynamic Data
def export_dynamic_data():
pass