jyio/vistassh-py
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Add type hints for ExpectQ methods

Browse Source
This commit is contained in:
Jiang Yio 2025-01-14 22:42:44 -05:00
parent d261d5200e
commit c0d9cbfc42
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
autoproc.py
View File

@ -9,7 +9,7 @@ import contextlib
import logging import logging
from collections import namedtuple from collections import namedtuple
from typing import Optional, Union, Sequence, NamedTuple, Callable from typing import Any, Optional, Union, Sequence, NamedTuple, Callable, AsyncGenerator
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -28,7 +28,7 @@ class ExpectQ(object):
"""Clear or restore buffer""" """Clear or restore buffer"""
self.buffer = buffer self.buffer = buffer
clear = reset clear = reset
async def prompts(self, endl: str='\r\n', timeout_settle: Optional[float]=None, throw: bool=False): async def prompts(self, endl: str='\r\n', timeout_settle: Optional[float]=None, throw: bool=False) -> AsyncGenerator[tuple[Optional[str], Optional[int]], None]:
len_endl = len(endl) len_endl = len(endl)
while True: while True:
if (pos := self.buffer.rfind(endl)) >= 0: if (pos := self.buffer.rfind(endl)) >= 0:
@ -43,7 +43,7 @@ class ExpectQ(object):
if throw: if throw:
raise raise
yield None, None yield None, None
async def promptmatches(self, *mappings: Union[str, re.Pattern, tuple, list], endl: str='\r\n', timeout_settle: Optional[float]=None, throw: bool=False): async def promptmatches(self, *mappings: Union[str, re.Pattern, tuple, list], endl: str='\r\n', timeout_settle: Optional[float]=None, throw: bool=False) -> AsyncGenerator[tuple[Optional[ExpectMatch], Any], Optional[bool]]:
for i, mapping in enumerate(mappings): for i, mapping in enumerate(mappings):
try: try:
match mapping: match mapping: