update load function to look in family games

Author: edwardchalstrey1

catalog/utils.py

@@ -22,15 +22,20 @@ def load(slug: str) -> gbt.Game:
     """
     slug = str(Path(slug)).replace("\\", "/")
 
+    # Try to load from file
     for suffix, reader in READERS.items():
         resource_path = _CATALOG_RESOURCE / f"{slug}{suffix}"
-
         if resource_path.is_file():
-            # as_file ensures we have a real filesystem path for the reader
             with as_file(resource_path) as path:
                 return reader(str(path))
 
-    raise FileNotFoundError(f"No catalog entry called {slug}.nfg or {slug}.efg")
+    # Try loading from family games
+    fg = family_games()
+    if slug in fg:
+        return fg[slug]
+
+    # Raise error if game does not exist
+    raise FileNotFoundError(f"No catalog entry called {slug}")
 
 
 def games() -> pd.DataFrame:

tests/test_catalog.py

@@ -30,6 +30,12 @@ def test_catalog_load_subdir_slug():
     assert isinstance(g, gbt.Game)
 
 
+def test_catalog_load_family_game():
+    """Test loading a game generated from code with a game family func."""
+    g = gbt.catalog.load("one_shot_trust")
+    assert isinstance(g, gbt.Game)
+
+
 def test_catalog_games():
     """Test games() function returns df of game slugs and titles"""
     all_games = gbt.catalog.games()