better stubs

django-stubs/contrib/messages/storage/base.pyi

@@ -1,26 +1,35 @@
+from typing import Any, List, Optional, Union
+
 from django.http.request import HttpRequest
 from django.http.response import HttpResponseBase
-from typing import List, Optional, Union
 
-class BaseStorage:
-    def __init__(self, request: HttpRequest, *args, **kwargs) -> None: ...
-    def __len__(self) -> int: ...
-    def _get_level(self) -> int: ...
-    @property
-    def _loaded_messages(self) -> Union[List[Message], List[str]]: ...
-    def _prepare_messages(self, messages: List[Message]) -> None: ...
-    def _set_level(self, value: int = ...) -> None: ...
-    def add(self, level: int, message: str, extra_tags: Optional[str] = ...) -> None: ...
-    def update(self, response: HttpResponseBase) -> None: ...
+LEVEL_TAGS: Any
 
 class Message:
-    def __eq__(self, other: Union[str, Message]) -> bool: ...
+    level: int = ...
+    message: str = ...
+    extra_tags: str = ...
     def __init__(
         self, level: int, message: str, extra_tags: Optional[str] = ...
     ) -> None: ...
-    def __str__(self) -> str: ...
-    def _prepare(self) -> None: ...
-    @property
-    def level_tag(self) -> str: ...
+    def __eq__(self, other: Union[str, Message]) -> bool: ...
     @property
     def tags(self) -> str: ...
+    @property
+    def level_tag(self) -> str: ...
+
+class BaseStorage:
+    request: Any = ...
+    used: bool = ...
+    added_new: bool = ...
+    def __init__(
+        self, request: HttpRequest, *args: Any, **kwargs: Any
+    ) -> None: ...
+    def __len__(self) -> int: ...
+    def __iter__(self): ...
+    def __contains__(self, item: Any): ...
+    def update(self, response: HttpResponseBase) -> Optional[List[Message]]: ...
+    def add(
+        self, level: int, message: str, extra_tags: Optional[str] = ...
+    ) -> None: ...
+    level: Any = ...