refactor: add types

jazzband · Nov 25, 2024 · dbccc9a · dbccc9a
1 parent af0e238
commit dbccc9a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/waffle/management/commands/waffle_flag.py b/waffle/management/commands/waffle_flag.py
@@ -5,6 +5,7 @@
 from django.core.management.base import BaseCommand, CommandError, CommandParser
 from django.db.models import Q
 
+from build.lib.waffle.models import AbstractBaseFlag
 from waffle import get_waffle_flag_model
 
 UserModel = get_user_model()
@@ -166,7 +167,7 @@ def handle(self, *args: Any, **options: Any) -> None:
 
         flag.save()
 
-    def log_flag_to_stdout(self, flag):
+    def log_flag_to_stdout(self, flag: AbstractBaseFlag) -> None:
         self.stdout.write(f'NAME: {flag.name}')
         self.stdout.write(f'SUPERUSERS: {flag.superusers}')
         self.stdout.write(f'EVERYONE: {flag.everyone}')