mirror of
https://github.com/FAUSheppy/tmnf-replay-server.git
synced 2025-12-07 15:41:36 +01:00
Compare commits
3 Commits
e54f70eaac
...
notificati
| Author | SHA1 | Date | |
|---|---|---|---|
| ccc3122f3c | |||
| 8433e0f47e | |||
| a2ee6065c1 |
2
.github/workflows/main.yaml
vendored
2
.github/workflows/main.yaml
vendored
@@ -4,8 +4,6 @@ on:
|
||||
push:
|
||||
branches:
|
||||
- "master"
|
||||
schedule:
|
||||
- cron: "0 2 * * 0"
|
||||
|
||||
jobs:
|
||||
docker:
|
||||
|
||||
@@ -10,22 +10,15 @@ def send_notification(app, target_user, mapname, old_replay, new_replay):
|
||||
return
|
||||
|
||||
# send to event dispatcher #
|
||||
message = "TM: Record broken on {}\n\n".format(mapname)
|
||||
message += "Old time: {}\n".format(old_replay.get_human_readable_time())
|
||||
message += "New time: {}\n".format(new_replay.get_human_readable_time())
|
||||
message += "\nby {}".format(new_replay.clean_login())
|
||||
message = "Trackmania: Record broken on {}".format(mapname)
|
||||
message += "Old time: {}".format(old_replay.get_human_readable_time())
|
||||
message += "New time: {}".format(new_replay.get_human_readable_time())
|
||||
message += "by {}".format(new_replay.clean_login())
|
||||
|
||||
payload = {
|
||||
"users": [target_user],
|
||||
"msg" : message,
|
||||
"method" : "any"
|
||||
}
|
||||
payload = { "users": [user], "msg" : message }
|
||||
|
||||
url_and_token = "/smart-send?dispatch-access-token={}".format(app.config["DISPATCH_TOKEN"])
|
||||
r = requests.post(app.config["DISPATCH_SERVER"] + url_and_token, json=payload)
|
||||
r = requests.post(app.config["DISPATCH_SERVER"] + "/smart-send",
|
||||
json=payload, auth=app.config["DISPATCH_AUTH"])
|
||||
|
||||
if not r.ok:
|
||||
msg = "Error handing off notification to dispatch ({} {})".format(r.status_code, r.content)
|
||||
print(msg, file=sys.stderr)
|
||||
else:
|
||||
print("Handed off notification for {} to dispatch".format(target_user), file=sys.stderr)
|
||||
print("Error handing off notification to dispatch ({})".format(r.status_code), file=sys.stderr)
|
||||
|
||||
138
server.py
138
server.py
@@ -10,7 +10,6 @@ import datetime
|
||||
|
||||
from pygbx import Gbx, GbxType
|
||||
import tm2020parser
|
||||
import notifications
|
||||
|
||||
import sqlalchemy
|
||||
from sqlalchemy import Column, Integer, String, Boolean, or_, and_, asc, desc
|
||||
@@ -22,27 +21,6 @@ app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
|
||||
app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get("SQLITE_LOCATION") or "sqlite:///sqlite.db"
|
||||
db = SQLAlchemy(app)
|
||||
|
||||
SEASON_ORDERING = ["Winter", "Spring", "Summer", "Fall"]
|
||||
def filter_for_current_season(maps):
|
||||
|
||||
maps = list(maps)
|
||||
|
||||
year = str(datetime.datetime.now().year)
|
||||
|
||||
season = 0
|
||||
for m in maps:
|
||||
|
||||
if year not in m.mapname:
|
||||
continue
|
||||
else:
|
||||
for i, season_name in enumerate(SEASON_ORDERING):
|
||||
if season_name in m.mapname and i > season:
|
||||
season = i
|
||||
|
||||
filter_func = lambda m: SEASON_ORDERING[season] in m.mapname and year in m.mapname
|
||||
maps_season = list(filter(filter_func, maps))
|
||||
return maps_season
|
||||
|
||||
class Map(db.Model):
|
||||
|
||||
__tablename__ = "maps"
|
||||
@@ -109,68 +87,8 @@ class UserSettings(db.Model):
|
||||
show_tmnf = Column(Boolean)
|
||||
show_tm_2020_current = Column(Boolean)
|
||||
|
||||
notifications_all = Column(Boolean)
|
||||
notifications_self = Column(Boolean)
|
||||
|
||||
@app.route("/update-user-settings", methods=["GET", "POST"])
|
||||
def update_user_settings():
|
||||
|
||||
user = flask.request.headers.get("X-Forwarded-Preferred-Username")
|
||||
user_helper = user or "anonymous"
|
||||
|
||||
settings = db.session.query(UserSettings).filter(UserSettings.user==user_helper).first()
|
||||
|
||||
# handle new settings #/
|
||||
if not settings:
|
||||
settings = UserSettings(user=user_helper, show_tm_2020=False, show_tmnf=False,
|
||||
show_tm_2020_current=True, notifications_self=True,
|
||||
notifications_all=False)
|
||||
db.session.add(settings)
|
||||
db.session.commit()
|
||||
settings = db.session.query(UserSettings).filter(UserSettings.user==user_helper).first()
|
||||
|
||||
|
||||
if flask.request.method == "GET":
|
||||
|
||||
key = flask.request.args.get("key")
|
||||
|
||||
# some sanity checks #
|
||||
if not key:
|
||||
return ("key missing in args", 422)
|
||||
if key == "user" or key.startswith("_") or "sql" in key:
|
||||
return ("key {} is not allowed".format(key), 422)
|
||||
|
||||
# return attribute #
|
||||
return (str(getattr(settings, key)), 200)
|
||||
|
||||
elif flask.request.method == "POST":
|
||||
json_dict = flask.request.json
|
||||
key_value_list = json_dict.get("payload")
|
||||
|
||||
if not key_value_list:
|
||||
return ("'payload' field empty", 422)
|
||||
|
||||
for el in key_value_list:
|
||||
|
||||
key = el.get("key")
|
||||
value = el.get("value")
|
||||
|
||||
if key == "user" or key.startswith("_") or "sql" in key:
|
||||
return ("key {} is not allowed".format(key), 422)
|
||||
|
||||
if key is None or value is None:
|
||||
return ("element in payload list does not contain key and value", 422)
|
||||
|
||||
try:
|
||||
getattr(settings, key)
|
||||
setattr(settings, key, bool(value))
|
||||
db.session.merge(settings)
|
||||
db.session.commit()
|
||||
return ("", 204)
|
||||
except AttributeError:
|
||||
return ("key {} not part of user settings".format(key), 422)
|
||||
else:
|
||||
raise AssertionError("Unsupported Method: {}".format(flask.request.method))
|
||||
notifcations_all = Column(Boolean)
|
||||
notifcations_self = Column(Boolean)
|
||||
|
||||
class ParsedReplay(db.Model):
|
||||
|
||||
@@ -198,16 +116,13 @@ class ParsedReplay(db.Model):
|
||||
else:
|
||||
return self.login
|
||||
|
||||
def get_human_readable_time(self):
|
||||
def get_human_readable_time(self, thousands=False):
|
||||
t = datetime.timedelta(microseconds=self.race_time*1000)
|
||||
t_string = str(t)
|
||||
if t.seconds < 60*60:
|
||||
t_string = t_string[2:]
|
||||
if t.microseconds != 0:
|
||||
if self.game == "tmnf":
|
||||
return t_string[:-4]
|
||||
else:
|
||||
return t_string[:-3]
|
||||
return t_string[:-4]
|
||||
return t_string + ".00"
|
||||
|
||||
def __repr__(self):
|
||||
@@ -281,7 +196,7 @@ class DataTable():
|
||||
if map_uid:
|
||||
print("Filter for map: {}".format(map_uid))
|
||||
query = query.filter(ParsedReplay.map_uid == map_uid)
|
||||
|
||||
|
||||
total = query.count()
|
||||
if self.searchValue:
|
||||
|
||||
@@ -333,7 +248,7 @@ class DataTable():
|
||||
|
||||
def _extracted_login_from_file(fullpath):
|
||||
'''Extract a login from a tmnf 2020 replay manually'''
|
||||
|
||||
|
||||
# TODO fix underscores in filenames #
|
||||
if "its_a_sheppy" in fullpath:
|
||||
login_from_filename = "its_a_sheppy"
|
||||
@@ -365,7 +280,7 @@ def replay_from_path(fullpath, uploader=None):
|
||||
upload_dt=ghost.upload_dt,
|
||||
cp_times=ghost.cp_times,
|
||||
game=ghost.game)
|
||||
|
||||
|
||||
# build database map object from replay #
|
||||
m = Map(map_uid=replay.map_uid, mapname=replay.map_uid, game=replay.game)
|
||||
|
||||
@@ -409,7 +324,7 @@ def ranks():
|
||||
|
||||
|
||||
@app.route("/map-info")
|
||||
def map_info():
|
||||
def list():
|
||||
player = flask.request.headers.get("X-Forwarded-Preferred-Username")
|
||||
header_col = ["Player", "Time", "Date", "Replay"]
|
||||
map_uid = flask.request.args.get("map_uid")
|
||||
@@ -421,24 +336,17 @@ def mapnames():
|
||||
'''Index Location'''
|
||||
|
||||
# TODO list by user
|
||||
player = flask.request.headers.get("X-Forwarded-Preferred-Username") or "anonymous"
|
||||
player = flask.request.headers.get("X-Forwarded-Preferred-Username")
|
||||
maps_query = db.session.query(Map).order_by(asc(Map.mapname))
|
||||
|
||||
# limit leaderboard to game #
|
||||
settings = db.session.query(UserSettings).filter(UserSettings.user==player).first()
|
||||
if settings:
|
||||
if not settings.show_tm_2020 and not settings.show_tmnf:
|
||||
maps_query = maps_query.filter(Map.game=="tm2020")
|
||||
latest_season = tm2020parser.get_latest_season_from_maps(maps_query.all())
|
||||
# handle no replays #
|
||||
if latest_season:
|
||||
maps_query = maps_query.filter(Map.map_uid.like("{}%".format(latest_season)))
|
||||
elif settings.show_tm_2020 and not settings.show_tmnf:
|
||||
maps_query = maps_query.filter(Map.game=="tm2020")
|
||||
elif not settings.show_tm_2020 and settings.show_tmnf:
|
||||
maps_query = maps_query.filter(Map.game=="tmnf")
|
||||
else:
|
||||
pass
|
||||
game = flask.request.args.get("game")
|
||||
if game == "tm2020":
|
||||
maps_query = maps_query.filter(Map.game=="tm2020")
|
||||
elif game=="tmnf":
|
||||
maps_query = maps_query.filter(Map.game!="tm2020")
|
||||
else:
|
||||
pass
|
||||
|
||||
maps = maps_query.all()
|
||||
|
||||
@@ -446,9 +354,6 @@ def mapnames():
|
||||
allowed = ("A", "B", "C", "D", "E", "Fall", "Winter", "Spring", "Summer")
|
||||
maps_filtered = filter(lambda m: m.mapname.startswith(allowed), maps)
|
||||
|
||||
if settings.show_tm_2020_current:
|
||||
maps_filtered = filter_for_current_season(maps_filtered)
|
||||
|
||||
return flask.render_template("index.html", maps=maps_filtered, player=player)
|
||||
|
||||
@app.route("/open-info")
|
||||
@@ -516,13 +421,12 @@ def upload():
|
||||
|
||||
def check_replay_trigger(replay):
|
||||
|
||||
map_obj = db.session.query(Map).filter(Map.map_uid == replay.map_uid).first()
|
||||
map_obj = db.session.get(Map).filter(Map.map_uid == replay.map_uid).first()
|
||||
assert(map_uid)
|
||||
|
||||
best = map_obj.get_best_replay()
|
||||
second = map_obj.get_second_best_replay()
|
||||
|
||||
if not second:
|
||||
return
|
||||
|
||||
if replay.filehash != best.filehash:
|
||||
return
|
||||
|
||||
@@ -530,7 +434,7 @@ def check_replay_trigger(replay):
|
||||
return
|
||||
|
||||
settings = db.session.query(UserSettings).filter(UserSettings.user == second.uploader).first()
|
||||
if settings and settings.notifications_self:
|
||||
if settings and settings.notifcations_self:
|
||||
notifications.send_notification(app, settings.user, map_obj.map_uid, second, replay)
|
||||
|
||||
def create_app():
|
||||
@@ -539,7 +443,7 @@ def create_app():
|
||||
|
||||
app.config["DISPATCH_SERVER"] = os.environ.get("DISPATCH_SERVER")
|
||||
if app.config["DISPATCH_SERVER"]:
|
||||
app.config["DISPATCH_TOKEN"] = os.environ["DISPATCH_TOKEN"]
|
||||
app.config["DISPATCH_AUTH"] = (os.environ["DISPATCH_AUTH_USER"], os.environ["DISPATCH_AUTH_PASSWORD"])
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
|
||||
@@ -27,15 +27,6 @@ table{
|
||||
border-color: transparent;
|
||||
}
|
||||
|
||||
.user-settings{
|
||||
display: inline-flex;
|
||||
color: white;
|
||||
}
|
||||
|
||||
.form-check{
|
||||
margin-right: 15px;
|
||||
}
|
||||
|
||||
.m-auto{
|
||||
margin: auto;
|
||||
}
|
||||
|
||||
@@ -1,47 +0,0 @@
|
||||
/* event listener for all sliders */
|
||||
const sliders = Array.from(document.getElementsByClassName("form-check-input"))
|
||||
|
||||
/* safety switch -> never run set before get */
|
||||
var sliders_set = false
|
||||
|
||||
/* defer */
|
||||
sliders_load_all()
|
||||
|
||||
/* get initial values & set listeners */
|
||||
function sliders_load_all(){
|
||||
Promise.all(sliders.map(s => {
|
||||
fetch("/update-user-settings?key=" + s.id, { credentials: "include" }).then(response => {
|
||||
response.text().then(data => {
|
||||
if(data == "True"){
|
||||
s.checked = true
|
||||
}
|
||||
})
|
||||
})
|
||||
s.addEventListener("change", submit)
|
||||
})).then(
|
||||
sliders_set = true
|
||||
)
|
||||
}
|
||||
|
||||
/* submit settings */
|
||||
function submit(e){
|
||||
|
||||
console.log("submit")
|
||||
if(!sliders_set){
|
||||
return
|
||||
}
|
||||
|
||||
const s = e.target
|
||||
const json_data = JSON.stringify({ payload : [ { key : s.id, value : s.checked } ] })
|
||||
fetch("/update-user-settings", {
|
||||
method: "POST",
|
||||
credentials: "include",
|
||||
headers: {'Content-Type': 'application/json'},
|
||||
body: json_data
|
||||
}
|
||||
).then(response => {
|
||||
if(s.id.startsWith("show")){
|
||||
window.location.reload()
|
||||
}
|
||||
})
|
||||
}
|
||||
@@ -3,28 +3,15 @@
|
||||
</head>
|
||||
<body>
|
||||
{% include "upload-button.html" %}
|
||||
|
||||
<div class="user-settings">
|
||||
<div class="form-check form-switch">
|
||||
<input class="form-check-input" type="checkbox" role="switch" id="show_tm_2020_current">
|
||||
<label class="form-check-label" for="show_tm_2020_current">Only current TM2020</label>
|
||||
</div>
|
||||
<div class="form-check form-switch">
|
||||
<input class="form-check-input" type="checkbox" role="switch" id="show_tm_2020">
|
||||
<label class="form-check-label" for="show_tm_2020">Show All TM2020</label>
|
||||
</div>
|
||||
<div class="form-check form-switch">
|
||||
<input class="form-check-input" type="checkbox" role="switch" id="show_tmnf">
|
||||
<label class="form-check-label" for="show_tmnf">Show All TMNF</label>
|
||||
</div>
|
||||
<div class="form-check form-switch">
|
||||
<input class="form-check-input" type="checkbox" role="switch" id="notifications_self">
|
||||
<label class="form-check-label" for="notifications_self">Send notifcations</label>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<script src="/static/user_settings.js" defer></script>
|
||||
|
||||
<button class="ml-4 mt-4 mb-4 btn btn-info" onclick="window.location.href='/?game=tmnf'">
|
||||
TMNF
|
||||
</button>
|
||||
<button class="mt-4 mb-4 btn btn-info" onclick="window.location.href='/?game=tm2020'">
|
||||
TM2020
|
||||
</button>
|
||||
<button class="mt-4 mb-4 btn btn-info" onclick="window.location.href='/'">
|
||||
All
|
||||
</button>
|
||||
<table class="m-auto">
|
||||
<thead>
|
||||
<tr>
|
||||
|
||||
@@ -5,41 +5,6 @@ import hashlib
|
||||
import pygbx
|
||||
import xmltodict
|
||||
|
||||
def get_latest_season_from_maps(maps):
|
||||
'''Determine the latest season in DB'''
|
||||
|
||||
order = ["Winter", "Spring", "Summer", "Fall"]
|
||||
|
||||
max_year = 0
|
||||
max_season = "Winter"
|
||||
|
||||
if not maps:
|
||||
return
|
||||
|
||||
for m in maps:
|
||||
splitted = m.map_uid.split(" ")
|
||||
|
||||
# not a campain map #
|
||||
if len(splitted) < 3:
|
||||
return
|
||||
|
||||
season = splitted[0]
|
||||
year = splitted[1]
|
||||
|
||||
# also not a campaing map #
|
||||
if not season in order:
|
||||
return
|
||||
try:
|
||||
year = int(year)
|
||||
except ValueError:
|
||||
return
|
||||
|
||||
if year > max_year or year == max_year and order.index(season) >= order.index(max_season):
|
||||
max_year = year
|
||||
max_season = season
|
||||
|
||||
return "{} {}".format(max_season, max_year)
|
||||
|
||||
class GhostWrapper():
|
||||
|
||||
def __init__(self, fullpath, uploader):
|
||||
|
||||
Reference in New Issue
Block a user