🚸 Show actual length (#23)

getcodelimit · Sep 3, 2023 · 35e376a · 35e376a
1 parent 917dbea
commit 35e376a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/codelimit/common/report/Report.py b/codelimit/common/report/Report.py
@@ -45,10 +45,10 @@ def summary(self) -> str:
         )
         high_risk = len([m for m in self.codebase.all_measurements() if m.value > 60])
         if high_risk > 0:
-            return f"⏰ Refactoring necessary, unmaintainable functions: {high_risk}"
+            return f"🚨 Refactoring necessary, unmaintainable functions: {high_risk}"
         elif medium_risk > 0:
             return (
-                f"⚠️ Don't forget to refactor, hard-to-maintain functions: "
+                f"🔔 Don't forget to refactor, hard-to-maintain functions: "
                 f"{medium_risk}"
             )
         else:

diff --git a/codelimit/common/utils.py b/codelimit/common/utils.py
@@ -93,7 +93,7 @@ def format_unit(name: str, length: int, file: Union[str, None] = None) -> Text:
     else:
         color = "green"
     result = Text()
-    result.append(f"{length:3}" if length < 61 else "60+")
+    result.append(f"{length:3}" if length < 1000 else str(length))
     result.append(Text(" | ", style=Style(color=color)))
     if file:
         result.append(Text(f"{file}:", style=Style(dim=True)))

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -4,4 +4,6 @@
 def test_format_unit():
     assert format_unit("bar()", 1).markup == "  1[green] | [/green]bar()"
     assert format_unit("bar()", 20).markup == " 20[yellow] | [/yellow]bar()"
-    assert format_unit("bar()", 70).markup == "60+[red] | [/red]bar()"
+    assert format_unit("bar()", 70).markup == " 70[red] | [/red]bar()"
+    assert format_unit("bar()", 700).markup == "700[red] | [/red]bar()"
+    assert format_unit("bar()", 7000).markup == "7000[red] | [/red]bar()"