mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 00:48:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Improve pyrepl type-annotation coverage (#119081)

Browse source
This commit is contained in:
Alex Waygood 2024-05-17 06:13:24 -04:00 • committed by GitHub
parent 100c7ab00a
commit 033f5c87f1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 31 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

22
Lib/_pyrepl/unix_console.py
View file

@ -40,9 +40,13 @@ from .unix_eventqueue import EventQueue
from .utils import wlen
TYPE_CHECKING = False
# types
if False:
from typing import IO
if TYPE_CHECKING:
from typing import IO, Literal, overload
else:
overload = lambda func: None
class InvalidTerminal(RuntimeError):
@ -157,7 +161,13 @@ class UnixConsole(Console):
curses.setupterm(term or None, self.output_fd)
self.term = term
def _my_getstr(cap, optional=0):
@overload
def _my_getstr(cap: str, optional: Literal[False] = False) -> bytes: ...
@overload
def _my_getstr(cap: str, optional: bool) -> bytes | None: ...
def _my_getstr(cap: str, optional: bool = False) -> bytes | None:
r = curses.tigetstr(cap)
if not optional and r is None:
raise InvalidTerminal(
@ -672,18 +682,18 @@ class UnixConsole(Console):
elif dy < 0:
self.__write_code(self._cuu, -dy)
def __move_x_hpa(self, x):
def __move_x_hpa(self, x: int) -> None:
if x != self.__posxy[0]:
self.__write_code(self._hpa, x)
def __move_x_cub1_cuf1(self, x):
def __move_x_cub1_cuf1(self, x: int) -> None:
dx = x - self.__posxy[0]
if dx > 0:
self.__write_code(self._cuf1 * dx)
elif dx < 0:
self.__write_code(self._cub1 * (-dx))
def __move_x_cub_cuf(self, x):
def __move_x_cub_cuf(self, x: int) -> None:
dx = x - self.__posxy[0]
if dx > 0:
self.__write_code(self._cuf, dx)