🗃️ Rendered 31-2 data, added shields

data/render/31-2/render_all.py

@@ -14,7 +14,7 @@ with sync_playwright() as p:
     browser = p.chromium.launch()
     page = browser.new_page(device_scale_factor=3)
     with db_session:
-        for m in Melee.select():
+        for m in [*Melee.select(), *Shield.select()]:
             print(f"rendering {m.name}...")
             page.goto(f"http://localhost:8080/{m.name}")
             page.locator(".container").screenshot(path=f"./output/{m.name}.png")

data/render/31-2/server.py

@@ -18,10 +18,17 @@ db.generate_mapping()
 def melee(name):
     with db_session:
         query = select(w for w in Melee if w.name == name)
-        weapon = query.first()
-    mechanism, _ = re.subn(
-        r"{(.*?)@(.*?)}", r"<span class='\2'>\1</span>", weapon.mechanism
-    )
+        if query.count() > 0:
+            weapon = query.first()
+        else:
+            query = select(w for w in Shield if w.name == name)
+            weapon = query.first()
+    if weapon.mechanism:
+        mechanism, _ = re.subn(
+            r"{(.*?)@(.*?)}", r"<span class='\2'>\1</span>", weapon.mechanism
+        )
+    else:
+        mechanism = ""
     stats_map = {
         1: "brutality",
         2: "survival",
@@ -35,7 +42,7 @@ def melee(name):
         "icon": base64.b64encode(weapon.icon),
         "mechanism": mechanism,
         "description": weapon.description,
-        "stats": stats_map[weapon.stats]
+        "stats": stats_map[weapon.stats],
     }