Update django.core.management.commands.* (#1829)

* Narrow overridden `Command.handle` signatures
* Add missing entries in `BaseCommand`
* Update `runserver` command
* Update `loaddata` command
* Update `flush` Command
* Update `inspectdb` command `get_meta` signature
* Update `makemessages` command types
* Add missing allowlist entry after `cached_property` update

Author: UnknownPlatypus

django-stubs/contrib/contenttypes/management/commands/remove_stale_contenttypes.pyi

@@ -1,10 +1,11 @@
-from typing import Literal
+from typing import Any, Literal
 
 from _typeshed import Unused
 from django.core.management import BaseCommand
 from django.db.models.deletion import Collector
 
-class Command(BaseCommand): ...
+class Command(BaseCommand):
+    def handle(self, **options: Any) -> None: ...
 
 class NoFastDeleteCollector(Collector):
     def can_fast_delete(self, *args: Unused, **kwargs: Unused) -> Literal[False]: ...

django-stubs/contrib/sessions/management/commands/clearsessions.pyi

@@ -1,3 +1,6 @@
+from typing import Any
+
 from django.core.management.base import BaseCommand
 
-class Command(BaseCommand): ...
+class Command(BaseCommand):
+    def handle(self, **options: Any) -> None: ...

django-stubs/core/management/base.pyi

@@ -7,6 +7,8 @@ from django.apps.config import AppConfig
 from django.core.management.color import Style
 from django.utils.datastructures import _ListOrTuple
 
+ALL_CHECKS: Literal["__all__"]
+
 class CommandError(Exception):
     def __init__(self, *args: Any, returncode: int = ..., **kwargs: Any) -> None: ...
 
@@ -49,6 +51,7 @@ class BaseCommand:
     requires_system_checks: _ListOrTuple[str] | Literal["__all__"]
     base_stealth_options: tuple[str, ...]
     stealth_options: tuple[str, ...]
+    suppressed_base_arguments: set[str]
     stdout: OutputWrapper
     stderr: OutputWrapper
     style: Style
@@ -62,6 +65,7 @@ class BaseCommand:
     def get_version(self) -> str: ...
     def create_parser(self, prog_name: str, subcommand: str, **kwargs: Any) -> CommandParser: ...
     def add_arguments(self, parser: CommandParser) -> None: ...
+    def add_base_argument(self, parser: CommandParser, *args: Any, **kwargs: Any) -> None: ...
     def print_help(self, prog_name: str, subcommand: str) -> None: ...
     def run_from_argv(self, argv: list[str]) -> None: ...
     def execute(self, *args: Any, **options: Any) -> str | None: ...

django-stubs/core/management/commands/compilemessages.pyi

@@ -1,4 +1,5 @@
 from pathlib import Path
+from typing import Any
 
 from django.core.management.base import BaseCommand
 from django.utils._os import _PathCompatible
@@ -11,4 +12,5 @@ class Command(BaseCommand):
     program_options: list[str]
     verbosity: int
     has_errors: bool
+    def handle(self, **options: Any) -> None: ...
     def compile_messages(self, locations: list[tuple[_PathCompatible, _PathCompatible]]) -> None: ...

django-stubs/core/management/commands/dbshell.pyi

@@ -1,3 +1,6 @@
+from typing import Any
+
 from django.core.management.base import BaseCommand
 
-class Command(BaseCommand): ...
+class Command(BaseCommand):
+    def handle(self, **options: Any) -> None: ...

django-stubs/core/management/commands/diffsettings.pyi

@@ -6,6 +6,7 @@ from django.core.management.base import BaseCommand
 def module_to_dict(module: Any, omittable: Callable[[str], bool] = ...) -> dict[str, str]: ...
 
 class Command(BaseCommand):
+    def handle(self, **options: Any) -> str: ...
     def output_hash(
         self, user_settings: dict[str, str], default_settings: dict[str, str], **options: Any
     ) -> list[str]: ...

django-stubs/core/management/commands/flush.pyi

@@ -1,6 +1,9 @@
+from typing import Any
+
 from django.core.management.base import BaseCommand
 from django.core.management.color import Style
 
 class Command(BaseCommand):
-    stealth_options: tuple[str]
     style: Style
+
+    def handle(self, **options: Any) -> None: ...

django-stubs/core/management/commands/inspectdb.pyi

@@ -5,8 +5,9 @@ from django.core.management.base import BaseCommand
 from django.db.backends.base.base import BaseDatabaseWrapper
 
 class Command(BaseCommand):
-    stealth_options: tuple[str]
     db_module: str
+
+    def handle(self, **options: Any) -> None: ...
     def handle_inspection(self, options: dict[str, Any]) -> Iterable[str]: ...
     def normalize_col_name(
         self, col_name: str, used_column_names: list[str], is_relation: bool
@@ -15,5 +16,11 @@ class Command(BaseCommand):
         self, connection: BaseDatabaseWrapper, table_name: str, row: Any
     ) -> tuple[str, dict[str, Any], list[str]]: ...
     def get_meta(
-        self, table_name: str, constraints: Any, column_to_field_name: Any, is_view: Any, is_partition: Any
+        self,
+        table_name: str,
+        constraints: Any,
+        column_to_field_name: Any,
+        is_view: bool,
+        is_partition: bool,
+        comment: str | None,
     ) -> list[str]: ...

django-stubs/core/management/commands/loaddata.pyi

@@ -1,27 +1,48 @@
 import zipfile
 from collections.abc import Sequence
+from io import BufferedReader
+from typing import Callable, Literal
 
 from django.apps.config import AppConfig
 from django.core.management.base import BaseCommand
+from django.core.serializers.base import DeserializedObject
+from django.db.backends.base.base import BaseDatabaseWrapper
 from django.db.models.base import Model
 from django.utils.functional import cached_property
+from typing_extensions import TypeAlias
 
 has_bz2: bool
 has_lzma: bool
 
 READ_STDIN: str
+_ReadBinaryMode: TypeAlias = Literal["r", "rb"]
 
 class Command(BaseCommand):
+    missing_args_message: str
     ignore: bool
     using: str
     app_label: str
     verbosity: int
     excluded_models: set[type[Model]]
     excluded_apps: set[AppConfig]
     format: str
-    missing_args_message: str
+    fixture_count: int
+    loaded_object_count: int
+    fixture_object_count: int
+    models: set[type[Model]]
+    serialization_formats: list[str]
+    objs_with_deferred_fields: list[DeserializedObject]
+    @cached_property
+    def compression_formats(self) -> dict[str | None, tuple[Callable[[str, _ReadBinaryMode], BufferedReader]]]: ...
+    def reset_sequences(self, connection: BaseDatabaseWrapper, models: set[type[Model]]) -> None: ...
     def loaddata(self, fixture_labels: Sequence[str]) -> None: ...
+    def save_obj(self, obj: DeserializedObject) -> bool: ...
     def load_label(self, fixture_label: str) -> None: ...
+    def get_fixture_name_and_dirs(self, fixture_name: str) -> tuple[str, list[str]]: ...
+    def get_targets(self, fixture_name: str, ser_fmt: str | None, cmp_fmt: str | None) -> set[str]: ...
+    def find_fixture_files_in_dir(
+        self, fixture_dir: str, fixture_name: str, targets: set[str]
+    ) -> list[tuple[str, str, str]]: ...
     def find_fixtures(self, fixture_label: str) -> list[tuple[str, str | None, str | None]]: ...
     @cached_property
     def fixture_dirs(self) -> list[str]: ...

django-stubs/core/management/commands/makemessages.pyi

@@ -1,4 +1,4 @@
-from re import Pattern
+from re import Match, Pattern
 from typing import Any
 
 from django.core.management.base import BaseCommand
@@ -9,6 +9,7 @@ STATUS_OK: int
 NO_LOCALE_DIR: Any
 
 def check_programs(*programs: str) -> None: ...
+def is_valid_locale(locale: str) -> Match[str] | None: ...
 
 class TranslatableFile:
     dirpath: str
@@ -21,6 +22,9 @@ class BuildFile:
     Represent the state of a translatable file during the build process.
     """
 
+    command: BaseCommand
+    domain: str
+    translatable: TranslatableFile
     def __init__(self, command: BaseCommand, domain: str, translatable: TranslatableFile) -> None: ...
     @cached_property
     def is_templatized(self) -> bool: ...
@@ -42,3 +46,25 @@ class Command(BaseCommand):
     msguniq_options: list[str]
     msgattrib_options: list[str]
     xgettext_options: list[str]
+
+    domain: str
+    verbosity: int
+    symlinks: bool
+    ignore_patterns: list[str]
+    no_obsolete: bool
+    keep_pot: bool
+    extensions: set[str]
+    invoked_for_django: bool
+    locale_paths: list[str]
+    default_locale_path: str | None
+    @cached_property
+    def gettext_version(self) -> tuple[int, int] | tuple[int, int, int]: ...
+    @cached_property
+    def settings_available(self) -> bool: ...
+    def build_potfiles(self) -> list[str]: ...
+    def remove_potfiles(self) -> None: ...
+    def find_files(self, root: str) -> list[TranslatableFile]: ...
+    def process_files(self, file_list: list[TranslatableFile]) -> None: ...
+    def process_locale_dir(self, locale_dir: str, files: list[TranslatableFile]) -> None: ...
+    def write_po_file(self, potfile: str, locale: str) -> None: ...
+    def copy_plural_forms(self, msgs: str, locale: str) -> str: ...