From dbccc9a9cb4a3546562d6aa713571f1acef9d842 Mon Sep 17 00:00:00 2001 From: dancergraham Date: Mon, 25 Nov 2024 23:11:50 +0100 Subject: [PATCH] refactor: add types --- waffle/management/commands/waffle_flag.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/waffle/management/commands/waffle_flag.py b/waffle/management/commands/waffle_flag.py index 3fca539a..72ec7073 100644 --- 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}')