christoph/flask-formula10
1
Fork 0
Code Issues 14 Actions Packages Activity

Implement dynamic data export

Browse Source
This commit is contained in:
Christoph Urlacher
2024-02-14 15:04:19 +01:00
parent 45be605094
commit 1dc995663e
3 changed files with 135 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
database_utils.py
View File

@ -3,21 +3,24 @@ from model import *
def load_csv(filename):
with open("init_data/" + filename + ".csv", newline="") as file:
with open("init_data/" + filename + ".csv", "r", 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("dynamic_data/" + filename + ".csv", newline="") as file:
def write_csv(filename, objects):
with open("dynamic_data/" + filename + ".csv", "w", newline="") as file:
writer = csv.writer(file, delimiter=",")
writer.writerow(objects[0].__csv_header__)
for obj in objects:
writer.writerow(obj.to_csv())
# Reload static database data, this has to be called from the app context
def reload_static_data(db):
print("Initializing DataBase with Static Values...")
print("Initializing Database with Static Values...")
# Create it (if it doesn't exist!)
db.create_all()
@ -40,6 +43,16 @@ def reload_static_data(db):
db.session.commit()
# @todo Export Dynamic Data
def export_dynamic_data():
pass
print("Exporting Userdata...")
raceresults = RaceResult.query.all()
raceguesses = RaceGuess.query.all()
seasonguesses = SeasonGuess.query.all()
if len(raceresults) > 0:
write_csv("raceresults", raceresults)
if len(raceguesses) > 0:
write_csv("raceguesses", raceguesses)
if len(seasonguesses) > 0:
write_csv("seasonguesses", seasonguesses)

6
formula10.py
View File

@ -22,6 +22,12 @@ def reload():
return redirect("/")
@app.route("/save")
def save():
export_dynamic_data()
return redirect("/")
@app.route("/race")
def race():
users = User.query.all()

116
model.py
View File

@ -6,7 +6,6 @@ from datetime import datetime
db = SQLAlchemy()
# This table contains users that can guess
class User(db.Model):
__tablename__ = "user"
@ -17,7 +16,6 @@ class User(db.Model):
name: Mapped[str] = mapped_column(String(32), primary_key=True)
# This table contains manifestations of GPs, with dates
class Race(db.Model):
__tablename__ = "race"
@ -44,7 +42,6 @@ class Team(db.Model):
name: Mapped[str] = mapped_column(String(32), primary_key=True)
# This table contains drivers and their team associations, e.g. Max Verschtappen
class Driver(db.Model):
__tablename__ = "driver"
@ -66,6 +63,50 @@ class Driver(db.Model):
class RaceResult(db.Model):
__tablename__ = "raceresult"
__csv_header__ = [
"id",
"race_id",
"p01_id",
"p02_id",
"p03_id",
"p04_id",
"p05_id",
"p06_id",
"p07_id",
"p08_id",
"p09_id",
"p10_id",
"p11_id",
"p12_id",
"p13_id",
"p14_id",
"p15_id",
"p16_id",
"p17_id",
"p18_id",
"p19_id",
"p20_id",
"dnf01_id",
"dnf02_id",
"dnf03_id",
"dnf04_id",
"dnf05_id",
"dnf06_id",
"dnf07_id",
"dnf08_id",
"dnf09_id",
"dnf10_id",
"dnf11_id",
"dnf12_id",
"dnf13_id",
"dnf14_id",
"dnf15_id",
"dnf16_id",
"dnf17_id",
"dnf18_id",
"dnf19_id",
"dnf20_id"
]
def from_csv(self, row):
self.id = int(row[0])
@ -112,6 +153,51 @@ class RaceResult(db.Model):
self.dnf20_id = str(row[41])
return self
def to_csv(self):
return [
self.id,
self.p01_id,
self.p02_id,
self.p03_id,
self.p04_id,
self.p05_id,
self.p06_id,
self.p07_id,
self.p08_id,
self.p09_id,
self.p10_id,
self.p11_id,
self.p12_id,
self.p13_id,
self.p14_id,
self.p15_id,
self.p16_id,
self.p17_id,
self.p18_id,
self.p19_id,
self.p20_id,
self.dnf01_id,
self.dnf02_id,
self.dnf03_id,
self.dnf04_id,
self.dnf05_id,
self.dnf06_id,
self.dnf07_id,
self.dnf08_id,
self.dnf09_id,
self.dnf10_id,
self.dnf11_id,
self.dnf12_id,
self.dnf13_id,
self.dnf14_id,
self.dnf15_id,
self.dnf16_id,
self.dnf17_id,
self.dnf18_id,
self.dnf19_id,
self.dnf20_id
]
id: Mapped[int] = mapped_column(Integer, primary_key=True)
race_id: Mapped[str] = mapped_column(ForeignKey("race.id"))
p01_id: Mapped[str] = mapped_column(ForeignKey("driver.name"))
@ -207,34 +293,52 @@ class RaceResult(db.Model):
# This table contains race guesses made by users
class RaceGuess(db.Model):
__tablename__ = "raceguess"
__csv_header__ = ["id", "user_id", "race_id", "pxx_id", "dnf_id"]
def from_csv(self, row):
self.id = int(row[0])
self.user_id = str(row[1])
self.race_id = str(row[2])
self.pXX_id = str(row[3])
self.pxx_id = str(row[3])
self.dnf_id = str(row[4])
return self
def to_csv(self, writer):
return [
self.id,
self.user_id,
self.race_id,
self.pxx_id,
self.dnf_id
]
id: Mapped[int] = mapped_column(Integer, primary_key=True)
user_id: Mapped[str] = mapped_column(ForeignKey("user.name"))
race_id: Mapped[str] = mapped_column(ForeignKey("race.id"))
pXX_id: Mapped[str] = mapped_column(ForeignKey("driver.name"))
pxx_id: Mapped[str] = mapped_column(ForeignKey("driver.name"))
dnf_id: Mapped[str] = mapped_column(ForeignKey("driver.name"))
# Relationships
user: Mapped["User"] = relationship("User", foreign_keys=[user_id])
race: Mapped["Race"] = relationship("Race", foreign_keys=[race_id])
pXX: Mapped["Driver"] = relationship("Driver", foreign_keys=[pXX_id])
pxx: Mapped["Driver"] = relationship("Driver", foreign_keys=[pxx_id])
dnf: Mapped["Driver"] = relationship("Driver", foreign_keys=[dnf_id])
class SeasonGuess(db.Model):
__tablename__ = "seasonguess"
__csv_header__ = ["id", "user_id", "hot_take"]
def from_csv(self, row):
return self
def to_csv(self, writer):
return [
self.id,
self.user_id,
self.hot_take
]
id: Mapped[int] = mapped_column(Integer, primary_key=True)
user_id: Mapped[str] = mapped_column(ForeignKey("user.name"))
hot_take: Mapped[str] = mapped_column(String(512))