remove enemy builds

Author: padtrack

src/render.py

@@ -1,3 +1,4 @@
+import json
 from pathlib import Path
 from renderer.render import Renderer
 from replay_parser import ReplayParser
@@ -25,6 +26,8 @@
             enable_chat=True,
             use_tqdm=True,
         )
+        with open(path.parent.joinpath(f"{path.stem}-builds.json"), "w") as fp:
+            json.dump(renderer.get_player_build(), fp, indent=4)
         renderer.start(str(video_path))
         LOGGER.info(f"The video file is at: {str(video_path)}")
         LOGGER.info("Done.")

src/renderer/render.py

@@ -348,6 +348,9 @@ def get_player_build(self) -> list[dict]:
         builds = []
 
         for player in self.replay_data.player_info.values():
+            if player.relation not in [-1, 0]:
+                continue
+
             try:
                 index, build_str = self._builder.get_build(player)
                 build_url = f"{url}{index}&build={build_str}"