Merge pull request #121 from se1exin/chore_chaos_env

chore: adding chaos env that makes cleanarr show all non duplicates instead of duplicates

backend/benchmarks.py

@@ -6,6 +6,7 @@
 # be invoked directly with ./backend/benchmark.py to simply run get_dupe_content()
 # and print traces to stdout (note: traces only available if DEBUG=1 set)
 
+import os
 import pytest
 import time
 from plexwrapper import PlexWrapper
@@ -14,14 +15,18 @@
 
 load_dotenv()
 
-def get_dupe_content():
-    PlexWrapper().get_dupe_content()
+def get_dupe_content(page):
+    return PlexWrapper().get_dupe_content(int(page))
 
 def test_get_dupe_content(benchmark):
     benchmark.pedantic(get_dupe_content, iterations=10, rounds=3)
 
 
 # allow for direct invocation, without pytest
 if __name__ == "__main__":
-    get_dupe_content()
+    dupes = get_dupe_content(os.getenv("PAGE", "1"))
+    if dupes:
+        print(f"found {len(dupes)} dupes")
+    else:
+        print("no data found ...")
     print_top_traces(10)

backend/plexwrapper.py

@@ -83,6 +83,12 @@ def get_dupe_content_for_section(self, page, section):
         if section.type not in ("movie", "show"):
             return {}
         dupes = []
+        duplicate=True
+        # undocumented environment variable purely for development purposes.
+        # instead of looking for duplicates, this will look for non-duplicates
+        # which can be useful when wanting to test the UI against a larger dataset
+        if os.getenv("CHAOS_NOT_DUPLICATE", "0") == "1":
+            duplicate=False
         to_dict_func = self.movie_to_dict
         if section.type == "episode":
             to_dict_func = self.episode_to_dict
@@ -101,9 +107,9 @@ def get_dupe_content_for_section(self, page, section):
             libtype = section.type
             if libtype == "show":
                 libtype = "episode"
-            results = section.search(duplicate=True, libtype=libtype, container_start=offset, limit=limit)
+            results = section.search(duplicate=duplicate, libtype=libtype, container_start=offset, limit=limit)
             for item in results:
-                if len(item.media) > 1:
+                if not duplicate or len(item.media) > 1:
                     future = executor.submit(to_dict_func, item, section.title)
                     futures.append(future)