Merge pull request #57 from GitRon/bugfix/scrub-validator-wrong-retur…

…n-on-whitelisted-field Whitelisted model causes wrong return code of "scrub_validaton" command
RegioHelden · Aug 20, 2024 · 9039515 · 9039515
2 parents b882085 + 9fb82ed
commit 9039515
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!')