From 67cbf674b1e0234ecc437db08d639b01da2d020f Mon Sep 17 00:00:00 2001
From: Zoe Roux <zoe.roux@zoriya.dev>
Date: Sun, 28 Apr 2024 13:00:17 +0200
Subject: [PATCH] Add optional advanced flag to guess.py for debugging

---
 scanner/matcher/parser/guess.py | 13 +++++++++----
 shell.nix                       |  2 +-
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/scanner/matcher/parser/guess.py b/scanner/matcher/parser/guess.py
index 431f6f45..54b96a0f 100644
--- a/scanner/matcher/parser/guess.py
+++ b/scanner/matcher/parser/guess.py
@@ -7,7 +7,7 @@ if __name__ == "__main__":
 	sys.path.append(str(Path(f"{__file__}/../../..").resolve()))
 
 from guessit.api import default_api
-from typing import cast, List
+from typing import cast, List, Any
 from rebulk import Rebulk
 
 try:
@@ -20,14 +20,15 @@ rblk = cast(Rebulk, default_api.rebulk)
 rblk.rules(rules)
 
 
-def guessit(name: str, *, xem_titles: List[str] = []):
+def guessit(name: str, *, xem_titles: List[str] = [], extra_flags: dict[str, Any] = {}):
 	return default_api.guessit(
 		name,
 		{
 			"episode_prefer_number": True,
 			"excludes": "language",
 			"xem_titles": xem_titles,
-		},
+		}
+		| extra_flags,
 	)
 
 
@@ -44,7 +45,11 @@ if __name__ == "__main__":
 		async with ClientSession() as client:
 			xem = TheXemClient(client)
 
-			ret = guessit(sys.argv[1], xem_titles=await xem.get_expected_titles())
+			ret = guessit(
+				sys.argv[1],
+				xem_titles=await xem.get_expected_titles(),
+				# extra_flags={"advanced": True},
+			)
 			print(json.dumps(ret, cls=GuessitEncoder, indent=4))
 
 	asyncio.run(main())
diff --git a/shell.nix b/shell.nix
index 6f92b2eb..406c8321 100644
--- a/shell.nix
+++ b/shell.nix
@@ -1,5 +1,5 @@
 {pkgs ? import <nixpkgs> {}}: let
-  python = pkgs.python311.withPackages (ps:
+  python = pkgs.python312.withPackages (ps:
     with ps; [
       guessit
       aiohttp