From c020349f2e495edf8b96fb302a2b71b82cb0db7c Mon Sep 17 00:00:00 2001 From: Jakub Hrabec Date: Fri, 24 Jun 2022 09:08:17 +0200 Subject: [PATCH] Update RunSQL type to include parametrized sql (#1015) * update RunSQL type to include parametrized sql * fix Black formatting --- django-stubs/db/migrations/operations/special.pyi | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/django-stubs/db/migrations/operations/special.pyi b/django-stubs/db/migrations/operations/special.pyi index 3b04f93..1495b1d 100644 --- 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 = ...,