Use Mypy's strict mode

Author: adamchainz

.pre-commit-config.yaml

@@ -57,3 +57,7 @@ repos:
   rev: v1.0.0
   hooks:
   - id: mypy
+    additional_dependencies:
+    - django-stubs==1.12.0
+    - mysqlclient
+    - pytest==7.1.2

pyproject.toml

@@ -5,17 +5,16 @@ build-backend = "setuptools.build_meta"
 [tool.black]
 target-version = ['py37']
 
+[tool.django-stubs]
+django_settings_module = "tests.settings"
+
 [tool.mypy]
-check_untyped_defs = true
-disallow_any_generics = true
-disallow_incomplete_defs = true
-disallow_untyped_defs = true
 mypy_path = "src/"
 namespace_packages = false
-no_implicit_optional = true
+plugins = ["mypy_django_plugin.main"]
 show_error_codes = true
+strict = true
 warn_unreachable = true
-warn_unused_ignores = true
 
 [[tool.mypy.overrides]]
 module = "tests.*"

src/django_mysql/compat.py

@@ -6,6 +6,8 @@
 from typing import cast
 from typing import TypeVar
 
+__all__ = ("cache",)
+
 if sys.version_info >= (3, 9):
     from functools import cache
 else:

src/django_mysql/models/__init__.py

@@ -4,7 +4,7 @@
 from django_mysql.models.aggregates import BitOr
 from django_mysql.models.aggregates import BitXor
 from django_mysql.models.aggregates import GroupConcat
-from django_mysql.models.base import Model  # noqa
+from django_mysql.models.base import Model
 from django_mysql.models.expressions import ListF
 from django_mysql.models.expressions import SetF
 from django_mysql.models.fields import Bit1BooleanField
@@ -25,3 +25,32 @@
 from django_mysql.models.query import QuerySetMixin
 from django_mysql.models.query import SmartChunkedIterator
 from django_mysql.models.query import SmartIterator
+
+
+__all__ = (
+    "add_QuerySetMixin",
+    "ApproximateInt",
+    "Bit1BooleanField",
+    "BitAnd",
+    "BitOr",
+    "BitXor",
+    "DynamicField",
+    "EnumField",
+    "FixedCharField",
+    "GroupConcat",
+    "ListCharField",
+    "ListF",
+    "ListTextField",
+    "Model",
+    "NullBit1BooleanField",
+    "pt_visual_explain",
+    "QuerySet",
+    "QuerySetMixin",
+    "SetCharField",
+    "SetF",
+    "SetTextField",
+    "SizedBinaryField",
+    "SizedTextField",
+    "SmartChunkedIterator",
+    "SmartIterator",
+)

src/django_mysql/models/fields/__init__.py

@@ -12,7 +12,7 @@
 from django_mysql.models.fields.sizes import SizedBinaryField
 from django_mysql.models.fields.sizes import SizedTextField
 
-__all__ = [
+__all__ = (
     "Bit1BooleanField",
     "DynamicField",
     "EnumField",
@@ -24,4 +24,4 @@
     "SetTextField",
     "SizedBinaryField",
     "SizedTextField",
-]
+)

src/django_mysql/models/fields/dynamic.py

@@ -25,9 +25,9 @@
 from django.forms import Field as FormField
 from django.utils.translation import gettext_lazy as _
 
-from django_mysql.checks import mysql_connections
 from django_mysql.models.lookups import DynColHasKey
 from django_mysql.typing import DeconstructResult
+from django_mysql.utils import mysql_connections
 
 try:
     import mariadb_dyncol
@@ -89,7 +89,7 @@ def check(self, **kwargs: Any) -> list[checks.CheckMessage]:
         return errors
 
     def _check_mariadb_dyncol(self) -> list[checks.CheckMessage]:
