Merge pull request #1003 from JLMSC/main

sebastinas · web-flow · commit 5db3e08caf54 · 2023-11-28T11:34:48.000+01:00
Added functions return typing in some files: __init__.py; keys.py and lazyre.py
diff --git a/bpython/__init__.py b/bpython/__init__.py
@@ -21,6 +21,7 @@
 # THE SOFTWARE.
 
 import os.path
+from typing import Any
 
 try:
     from ._version import __version__ as version  # type: ignore
@@ -36,7 +37,7 @@
 package_dir = os.path.abspath(os.path.dirname(__file__))
 
 
-def embed(locals_=None, args=None, banner=None):
+def embed(locals_=None, args=None, banner=None) -> Any:
     if args is None:
         args = ["-i", "-q"]
 
diff --git a/bpython/keys.py b/bpython/keys.py
@@ -42,10 +42,10 @@ def __getitem__(self, key: str) -> T:
                 f"Configured keymap ({key}) does not exist in bpython.keys"
             )
 
-    def __delitem__(self, key: str):
+    def __delitem__(self, key: str) -> None:
         del self.map[key]
 
-    def __setitem__(self, key: str, value: T):
+    def __setitem__(self, key: str, value: T) -> None:
         self.map[key] = value
 
 
diff --git a/bpython/lazyre.py b/bpython/lazyre.py
@@ -21,7 +21,7 @@
 # THE SOFTWARE.
 
 import re
-from typing import Optional, Pattern, Match, Optional
+from typing import Optional, Pattern, Match, Optional, Iterator
 
 try:
     from functools import cached_property
@@ -43,7 +43,7 @@ def __init__(self, regex: str, flags: int = 0) -> None:
     def compiled(self) -> Pattern[str]:
         return re.compile(self.regex, self.flags)
 
-    def finditer(self, *args, **kwargs):
+    def finditer(self, *args, **kwargs) -> Iterator[Match[str]]:
         return self.compiled.finditer(*args, **kwargs)
 
     def search(self, *args, **kwargs) -> Optional[Match[str]]:

Original file line number	Diff line number	Diff line change
`@@ -42,10 +42,10 @@ def __getitem__(self, key: str) -> T:`
`42`	`42`	`f"Configured keymap ({key}) does not exist in bpython.keys"`
`43`	`43`	`)`
`44`	`44`
`45`		`- def __delitem__(self, key: str):`
	`45`	`+ def __delitem__(self, key: str) -> None:`
`46`	`46`	`del self.map[key]`
`47`	`47`
`48`		`- def __setitem__(self, key: str, value: T):`
	`48`	`+ def __setitem__(self, key: str, value: T) -> None:`
`49`	`49`	`self.map[key] = value`
`50`	`50`
`51`	`51`