Add some return types

sebastinas · sebastinas · commit 2ba5de04ef86 · 2021-11-07T19:33:10.000+01:00
diff --git a/bpython/inspection.py b/bpython/inspection.py
@@ -267,7 +267,7 @@ def getfuncprops(func, f):
     return fprops
 
 
-def is_eval_safe_name(string):
+def is_eval_safe_name(string: str) -> bool:
     return all(
         part.isidentifier() and not keyword.iskeyword(part)
         for part in string.split(".")
@@ -334,7 +334,7 @@ def get_argspec_from_signature(f):
 get_encoding_line_re = LazyReCompile(r"^.*coding[:=]\s*([-\w.]+).*$")
 
 
-def get_encoding(obj):
+def get_encoding(obj) -> str:
     """Try to obtain encoding information of the source of an object."""
     for line in inspect.findsource(obj)[0][:2]:
         m = get_encoding_line_re.search(line)
@@ -354,7 +354,7 @@ def get_encoding_file(fname: str) -> str:
     return "utf8"
 
 
-def getattr_safe(obj: Any, name: str):
+def getattr_safe(obj: Any, name: str) -> Any:
     """side effect free getattr (calls getattr_static)."""
     result = inspect.getattr_static(obj, name)
     # Slots are a MemberDescriptorType
@@ -371,6 +371,6 @@ def hasattr_safe(obj: Any, name: str) -> bool:
         return False
 
 
-def get_source_unicode(obj):
+def get_source_unicode(obj) -> str:
     """Returns a decoded source of object"""
     return inspect.getsource(obj)
