mirror of
https://github.com/OpenFusionProject/scripts.git
synced 2024-11-14 19:20:05 +00:00
130 lines
4.0 KiB
Python
130 lines
4.0 KiB
Python
from flask import Flask, request
|
|
app = Flask(__name__)
|
|
|
|
import sqlite3
|
|
import sys
|
|
|
|
header = "SUCCESS"
|
|
|
|
def main(db_path):
|
|
# Opens database in read-only mode
|
|
# Checking same thread disabled for now, which is fine since we never modify anything
|
|
try:
|
|
db = sqlite3.connect('file:{}?mode=ro'.format(db_path), uri=True, check_same_thread=False)
|
|
cur = db.cursor()
|
|
except Exception as ex:
|
|
print(ex)
|
|
sys.exit()
|
|
|
|
#db.set_trace_callback(print)
|
|
|
|
def fetch_ranks(epid, date, num):
|
|
sql = """
|
|
SELECT * FROM (
|
|
SELECT RaceResults.PlayerID,
|
|
Players.FirstName,
|
|
Players.LastName,
|
|
RaceResults.Score
|
|
FROM RaceResults
|
|
INNER JOIN Players ON RaceResults.PlayerID=Players.PlayerID
|
|
WHERE EPID=? AND
|
|
DATETIME(Timestamp,'unixepoch') > (SELECT DATETIME('now', ?))
|
|
ORDER BY Score DESC
|
|
)
|
|
GROUP BY PlayerID
|
|
ORDER BY Score DESC
|
|
"""
|
|
|
|
if num > -1:
|
|
sql += "LIMIT ?"
|
|
args = (epid, date, num)
|
|
else:
|
|
args = (epid, date)
|
|
|
|
cur = db.execute(sql + ";", args)
|
|
rows = cur.fetchall()
|
|
|
|
return rows
|
|
|
|
def fetch_my_ranks(pcuid, epid, date):
|
|
sql = """
|
|
SELECT RaceResults.PlayerID,
|
|
Players.FirstName,
|
|
Players.LastName,
|
|
RaceResults.Score
|
|
FROM RaceResults
|
|
INNER JOIN Players ON RaceResults.PlayerID=Players.PlayerID
|
|
WHERE RaceResults.PlayerID=? AND EPID=? AND
|
|
DATETIME(Timestamp,'unixepoch') > (SELECT DATETIME('now', ?))
|
|
ORDER BY Score DESC LIMIT 1;
|
|
"""
|
|
|
|
args = (pcuid, epid, date)
|
|
cur = db.execute(sql, args)
|
|
rows = cur.fetchall()
|
|
|
|
return rows
|
|
|
|
def get_score_entries(data, name):
|
|
# Uncomment if you want placeholders in top 10 ranks ala Retro
|
|
#if not name.startswith("my"):
|
|
# while len(data) < 10:
|
|
# data.append(((999, 'hehe', 'dong', 1)))
|
|
|
|
scores="<{}>\n".format(name)
|
|
for rank, item in enumerate(data, start=1):
|
|
scores+='\t<score>PCUID="{}" Score="{}" Rank="{}" FirstName="{}" LastName="{}"</score>\n'.format(item[0], item[3], rank, item[1], item[2])
|
|
scores+="</{}>\n".format(name)
|
|
|
|
return scores
|
|
|
|
@app.route('/getranks', methods=['POST'])
|
|
def rankings():
|
|
#print("PCUID:", request.form['PCUID'])
|
|
#print("EP_ID:", request.form['EP_ID'])
|
|
|
|
# Input Validation
|
|
try:
|
|
pcuid = int(request.form['PCUID'])
|
|
epid = int(request.form['EP_ID'])
|
|
num = 10 if 'NUM' not in request.form else int(request.form['NUM'])
|
|
except ValueError as verr:
|
|
return "Request param does not convert to int", 400
|
|
except Exception as ex:
|
|
return "Error converting request param to int", 500
|
|
|
|
# EP_ID must be between 1 and 33. also, ep #6 doesn't exist
|
|
if not (1 <= epid <= 33) or (epid == 6):
|
|
return "Invalid EP_ID", 400
|
|
|
|
# Get everything we need from the DB...
|
|
myday = fetch_my_ranks(pcuid, epid, '-1 day')
|
|
day = fetch_ranks(epid, '-1 day', num)
|
|
myweek = fetch_my_ranks(pcuid, epid, '-7 day')
|
|
week = fetch_ranks(epid, '-7 day', num)
|
|
mymonth = fetch_my_ranks(pcuid, epid, '-1 month')
|
|
month = fetch_ranks(epid, '-1 month', num)
|
|
myalltime = fetch_my_ranks(pcuid, epid, '-999 year')
|
|
alltime = fetch_ranks(epid, '-999 year', num)
|
|
|
|
# Slap that all into an "xml"...
|
|
xmlbody = ""
|
|
xmlbody += get_score_entries(myday, "myday")
|
|
xmlbody += get_score_entries(day, "day")
|
|
xmlbody += get_score_entries(myweek, "myweek")
|
|
xmlbody += get_score_entries(week, "week")
|
|
xmlbody += get_score_entries(mymonth, "mymonth")
|
|
xmlbody += get_score_entries(month, "month")
|
|
xmlbody += get_score_entries(myalltime, "myalltime")
|
|
xmlbody += get_score_entries(alltime, "alltime")
|
|
|
|
# and send it off!
|
|
return header + xmlbody
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 2:
|
|
print("Usage: {} <db file>".format(sys.argv[0]))
|
|
else:
|
|
main(sys.argv[1])
|
|
|