Update RunSQL type to include parametrized sql (#1015)

* update RunSQL type to include parametrized sql * fix Black formatting
typeddjango · Jun 24, 2022 · c020349 · c020349
1 parent 6440521
commit c020349
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/django-stubs/db/migrations/operations/special.pyi b/django-stubs/db/migrations/operations/special.pyi
@@ -1,5 +1,5 @@
 import sys
-from typing import Any, Mapping, Optional, Sequence, Union
+from typing import Any, Mapping, Optional, Sequence, Tuple, Union
 
 from django.db.backends.base.schema import BaseDatabaseSchemaEditor
 from django.db.migrations.state import StateApps
@@ -21,14 +21,16 @@ class SeparateDatabaseAndState(Operation):
 
 class RunSQL(Operation):
     noop: Literal[""] = ...
-    sql: Union[str, _ListOrTuple[str]] = ...
-    reverse_sql: Optional[Union[str, _ListOrTuple[str]]] = ...
+    sql: Union[str, _ListOrTuple[str], _ListOrTuple[Tuple[str, Optional[_ListOrTuple[str]]]]] = ...
+    reverse_sql: Optional[Union[str, _ListOrTuple[str], _ListOrTuple[Tuple[str, Optional[_ListOrTuple[str]]]]]] = ...
     state_operations: Sequence[Operation] = ...
     hints: Mapping[str, Any] = ...
     def __init__(
         self,
-        sql: Union[str, _ListOrTuple[str]],
-        reverse_sql: Optional[Union[str, _ListOrTuple[str]]] = ...,
+        sql: Union[str, _ListOrTuple[str], _ListOrTuple[Tuple[str, Optional[_ListOrTuple[str]]]]],
+        reverse_sql: Optional[
+            Union[str, _ListOrTuple[str], _ListOrTuple[Tuple[str, Optional[_ListOrTuple[str]]]]]
+        ] = ...,
         state_operations: Sequence[Operation] = ...,
         hints: Optional[Mapping[str, Any]] = ...,
         elidable: bool = ...,