Fixing the types

Author: vrdmr

azure_functions_worker/bindings/generic.py

@@ -3,7 +3,6 @@
 import typing
 
 from . import datumdef
-from azure_functions_worker.bindings.datumdef import Datum
 from typing import Any, Optional
 
 
@@ -22,7 +21,8 @@ def check_output_type_annotation(cls, pytype: type) -> bool:
         return issubclass(pytype, (str, bytes, bytearray))
 
     @classmethod
-    def encode(cls, obj: typing.Any, *, expected_type: typing.Optional[type]) -> datumdef.Datum:
+    def encode(cls, obj: Any, *,
+               expected_type: Optional[type]) -> datumdef.Datum:
         if isinstance(obj, str):
             return datumdef.Datum(type='string', value=obj)
 

azure_functions_worker/dispatcher.py

@@ -49,8 +49,10 @@ class Dispatcher(metaclass=DispatcherMeta):
 
     _GRPC_STOP_RESPONSE = object()
 
-    def __init__(self, loop: _UnixSelectorEventLoop, host: str, port: int, worker_id: str, request_id: str,
-                 grpc_connect_timeout: float, grpc_max_msg_len: int = -1) -> None:
+    def __init__(self, loop: _UnixSelectorEventLoop, host: str, port: int,
+                 worker_id: str, request_id: str,
+                 grpc_connect_timeout: float,
+                 grpc_max_msg_len: int = -1) -> None:
         self._loop = loop
         self._host = host
         self._port = port
@@ -92,8 +94,8 @@ def load_bindings() -> Dict[Any, Any]:
         return services
 
     @classmethod
-    async def connect(cls, host: str, port: int, worker_id: str, request_id: str,
-                      connect_timeout: float):
+    async def connect(cls, host: str, port: int, worker_id: str,
+                      request_id: str, connect_timeout: float):
         loop = asyncio.events.get_event_loop()
         disp = cls(loop, host, port, worker_id, request_id, connect_timeout)
         disp._grpc_thread.start()

azure_functions_worker/loader.py

@@ -14,16 +14,16 @@
 
 from .constants import MODULE_NOT_FOUND_TS_URL
 from .utils.wrappers import attach_message_to_exception
-from os import PathLike
+from os import PathLike, fspath
 
 
 _AZURE_NAMESPACE = '__app__'
 
 _submodule_dirs = []
 
 
-def register_function_dir(path: os.PathLike) -> None:
-    _submodule_dirs.append(os.fspath(path))
+def register_function_dir(path: PathLike) -> None:
+    _submodule_dirs.append(fspath(path))
 
 
 def install() -> None:

azure_functions_worker/testutils.py

@@ -458,17 +458,18 @@ async def __aenter__(self):
 
         await self._host.start()
 
-        self._worker = await dispatcher.Dispatcher.connect('127.0.0.1',
-                       self._host._port, self._host.worker_id,
-                       self._host.request_id, connect_timeout=5.0)
+        self._worker = await dispatcher. \
+            Dispatcher.connect('127.0.0.1', self._host._port,
+                               self._host.worker_id,
+                               self._host.request_id, connect_timeout=5.0)
 
         self._worker.load_bindings()
 
         self._worker_task = loop.create_task(self._worker.dispatch_forever())
 
-        done, pending = await asyncio.wait([self._host._connected_fut,
-                                            self._worker_task],
-                                            return_when=asyncio.FIRST_COMPLETED)
+        done, pending = await asyncio. \
+            wait([self._host._connected_fut, self._worker_task],
+                 return_when=asyncio.FIRST_COMPLETED)
 
         try:
             if self._worker_task in done:

azure_functions_worker/utils/tracing.py

@@ -2,7 +2,7 @@
 # Licensed under the MIT License.
 from typing import List
 import traceback
-from traceback import StackSummary
+from traceback import StackSummary, extract_tb
 
 
 def extend_exception_message(exc: Exception, msg: str) -> Exception:
@@ -17,28 +17,23 @@ def extend_exception_message(exc: Exception, msg: str) -> Exception:
 
 
 def marshall_exception_trace(exc: Exception) -> str:
-    stack_summary: traceback.StackSummary = traceback.extract_tb(
-        exc.__traceback__)
+    stack_summary: StackSummary = extract_tb(exc.__traceback__)
     if isinstance(exc, ModuleNotFoundError):
         stack_summary = _marshall_module_not_found_error(stack_summary)
     return ''.join(stack_summary.format())
 
 
-def _marshall_module_not_found_error(
-    tbss: traceback.StackSummary
-) -> traceback.StackSummary:
+def _marshall_module_not_found_error(tbss: StackSummary) -> StackSummary:
     tbss = _remove_frame_from_stack(tbss, '<frozen importlib._bootstrap>')
     tbss = _remove_frame_from_stack(
         tbss, '<frozen importlib._bootstrap_external>')
     return tbss
 
 
-def _remove_frame_from_stack(
-    tbss: traceback.StackSummary,
-    framename: str
-) -> traceback.StackSummary:
-    filtered_stack_list: List[traceback.FrameSummary] = list(
-        filter(lambda frame: getattr(frame, 'filename') != framename, tbss))
-    filtered_stack: traceback.StackSummary = traceback.StackSummary.from_list(
-        filtered_stack_list)
+def _remove_frame_from_stack(tbss: StackSummary,
+                             framename: str) -> StackSummary:
+    filtered_stack_list: List[traceback.FrameSummary] = \
+        list(filter(lambda frame: getattr(frame,
+                                          'filename') != framename, tbss))
+    filtered_stack: StackSummary = StackSummary.from_list(filtered_stack_list)
     return filtered_stack

azure_functions_worker/utils/wrappers.py

@@ -5,7 +5,7 @@
 from typing import Callable, Optional
 
 
-def enable_feature_by(flag: str, default: Optional[int]=None) -> Callable:
+def enable_feature_by(flag: str, default: Optional[int] = None) -> Callable:
     def decorate(func):
         def call(*args, **kwargs):
             if is_envvar_true(flag):
@@ -15,7 +15,7 @@ def call(*args, **kwargs):
     return decorate
 
 
-def disable_feature_by(flag: str, default: None=None) -> Callable:
+def disable_feature_by(flag: str, default: None = None) -> Callable:
     def decorate(func):
         def call(*args, **kwargs):
             if not is_envvar_true(flag):