@@ -54,7 +54,7 @@ from typing import ( # noqa: Y022
54
54
overload ,
55
55
type_check_only ,
56
56
)
57
- from typing_extensions import Literal , Self , SupportsIndex , TypeAlias , TypeGuard , final
57
+ from typing_extensions import Literal , LiteralString , Self , SupportsIndex , TypeAlias , TypeGuard , final
58
58
59
59
if sys .version_info >= (3 , 9 ):
60
60
from types import GenericAlias
@@ -413,20 +413,38 @@ class str(Sequence[str]):
413
413
def __new__ (cls , object : object = ...) -> Self : ...
414
414
@overload
415
415
def __new__ (cls , object : ReadableBuffer , encoding : str = ..., errors : str = ...) -> Self : ...
416
+ @overload
417
+ def capitalize (self : LiteralString ) -> LiteralString : ...
418
+ @overload
416
419
def capitalize (self ) -> str : ... # type: ignore[misc]
420
+ @overload
421
+ def casefold (self : LiteralString ) -> LiteralString : ...
422
+ @overload
417
423
def casefold (self ) -> str : ... # type: ignore[misc]
424
+ @overload
425
+ def center (self : LiteralString , __width : SupportsIndex , __fillchar : LiteralString = " " ) -> LiteralString : ...
426
+ @overload
418
427
def center (self , __width : SupportsIndex , __fillchar : str = " " ) -> str : ... # type: ignore[misc]
419
428
def count (self , x : str , __start : SupportsIndex | None = ..., __end : SupportsIndex | None = ...) -> int : ...
420
429
def encode (self , encoding : str = "utf-8" , errors : str = "strict" ) -> bytes : ...
421
430
def endswith (
422
431
self , __suffix : str | tuple [str , ...], __start : SupportsIndex | None = ..., __end : SupportsIndex | None = ...
423
432
) -> bool : ...
424
433
if sys .version_info >= (3 , 8 ):
434
+ @overload
435
+ def expandtabs (self : LiteralString , tabsize : SupportsIndex = 8 ) -> LiteralString : ...
436
+ @overload
425
437
def expandtabs (self , tabsize : SupportsIndex = 8 ) -> str : ... # type: ignore[misc]
426
438
else :
439
+ @overload
440
+ def expandtabs (self : LiteralString , tabsize : int = 8 ) -> LiteralString : ...
441
+ @overload
427
442
def expandtabs (self , tabsize : int = 8 ) -> str : ... # type: ignore[misc]
428
443
429
444
def find (self , __sub : str , __start : SupportsIndex | None = ..., __end : SupportsIndex | None = ...) -> int : ...
445
+ @overload
446
+ def format (self : LiteralString , * args : LiteralString , ** kwargs : LiteralString ) -> LiteralString : ...
447
+ @overload
430
448
def format (self , * args : object , ** kwargs : object ) -> str : ... # type: ignore[misc]
431
449
def format_map (self , map : _FormatMapMapping ) -> str : ...
432
450
def index (self , __sub : str , __start : SupportsIndex | None = ..., __end : SupportsIndex | None = ...) -> int : ...
@@ -442,32 +460,91 @@ class str(Sequence[str]):
442
460
def isspace (self ) -> bool : ...
443
461
def istitle (self ) -> bool : ...
444
462
def isupper (self ) -> bool : ...
463
+ @overload
464
+ def join (self : LiteralString , __iterable : Iterable [LiteralString ]) -> LiteralString : ...
465
+ @overload
445
466
def join (self , __iterable : Iterable [str ]) -> str : ... # type: ignore[misc]
467
+ @overload
468
+ def ljust (self : LiteralString , __width : SupportsIndex , __fillchar : LiteralString = " " ) -> LiteralString : ...
469
+ @overload
446
470
def ljust (self , __width : SupportsIndex , __fillchar : str = " " ) -> str : ... # type: ignore[misc]
471
+ @overload
472
+ def lower (self : LiteralString ) -> LiteralString : ...
473
+ @overload
447
474
def lower (self ) -> str : ... # type: ignore[misc]
475
+ @overload
476
+ def lstrip (self : LiteralString , __chars : LiteralString | None = None ) -> LiteralString : ...
477
+ @overload
448
478
def lstrip (self , __chars : str | None = None ) -> str : ... # type: ignore[misc]
479
+ @overload
480
+ def partition (self : LiteralString , __sep : LiteralString ) -> tuple [LiteralString , LiteralString , LiteralString ]: ...
481
+ @overload
449
482
def partition (self , __sep : str ) -> tuple [str , str , str ]: ... # type: ignore[misc]
483
+ @overload
484
+ def replace (
485
+ self : LiteralString , __old : LiteralString , __new : LiteralString , __count : SupportsIndex = - 1
486
+ ) -> LiteralString : ...
487
+ @overload
450
488
def replace (self , __old : str , __new : str , __count : SupportsIndex = - 1 ) -> str : ... # type: ignore[misc]
451
489
if sys .version_info >= (3 , 9 ):
490
+ @overload
491
+ def removeprefix (self : LiteralString , __prefix : LiteralString ) -> LiteralString : ...
492
+ @overload
452
493
def removeprefix (self , __prefix : str ) -> str : ... # type: ignore[misc]
494
+ @overload
495
+ def removesuffix (self : LiteralString , __suffix : LiteralString ) -> LiteralString : ...
496
+ @overload
453
497
def removesuffix (self , __suffix : str ) -> str : ... # type: ignore[misc]
454
498
455
499
def rfind (self , __sub : str , __start : SupportsIndex | None = ..., __end : SupportsIndex | None = ...) -> int : ...
456
500
def rindex (self , __sub : str , __start : SupportsIndex | None = ..., __end : SupportsIndex | None = ...) -> int : ...
501
+ @overload
502
+ def rjust (self : LiteralString , __width : SupportsIndex , __fillchar : LiteralString = " " ) -> LiteralString : ...
503
+ @overload
457
504
def rjust (self , __width : SupportsIndex , __fillchar : str = " " ) -> str : ... # type: ignore[misc]
505
+ @overload
506
+ def rpartition (self : LiteralString , __sep : LiteralString ) -> tuple [LiteralString , LiteralString , LiteralString ]: ...
507
+ @overload
458
508
def rpartition (self , __sep : str ) -> tuple [str , str , str ]: ... # type: ignore[misc]
509
+ @overload
510
+ def rsplit (self : LiteralString , sep : LiteralString | None = None , maxsplit : SupportsIndex = - 1 ) -> list [LiteralString ]: ...
511
+ @overload
459
512
def rsplit (self , sep : str | None = None , maxsplit : SupportsIndex = - 1 ) -> list [str ]: ... # type: ignore[misc]
513
+ @overload
514
+ def rstrip (self : LiteralString , __chars : LiteralString | None = None ) -> LiteralString : ...
515
+ @overload
460
516
def rstrip (self , __chars : str | None = None ) -> str : ... # type: ignore[misc]
517
+ @overload
518
+ def split (self : LiteralString , sep : LiteralString | None = None , maxsplit : SupportsIndex = - 1 ) -> list [LiteralString ]: ...
519
+ @overload
461
520
def split (self , sep : str | None = None , maxsplit : SupportsIndex = - 1 ) -> list [str ]: ... # type: ignore[misc]
521
+ @overload
522
+ def splitlines (self : LiteralString , keepends : bool = False ) -> list [LiteralString ]: ...
523
+ @overload
462
524
def splitlines (self , keepends : bool = False ) -> list [str ]: ... # type: ignore[misc]
463
525
def startswith (
464
526
self , __prefix : str | tuple [str , ...], __start : SupportsIndex | None = ..., __end : SupportsIndex | None = ...
465
527
) -> bool : ...
528
+ @overload
529
+ def strip (self : LiteralString , __chars : LiteralString | None = None ) -> LiteralString : ...
530
+ @overload
466
531
def strip (self , __chars : str | None = None ) -> str : ... # type: ignore[misc]
532
+ @overload
533
+ def swapcase (self : LiteralString ) -> LiteralString : ...
534
+ @overload
467
535
def swapcase (self ) -> str : ... # type: ignore[misc]
536
+ @overload
537
+ def title (self : LiteralString ) -> LiteralString : ...
538
+ @overload
468
539
def title (self ) -> str : ... # type: ignore[misc]
469
540
def translate (self , __table : _TranslateTable ) -> str : ...
541
+ @overload
542
+ def upper (self : LiteralString ) -> LiteralString : ...
543
+ @overload
470
544
def upper (self ) -> str : ... # type: ignore[misc]
545
+ @overload
546
+ def zfill (self : LiteralString , __width : SupportsIndex ) -> LiteralString : ...
547
+ @overload
471
548
def zfill (self , __width : SupportsIndex ) -> str : ... # type: ignore[misc]
472
549
@staticmethod
473
550
@overload
@@ -478,20 +555,35 @@ class str(Sequence[str]):
478
555
@staticmethod
479
556
@overload
480
557
def maketrans (__x : str , __y : str , __z : str ) -> dict [int , int | None ]: ...
558
+ @overload
559
+ def __add__ (self : LiteralString , __s : LiteralString ) -> LiteralString : ...
560
+ @overload
481
561
def __add__ (self , __s : str ) -> str : ... # type: ignore[misc]
482
562
# Incompatible with Sequence.__contains__
483
563
def __contains__ (self , __o : str ) -> bool : ... # type: ignore[override]
484
564
def __eq__ (self , __x : object ) -> bool : ...
485
565
def __ge__ (self , __x : str ) -> bool : ...
486
566
def __getitem__ (self , __i : SupportsIndex | slice ) -> str : ...
487
567
def __gt__ (self , __x : str ) -> bool : ...
568
+ @overload
569
+ def __iter__ (self : LiteralString ) -> Iterator [LiteralString ]: ...
570
+ @overload
488
571
def __iter__ (self ) -> Iterator [str ]: ... # type: ignore[misc]
489
572
def __le__ (self , __x : str ) -> bool : ...
490
573
def __len__ (self ) -> int : ...
491
574
def __lt__ (self , __x : str ) -> bool : ...
575
+ @overload
576
+ def __mod__ (self : LiteralString , __x : LiteralString | tuple [LiteralString , ...]) -> LiteralString : ...
577
+ @overload
492
578
def __mod__ (self , __x : Any ) -> str : ... # type: ignore[misc]
579
+ @overload
580
+ def __mul__ (self : LiteralString , __n : SupportsIndex ) -> LiteralString : ...
581
+ @overload
493
582
def __mul__ (self , __n : SupportsIndex ) -> str : ... # type: ignore[misc]
494
583
def __ne__ (self , __x : object ) -> bool : ...
584
+ @overload
585
+ def __rmul__ (self : LiteralString , __n : SupportsIndex ) -> LiteralString : ...
586
+ @overload
495
587
def __rmul__ (self , __n : SupportsIndex ) -> str : ... # type: ignore[misc]
496
588
def __getnewargs__ (self ) -> tuple [str ]: ...
497
589
@@ -1634,11 +1726,11 @@ _SupportsSumNoDefaultT = TypeVar("_SupportsSumNoDefaultT", bound=_SupportsSumWit
1634
1726
# Instead, we special-case the most common examples of this: bool and literal integers.
1635
1727
if sys .version_info >= (3 , 8 ):
1636
1728
@overload
1637
- def sum (__iterable : Iterable [bool ], start : int = 0 ) -> int : ... # type: ignore[misc]
1729
+ def sum (__iterable : Iterable [bool | _LiteralInteger ], start : int = 0 ) -> int : ... # type: ignore[misc]
1638
1730
1639
1731
else :
1640
1732
@overload
1641
- def sum (__iterable : Iterable [bool ], __start : int = 0 ) -> int : ... # type: ignore[misc]
1733
+ def sum (__iterable : Iterable [bool | _LiteralInteger ], __start : int = 0 ) -> int : ... # type: ignore[misc]
1642
1734
1643
1735
@overload
1644
1736
def sum (__iterable : Iterable [_SupportsSumNoDefaultT ]) -> _SupportsSumNoDefaultT | Literal [0 ]: ...
0 commit comments