truncate set repr after 3 paths

ArcanaFramework · Sep 17, 2024 · c1c15bc · c1c15bc
1 parent c040b82
commit c1c15bc
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/fileformats/generic/set.py b/fileformats/generic/set.py
@@ -13,6 +13,18 @@ class TypedSet(FileSet):
 
     content_types: ty.Tuple[ty.Type[FileSet], ...] = ()
 
+    MAX_REPR_PATHS = 3
+
+    def __repr__(self) -> str:
+        paths_repr = (
+            "'"
+            + "', '".join(str(p) for p in sorted(self.fspaths)[: self.MAX_REPR_PATHS])
+            + "'"
+        )
+        if len(self.fspaths) > self.MAX_REPR_PATHS:
+            paths_repr += ", ..."
+        return f"{self.type_name}({paths_repr})"
+
     @cached_property
     def contents(self) -> ty.List[FileSet]:
         contnts = []