-        errors = []
+        errors: list[checks.CheckMessage] = []
         if mariadb_dyncol is None:
             errors.append(
                 checks.Error(
@@ -102,7 +102,7 @@ def _check_mariadb_dyncol(self) -> list[checks.CheckMessage]:
         return errors
 
     def _check_mariadb_version(self) -> list[checks.CheckMessage]:
-        errors = []
+        errors: list[checks.CheckMessage] = []
 
         any_conn_works = any(
             (conn.vendor == "mysql" and conn.mysql_is_mariadb)
@@ -121,7 +121,7 @@ def _check_mariadb_version(self) -> list[checks.CheckMessage]:
         return errors
 
     def _check_character_set(self) -> list[checks.CheckMessage]:
-        errors = []
+        errors: list[checks.CheckMessage] = []
 
         conn = None
         for _alias, check_conn in mysql_connections():
@@ -153,7 +153,7 @@ def _check_character_set(self) -> list[checks.CheckMessage]:
     def _check_spec_recursively(
         self, spec: Any, path: str = ""
     ) -> list[checks.CheckMessage]:
-        errors = []
+        errors: list[checks.CheckMessage] = []
 
         if not isinstance(spec, dict):
             errors.append(

src/django_mysql/operations.py

@@ -2,7 +2,7 @@
 
 from django.db.backends.base.schema import BaseDatabaseSchemaEditor
 from django.db.migrations.operations.base import Operation
-from django.db.migrations.state import ModelState
+from django.db.migrations.state import ProjectState
 from django.utils.functional import cached_property
 
 
@@ -15,15 +15,15 @@ def __init__(self, name: str, soname: str) -> None:
         self.name = name
         self.soname = soname
 
-    def state_forwards(self, app_label: str, state: ModelState) -> None:
+    def state_forwards(self, app_label: str, state: ProjectState) -> None:
         pass  # pragma: no cover
 
     def database_forwards(
         self,
         app_label: str,
         schema_editor: BaseDatabaseSchemaEditor,
-        from_st: ModelState,
-        to_st: ModelState,
+        from_st: ProjectState,
+        to_st: ProjectState,
     ) -> None:
         if not self.plugin_installed(schema_editor):
             schema_editor.execute(
@@ -34,8 +34,8 @@ def database_backwards(
         self,
         app_label: str,
         schema_editor: BaseDatabaseSchemaEditor,
-        from_st: ModelState,
-        to_st: ModelState,
+        from_st: ProjectState,
+        to_st: ProjectState,
     ) -> None:
         if self.plugin_installed(schema_editor):
             schema_editor.execute("UNINSTALL PLUGIN %s" % self.name)
@@ -63,24 +63,24 @@ class InstallSOName(Operation):
     def __init__(self, soname: str) -> None:
         self.soname = soname
 
-    def state_forwards(self, app_label: str, state: ModelState) -> None:
+    def state_forwards(self, app_label: str, state: ProjectState) -> None:
         pass  # pragma: no cover
 
     def database_forwards(
         self,
         app_label: str,
         schema_editor: BaseDatabaseSchemaEditor,
-        from_st: ModelState,
-        to_st: ModelState,
+        from_st: ProjectState,
+        to_st: ProjectState,
     ) -> None:
         schema_editor.execute("INSTALL SONAME %s", (self.soname,))
 
     def database_backwards(
         self,
         app_label: str,
         schema_editor: BaseDatabaseSchemaEditor,
-        from_st: ModelState,
-        to_st: ModelState,
+        from_st: ProjectState,
+        to_st: ProjectState,
     ) -> None:
         schema_editor.execute("UNINSTALL SONAME %s", (self.soname,))
 
@@ -100,24 +100,24 @@ def __init__(
     def reversible(self) -> bool:
         return self.from_engine is not None
 
-    def state_forwards(self, app_label: str, state: ModelState) -> None:
+    def state_forwards(self, app_label: str, state: ProjectState) -> None:
         pass
 
     def database_forwards(
         self,
         app_label: str,
         schema_editor: BaseDatabaseSchemaEditor,
-        from_state: ModelState,
-        to_state: ModelState,
+        from_state: ProjectState,
+        to_state: ProjectState,
     ) -> None:
         self._change_engine(app_label, schema_editor, to_state, engine=self.engine)
 
     def database_backwards(
         self,
         app_label: str,
         schema_editor: BaseDatabaseSchemaEditor,
-        from_state: ModelState,
-        to_state: ModelState,
+        from_state: ProjectState,
+        to_state: ProjectState,
     ) -> None:
         if self.from_engine is None:
             raise NotImplementedError("You cannot reverse this operation")
@@ -128,7 +128,7 @@ def _change_engine(
         self,
         app_label: str,
         schema_editor: BaseDatabaseSchemaEditor,
-        to_state: ModelState,
+        to_state: ProjectState,
         engine: str,
     ) -> None:
         new_model = to_state.apps.get_model(app_label, self.name)

src/django_mysql/status.py

@@ -6,7 +6,6 @@
 from django.db import connections
 from django.db.backends.utils import CursorWrapper
 from django.db.utils import DEFAULT_DB_ALIAS
-from django.utils.functional import SimpleLazyObject
 
 from django_mysql.exceptions import TimeoutError
 
@@ -16,6 +15,7 @@ class BaseStatus:
     Base class for the status classes
     """
 
+    __slots__ = ("db",)
     query = ""
 
     def __init__(self, using: str | None = None) -> None:
@@ -127,5 +127,5 @@ class SessionStatus(BaseStatus):
     query = "SHOW SESSION STATUS"
 
 
-global_status = SimpleLazyObject(GlobalStatus)
-session_status = SimpleLazyObject(SessionStatus)
+global_status = GlobalStatus()
+session_status = SessionStatus()