rankendpoint.py

@@ -1,33 +1,18 @@
-# This script serves an HTTP endpoint that provides the racing scores.
-#
-# Example invocation for testing:
-# $ RANKENDPOINT_DBPATH=/path/to/database.db RANKENDPOINT_ROUTE=/getranks flask --app rankendpoint.py run
-#
-# Example invocation in production (behind a properly configured gateway like nginx):
-# $ RANKENDPOINT_DBPATH=/path/to/database.db RANKENDPOINT_ROUTE=/getranks uwsgi \
-#     -s localhost:3031 --manage-script-name --mount /=rankendpoint:app --plugin python3
-
 from flask import Flask, request
 app = Flask(__name__)
 
 import sqlite3
 import sys
-import os
 
 header = "SUCCESS"
 
-db_path = os.environ.get('RANKENDPOINT_DBPATH')
-route = os.environ.get('RANKENDPOINT_ROUTE')
-
-if None in (db_path, route):
-    sys.exit('must set RANKENDPOINT_DBPATH and RANKENDPOINT_ROUTE environment variables')
-
-# Opens database in read-only mode
-# Checking same thread disabled for now, which is fine since we never modify anything
-try:
+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:
+    except Exception as ex:
         print(ex)
         sys.exit()
 
@@ -100,8 +85,7 @@ def get_score_entries(data, name):
 
     return scores
 
-# route should be something like /getranks
-@app.route(f'{route}', methods=['POST'])
+@app.route('/getranks', methods=['POST'])
 def rankings():
     #print("PCUID:", request.form['PCUID'])
     #print("EP_ID:", request.form['EP_ID'])
@@ -144,3 +128,9 @@ def rankings():
     # 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])
+