-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathweb.py
45 lines (42 loc) · 1.63 KB
/
web.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
from flask import Flask, render_template, request, jsonify, make_response
import os.path, os
import models
from bot import db
from sqlalchemy.orm import scoped_session
session = scoped_session(db.session_fact)
from flask.helpers import send_from_directory
app = Flask(__name__)
class Mii:
def __init__(self, did, file):
self.did = did
self.file = file
def get_discord_id(self):
return self.did
def get_file(self):
return self.file
def get_full_path(self):
return os.path.join("miis", self.file)
@app.route('/')
def home():
return render_template("home.html.jinja2")
@app.route("/load")
def load_miis():
miis = [{"loc": f, "username": session.query(models.CachedUser).get(f.split(".")[0]).username + "#" + session.query(models.CachedUser).get(f.split(".")[0]).discriminator} for f in os.listdir("miis") if os.path.isfile(os.path.join("miis", f))]
if request.args:
page = int(request.args.get("page"))
if page == 0:
return {"miis": miis[0:10]}
elif page == len(miis):
return make_response(jsonify({"miis": []})) # I decided to pick this code because I had no clue about anything in the HTTP spec that would make sense.
else:
return {"miis": miis[page: page + 10]}
else:
return make_response({}, 400)
if __name__ == "__main__":
# In production, these are served by a reverse proxy.
app.static_folder = "static"
app.static_url_path = "/static"
@app.route("/miis/<file>")
def miis(file):
return send_from_directory("miis", file)
app.run(debug=True)