111 lines
2.9 KiB
Python
111 lines
2.9 KiB
Python
from flask import Flask, render_template, request, redirect, url_for
|
|
import sqlite3
|
|
|
|
app = Flask(__name__)
|
|
DB_PATH = "tarkov.db"
|
|
|
|
|
|
def get_db():
|
|
conn = sqlite3.connect(DB_PATH)
|
|
conn.row_factory = sqlite3.Row
|
|
return conn
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
conn = get_db()
|
|
maps = conn.execute("""
|
|
SELECT id, name
|
|
FROM maps
|
|
ORDER BY name
|
|
""").fetchall()
|
|
map_filter = request.args.get("map_id", type=int)
|
|
key_map_rows = conn.execute("""
|
|
SELECT key_id, map_id
|
|
FROM key_maps
|
|
""").fetchall()
|
|
key_maps = {}
|
|
for row in key_map_rows:
|
|
key_maps.setdefault(row["key_id"], set()).add(row["map_id"])
|
|
|
|
key_query = """
|
|
SELECT
|
|
k.id,
|
|
k.name,
|
|
k.icon_url,
|
|
k.grid_image_url,
|
|
k.wiki_url,
|
|
r.priority,
|
|
r.reason,
|
|
COALESCE(r.used_in_quest, 0) AS used_in_quest
|
|
FROM keys k
|
|
"""
|
|
params = []
|
|
if map_filter:
|
|
key_query += """
|
|
JOIN key_maps kmf
|
|
ON k.id = kmf.key_id
|
|
AND kmf.map_id = ?
|
|
"""
|
|
params.append(map_filter)
|
|
key_query += """
|
|
LEFT JOIN key_ratings r ON k.id = r.key_id
|
|
ORDER BY
|
|
CASE WHEN r.priority IS NULL THEN 1 ELSE 0 END,
|
|
r.priority DESC,
|
|
k.name
|
|
"""
|
|
keys = conn.execute(key_query, params).fetchall()
|
|
conn.close()
|
|
|
|
key_maps = {k: sorted(v) for k, v in key_maps.items()}
|
|
return render_template(
|
|
"index.html",
|
|
keys=keys,
|
|
maps=maps,
|
|
key_maps=key_maps,
|
|
map_filter=map_filter,
|
|
)
|
|
|
|
|
|
@app.route("/rate", methods=["POST"])
|
|
def rate_key():
|
|
key_id = request.form["key_id"]
|
|
priority = request.form["priority"]
|
|
reason = request.form.get("reason", "")
|
|
used_in_quest = 1 if request.form.get("used_in_quest") == "on" else 0
|
|
map_filter = request.form.get("map_id")
|
|
map_ids = []
|
|
for value in request.form.getlist("map_ids"):
|
|
try:
|
|
map_ids.append(int(value))
|
|
except ValueError:
|
|
continue
|
|
|
|
conn = get_db()
|
|
conn.execute("""
|
|
INSERT INTO key_ratings (key_id, priority, reason, used_in_quest)
|
|
VALUES (?, ?, ?, ?)
|
|
ON CONFLICT(key_id) DO UPDATE SET
|
|
priority = excluded.priority,
|
|
reason = excluded.reason,
|
|
used_in_quest = excluded.used_in_quest,
|
|
updated_at = CURRENT_TIMESTAMP
|
|
""", (key_id, priority, reason, used_in_quest))
|
|
conn.execute("DELETE FROM key_maps WHERE key_id = ?", (key_id,))
|
|
if map_ids:
|
|
conn.executemany(
|
|
"INSERT OR IGNORE INTO key_maps (key_id, map_id) VALUES (?, ?)",
|
|
[(key_id, map_id) for map_id in map_ids],
|
|
)
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
if map_filter:
|
|
return redirect(url_for("index", map_id=map_filter))
|
|
return redirect(url_for("index"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port=5000, debug=True)
|