Add ranked endpoint server

rankendpoint.py

@@ -0,0 +1,129 @@
+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])
+