🩹 (sync/memorious) Tweak strip prefix func

investigativedata · Jan 23, 2025 · 4389eb8 · 4389eb8
1 parent f0b6f50
commit 4389eb8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/leakrfc/sync/memorious.py b/leakrfc/sync/memorious.py
@@ -138,7 +138,12 @@ def _func(data: dict[str, Any]) -> str:
 
 
 def get_file_name_strip_func(strip_prefix: str) -> Callable:
+    strip_prefix = strip_prefix.strip("/")
+
     def _func(data: dict[str, Any]) -> str:
-        return get_file_key(data)[len(strip_prefix) :].strip("/")
+        key = get_file_key(data)
+        if key.startswith(strip_prefix):
+            key = key[len(strip_prefix) :].strip("/")
+        return key
 
     return _func
diff --git a/tests/test_sync.py b/tests/test_sync.py
@@ -68,4 +68,8 @@ def get_key(data):
         get_file_name_strip_func("en/request")(data)
         == "14928/response/55317/attach/5/Communication from the Commission SG 2009 D 51604.pdf"
     )
+    assert (
+        get_file_name_strip_func("foo/bar")(data)
+        == "en/request/14928/response/55317/attach/5/Communication from the Commission SG 2009 D 51604.pdf"
+    )
     assert get_file_name_templ_func("{{ headers.Server }}.pdf")(data) == "nginx.pdf"