Modernize TYPE_CHECKING blocks

Author: bluetech

pytest_django/asserts.py

@@ -2,16 +2,15 @@
 Dynamically load all Django assertion cases and expose them for importing.
 """
 from functools import wraps
-from typing import Any, Callable, Optional, Sequence, Set, Type, Union
+from typing import (
+    TYPE_CHECKING, Any, Callable, Optional, Sequence, Set, Type, Union,
+)
 
 from django.test import (
     LiveServerTestCase, SimpleTestCase, TestCase, TransactionTestCase,
 )
 
 
-TYPE_CHECKING = False
-
-
 test_case = TestCase("run")
 
 

pytest_django/fixtures.py

@@ -2,7 +2,10 @@
 import os
 from contextlib import contextmanager
 from functools import partial
-from typing import Any, Generator, Iterable, List, Optional, Tuple, Union
+from typing import (
+    TYPE_CHECKING, Any, Generator, Iterable, List, Literal, Optional, Tuple,
+    Union,
+)
 
 import pytest
 
@@ -11,16 +14,14 @@
 from .lazy_django import skip_if_no_django
 
 
-TYPE_CHECKING = False
 if TYPE_CHECKING:
-    from typing import Literal
-
     import django
 
-    _DjangoDbDatabases = Optional[Union["Literal['__all__']", Iterable[str]]]
-    _DjangoDbAvailableApps = Optional[List[str]]
-    # transaction, reset_sequences, databases, serialized_rollback, available_apps
-    _DjangoDb = Tuple[bool, bool, _DjangoDbDatabases, bool, _DjangoDbAvailableApps]
+
+_DjangoDbDatabases = Optional[Union[Literal['__all__'], Iterable[str]]]
+_DjangoDbAvailableApps = Optional[List[str]]
+# transaction, reset_sequences, databases, serialized_rollback, available_apps
+_DjangoDb = Tuple[bool, bool, _DjangoDbDatabases, bool, _DjangoDbAvailableApps]
 
 
 __all__ = [
@@ -239,7 +240,7 @@ def tearDownClass(cls) -> None:
     request.addfinalizer(test_case._post_teardown)
 
 
-def validate_django_db(marker) -> "_DjangoDb":
+def validate_django_db(marker) -> _DjangoDb:
     """Validate the django_db marker.
 
     It checks the signature and creates the ``transaction``,
@@ -254,10 +255,10 @@ def validate_django_db(marker) -> "_DjangoDb":
     def apifun(
         transaction: bool = False,
         reset_sequences: bool = False,
-        databases: "_DjangoDbDatabases" = None,
+        databases: _DjangoDbDatabases = None,
         serialized_rollback: bool = False,
-        available_apps: "_DjangoDbAvailableApps" = None,
-    ) -> "_DjangoDb":
+        available_apps: _DjangoDbAvailableApps = None,
+    ) -> _DjangoDb:
         return transaction, reset_sequences, databases, serialized_rollback, available_apps
 
     return apifun(*marker.args, **marker.kwargs)

pytest_django/plugin.py

@@ -10,7 +10,10 @@
 import pathlib
 import sys
 from functools import reduce
-from typing import Generator, List, Optional, Tuple, Union
+from typing import (
+    TYPE_CHECKING, ContextManager, Generator, List, NoReturn, Optional, Tuple,
+    Union,
+)
 
 import pytest
 
@@ -46,10 +49,7 @@
 from .lazy_django import django_settings_is_configured, skip_if_no_django
 
 
-TYPE_CHECKING = False
 if TYPE_CHECKING:
-    from typing import ContextManager, NoReturn
-
     import django