From 5ca2d77f85f25d119ddaac7a65904b1b95dcdca2 Mon Sep 17 00:00:00 2001
From: Nikita Sobolev <mail@sobolevn.me>
Date: Wed, 12 Oct 2022 00:28:39 +0300
Subject: [PATCH] Mark `atomicwrites` as complete (#8883)

---
 stubs/atomicwrites/@tests/stubtest_allowlist.txt | 1 -
 stubs/atomicwrites/METADATA.toml                 | 3 +++
 stubs/atomicwrites/atomicwrites/__init__.pyi     | 8 +++++++-
 3 files changed, 10 insertions(+), 2 deletions(-)
 delete mode 100644 stubs/atomicwrites/@tests/stubtest_allowlist.txt

diff --git a/stubs/atomicwrites/@tests/stubtest_allowlist.txt b/stubs/atomicwrites/@tests/stubtest_allowlist.txt
deleted file mode 100644
index 3bf9df36c..000000000
--- a/stubs/atomicwrites/@tests/stubtest_allowlist.txt
+++ /dev/null
@@ -1 +0,0 @@
-atomicwrites.AtomicWriter.get_fileobject
diff --git a/stubs/atomicwrites/METADATA.toml b/stubs/atomicwrites/METADATA.toml
index 6d3b92238..3a60aff54 100644
--- a/stubs/atomicwrites/METADATA.toml
+++ b/stubs/atomicwrites/METADATA.toml
@@ -1 +1,4 @@
 version = "1.4.*"
+
+[tool.stubtest]
+ignore_missing_stub = false
diff --git a/stubs/atomicwrites/atomicwrites/__init__.pyi b/stubs/atomicwrites/atomicwrites/__init__.pyi
index f5ea4ac29..981f0b601 100644
--- a/stubs/atomicwrites/atomicwrites/__init__.pyi
+++ b/stubs/atomicwrites/atomicwrites/__init__.pyi
@@ -2,6 +2,10 @@ from _typeshed import StrOrBytesPath
 from collections.abc import Callable
 from contextlib import AbstractContextManager
 from typing import IO, Any, AnyStr
+from typing_extensions import Literal
+
+PY2: Literal[False]
+DEFAULT_MODE: Literal["w"]
 
 def replace_atomic(src: AnyStr, dst: AnyStr) -> None: ...
 def move_atomic(src: AnyStr, dst: AnyStr) -> None: ...
@@ -10,7 +14,9 @@ class AtomicWriter:
     def __init__(self, path: StrOrBytesPath, mode: str = ..., overwrite: bool = ...) -> None: ...
     def open(self) -> AbstractContextManager[IO[Any]]: ...
     def _open(self, get_fileobject: Callable[..., IO[AnyStr]]) -> AbstractContextManager[IO[AnyStr]]: ...
-    def get_fileobject(self, dir: StrOrBytesPath | None = ..., **kwargs: Any) -> IO[Any]: ...
+    def get_fileobject(
+        self, suffix: str = ..., prefix: str = ..., dir: StrOrBytesPath | None = ..., **kwargs: Any
+    ) -> IO[Any]: ...
     def sync(self, f: IO[Any]) -> None: ...
     def commit(self, f: IO[Any]) -> None: ...
     def rollback(self, f: IO[Any]) -> None: ...