updated quality for ore

trade-hub/backend/app/loader.py

@@ -17,6 +17,7 @@ Real schemas this loader consumes:
 """
 import json
 import os
+import re
 
 from app import config
 
@@ -27,6 +28,68 @@ CATEGORY_SHIP_ITEMS = "ship-items"
 CATEGORY_FPS_ITEMS = "fps-items"
 CATEGORY_COMMODITIES = "commodities"
 
+# Commodities that can carry a mining/refining quality value (1-1000). These are
+# the ores, minerals and gems that players actually mine (and their refined
+# counterparts); crafted alloys, gases, organics, processed goods and junk
+# by-products are deliberately excluded. Keep in sync with the Ore quality table
+# at https://starcitizen.tools/Ore_quality.
+QUALITY_MATERIALS = frozenset(
+    name.lower()
+    for name in (
+        "Agricium",
+        "Aluminum",
+        "Aphorite",
+        "Aslarite",
+        "Beradom",
+        "Beryl",
+        "Bexalite",
+        "Borase",
+        "Carinite",
+        "Copper",
+        "Corundum",
+        "Diamond",
+        "Dolivine",
+        "Feynmaline",
+        "Glacosite",
+        "Gold",
+        "Hadanite",
+        "Hephaestanite",
+        "Iron",
+        "Jaclium",
+        "Janalite",
+        "Laranite",
+        "Lindinium",
+        "Ouratite",
+        "Pressurized Ice",
+        "Quantainium",
+        "Quartz",
+        "Riccite",
+        "Sadaryx",
+        "Saldynium",
+        "Savrilium",
+        "Silicon",
+        "Stileron",
+        "Taranite",
+        "Tin",
+        "Titanium",
+        "Torite",
+        "Tungsten",
+    )
+)
+
+# Variant suffixes are presentation labels, not a different material: "Iron
+# (Ore)" and "Iron" share a quality value, as do "(Raw)", "(Pure)" and "(R)".
+_QUALITY_VARIANT = re.compile(r"\s*\((?:Ore|Raw|Pure|R|\d+\s*SCU)\)\s*$", re.IGNORECASE)
+_RAW_PREFIX = re.compile(r"^Raw\s+", re.IGNORECASE)
+
+
+def _quality_applicable(name, category):
+    if category != CATEGORY_COMMODITIES:
+        return False
+    base_name = _QUALITY_VARIANT.sub("", name or "").strip()
+    base_name = _RAW_PREFIX.sub("", base_name).strip().lower()
+    return base_name in QUALITY_MATERIALS
+
 
 def _load_json(data_dir, filename):
     path = os.path.join(data_dir, filename)
@@ -80,6 +143,7 @@ def _normalize_ship(ship):
         "classification": "Ship",
         "description": ship.get("Description") or ship.get("DescriptionText") or "",
         "tags": tags,
+        "quality_applicable": False,
     }, {
         "career": ship.get("Career") or "",
         "role": ship.get("Role") or "",
@@ -111,6 +175,10 @@ def _normalize_item(item, category):
         "classification": item.get("classification") or "",
         "description": std.get("Description") or std.get("DescriptionText") or "",
         "tags": tags,
+        "quality_applicable": _quality_applicable(
+            _resolve_name(std.get("Name"), item.get("name"), class_name=class_name),
+            category,
+        ),
     }
     if category == CATEGORY_COMMODITIES:
         summary["size"] = None