From 44f7c25a4e533e37c2ca4ef80a067e0c5e1733cc Mon Sep 17 00:00:00 2001
From: Tim Nyborg <tim.nyborg@gmail.com>
Date: Tue, 3 Aug 2021 09:13:15 +0100
Subject: [PATCH] extend ProtectedError and RestrictedError stubs (#682)

types for their properties
---
 django-stubs/db/models/deletion.pyi | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/django-stubs/db/models/deletion.pyi b/django-stubs/db/models/deletion.pyi
index 3d5f175..5fea4a8 100644
--- a/django-stubs/db/models/deletion.pyi
+++ b/django-stubs/db/models/deletion.pyi
@@ -1,4 +1,4 @@
-from typing import Any, Callable, Collection, Iterable, Optional, Type, Union
+from typing import Any, Callable, Collection, Iterable, Optional, Set, Type, Union
 
 from django.db import IntegrityError
 from django.db.models.base import Model
@@ -14,8 +14,13 @@ def RESTRICT(collector, field, sub_objs, using): ...
 def SET(value: Any) -> Callable: ...
 def get_candidate_relations_to_delete(opts: Options) -> Iterable[Field]: ...
 
-class ProtectedError(IntegrityError): ...
-class RestrictedError(IntegrityError): ...
+class ProtectedError(IntegrityError):
+    protected_objects: Set[Model]
+    def __init__(self, msg: str, protected_objects: Set[Model]) -> None: ...
+
+class RestrictedError(IntegrityError):
+    restricted_objects: Set[Model]
+    def __init__(self, msg: str, restricted_objects: Set[Model]) -> None: ...
 
 class Collector:
     def __init__(self, using: str) -> None: ...