csv.pyi: minor cleanup (#7790)

This commit is contained in:
Alex Waygood
2022-05-07 09:49:27 +01:00
committed by GitHub
parent 5c6178a821
commit 4e946b74bf

View File

@@ -1,4 +1,6 @@
import sys
# actually csv.Dialect is a different class to _csv.Dialect at runtime, but for typing purposes, they're identical
from _csv import (
QUOTE_ALL as QUOTE_ALL,
QUOTE_MINIMAL as QUOTE_MINIMAL,
@@ -19,9 +21,10 @@ from _csv import (
unregister_dialect as unregister_dialect,
writer as writer,
)
from _typeshed import Self
from _typeshed import Self, SupportsWrite
from collections.abc import Collection, Iterable, Iterator, Mapping, Sequence
from typing import Any, Generic, TypeVar, overload
from typing_extensions import Literal
if sys.version_info >= (3, 8):
from builtins import dict as _DictReadMapping
@@ -108,14 +111,14 @@ class DictReader(Generic[_T], Iterator[_DictReadMapping[_T, str]]):
class DictWriter(Generic[_T]):
fieldnames: Collection[_T]
restval: Any | None
extrasaction: str
extrasaction: Literal["raise", "ignore"]
writer: _writer
def __init__(
self,
f: Any,
f: SupportsWrite[str],
fieldnames: Collection[_T],
restval: Any | None = ...,
extrasaction: str = ...,
extrasaction: Literal["raise", "ignore"] = ...,
dialect: _DialectLike = ...,
*args: Any,
**kwds: Any,