better stubs

django-stubs/db/migrations/writer.pyi

@@ -1,8 +1,32 @@
+from typing import Any, Optional, Set, Tuple, Type
+
 from django.db.migrations.migration import Migration
 from django.db.migrations.operations.base import Operation
-from typing import Any, Set, Tuple, Type, Union
+
+
+class SettingsReference(str):
+    def __new__(
+        self: Type[SettingsReference], value: str, setting_name: str
+    ) -> SettingsReference: ...
+    setting_name: str = ...
+    def __init__(self, value: str, setting_name: str) -> None: ...
+
+class OperationWriter:
+    operation: django.db.migrations.operations.models.CreateModel = ...
+    buff: List[Any] = ...
+    indentation: int = ...
+    def __init__(
+        self, operation: Operation, indentation: int = ...
+    ) -> None: ...
+    def serialize(self) -> Tuple[str, Set[str]]: ...
+    def indent(self) -> None: ...
+    def unindent(self) -> None: ...
+    def feed(self, line: str) -> None: ...
+    def render(self) -> str: ...
 
 class MigrationWriter:
+    migration: django.db.migrations.migration.Migration = ...
+    needs_manual_porting: bool = ...
     def __init__(self, migration: Migration) -> None: ...
     def as_string(self) -> str: ...
     @property
@@ -12,19 +36,6 @@ class MigrationWriter:
     @property
     def path(self) -> str: ...
     @classmethod
-    def serialize(cls, value: Any) -> Union[Tuple[str, Set[str]], Tuple[str, Set[Any]]]: ...
+    def serialize(cls, value: Any) -> Tuple[str, Set[str]]: ...
 
-class OperationWriter:
-    def __init__(self, operation: Operation, indentation: int = ...) -> None: ...
-    def feed(self, line: str) -> None: ...
-    def indent(self) -> None: ...
-    def render(self) -> str: ...
-    def serialize(self) -> Union[Tuple[str, Set[str]], Tuple[str, Set[Any]]]: ...
-    def unindent(self) -> None: ...
-
-class SettingsReference:
-    def __init__(self, value: str, setting_name: str) -> None: ...
-    @staticmethod
-    def __new__(
-        self: Type[SettingsReference], value: str, setting_name: str
-    ) -> SettingsReference: ...
+MIGRATION_TEMPLATE: str