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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
|
from functools import reduce
from werkzeug.exceptions import NotFound
from werkzeug.utils import redirect
from .application import Page
from .utils import unicodecmp
class ServerList(Page):
url_rule = "/"
def order_link(self, name, title):
cls = ""
link = f"?order_by={name}"
desc = False
if name == self.order_by:
desc = not self.order_desc
cls = f' class="{"down" if desc else "up"}"'
if desc:
link += "&dir=desc"
return f'<a href="{link}"{cls}>{title}</a>'
def process(self):
self.order_by = self.request.args.get("order_by") or "name"
sort_func = {
"name": lambda x: x,
"map": lambda x: x.map,
"gametype": lambda x: x.gametype,
"players": lambda x: x.player_count,
"progression": lambda x: x.progression,
}.get(self.order_by)
if sort_func is None:
return redirect(self.url_for("serverlist"))
self.servers = self.cup.server_browser.servers.values()
self.servers.sort(key=sort_func)
if self.request.args.get("dir") == "desc":
self.servers.reverse()
self.order_desc = True
else:
self.order_desc = False
self.players = reduce(lambda a, b: a + b.players, self.servers, [])
self.players = sorted(self.players, key=lambda a, b: unicodecmp(a.name, b.name))
class Server(Page):
url_rule = "/server/<id>"
def process(self, id):
try:
self.server = self.cup.server_browser.servers[id]
except KeyError:
raise NotFound() from None
class Search(Page):
url_rule = "/search"
def process(self):
self.user = self.request.args.get("user")
if self.user:
self.results = []
for server in self.cup.server_browser.servers.values():
for player in server.players:
if player.name == self.user:
self.results.append(server)
class MissingPage(Page):
def get_response(self):
response = super().get_response()
response.status_code = 404
return response
|