Whitelisted model causes wrong return code of "scrub_validaton" command

RegioHelden · Aug 20, 2024 · 9fb82ed · 9fb82ed
1 parent b882085
commit 9fb82ed
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/django_scrubber/management/commands/scrub_validation.py b/django_scrubber/management/commands/scrub_validation.py
@@ -31,7 +31,8 @@ def handle(self, *args, **options):
                     print(f"- {field}")
                     found_fields += 1
 
-            print(f'{found_models} model(s) having {found_fields} unscrubbed field(s) detected.')
-            sys.exit(1)
+            if found_models > 0 or found_fields > 0:
+                print(f'{found_models} model(s) having {found_fields} unscrubbed field(s) detected.')
+                sys.exit(1)
 
         print('No unscrubbed fields detected. Yeah!')