mirror of
https://github.com/FAUSheppy/skillbird
synced 2025-12-06 06:51:34 +01:00
add db importer
This commit is contained in:
57
helper_scripts/import_rounds_from_db.py
Executable file
57
helper_scripts/import_rounds_from_db.py
Executable file
@@ -0,0 +1,57 @@
|
||||
#!/usr/bin/python3
|
||||
|
||||
import sqlite3
|
||||
import datetime
|
||||
import json
|
||||
import requests
|
||||
|
||||
url = "http://127.0.0.1:5000/submitt-round"
|
||||
db = "rounds.sqlite"
|
||||
|
||||
if __name__ == "__main__":
|
||||
conn = sqlite3.connect(db)
|
||||
cursor = conn.cursor()
|
||||
|
||||
# status display #
|
||||
start = datetime.datetime.now()
|
||||
cursor.execute("select count(*) from rounds")
|
||||
count = cursor.fetchone()[0]
|
||||
i = 1
|
||||
|
||||
cursor.execute("select * from rounds")
|
||||
for row in cursor:
|
||||
|
||||
# load from db
|
||||
timestamp, winners, losers, winnerSide, mapName, duration, prediction, confidence = row
|
||||
startTime = datetime.datetime.fromtimestamp(int(float(timestamp)))
|
||||
winnersParsed = json.loads(winners)
|
||||
losersParsed = json.loads(losers)
|
||||
|
||||
# fix id/playerId db field name for now #
|
||||
for pElement in winnersParsed:
|
||||
pElement["playerId"] = pElement.pop("id")
|
||||
pElement["playerName"] = pElement.pop("name")
|
||||
pElement["activeTime"] = pElement.pop("active_time")
|
||||
for pElement in losersParsed:
|
||||
pElement["playerId"] = pElement.pop("id")
|
||||
pElement["playerName"] = pElement.pop("name")
|
||||
pElement["activeTime"] = pElement.pop("active_time")
|
||||
|
||||
data = { "map" : mapName,
|
||||
"winner-side" : winnerSide,
|
||||
"winners" : winnersParsed,
|
||||
"losers" : losersParsed,
|
||||
"duration" : duration,
|
||||
"startTime" : startTime.isoformat()}
|
||||
|
||||
# submit
|
||||
requests.post(url, json=data)
|
||||
|
||||
# status update #
|
||||
cur = datetime.datetime.now()
|
||||
percent = int(i/count*100);
|
||||
elapsed = str(cur-start).split(".")[0]
|
||||
estRem = str((cur-start)/i*count).split(".")[0]
|
||||
i += 1
|
||||
print("Round: {} ({}%) - elapsed: {}, estimated remaining: {}\r".format(
|
||||
i, percent, elapsed, estRem), end="")
|
||||
Reference in New Issue
Block a user