formatting fixes

Author: Victoria Hall

azure_functions_worker/dispatcher.py

@@ -140,6 +140,7 @@ def get_sync_tp_workers_set(self):
         """We don't know the exact value of the threadcount set for the Python
          3.9 scenarios (as we'll start passing only None by default), and we
          need to get that information.
+
          Ref: concurrent.futures.thread.ThreadPoolExecutor.__init__._max_workers
         """
         return self._sync_call_tp._max_workers
@@ -193,19 +194,19 @@ async def dispatch_forever(self):  # sourcery skip: swap-if-expression
 
             root_logger.setLevel(log_level)
             root_logger.addHandler(logging_handler)
-            logger.info("Switched to gRPC logging.")
+            logger.info('Switched to gRPC logging.')
             logging_handler.flush()
 
             try:
                 await forever
             finally:
-                logger.warning("Detaching gRPC logging due to exception.")
+                logger.warning('Detaching gRPC logging due to exception.')
                 logging_handler.flush()
                 root_logger.removeHandler(logging_handler)
 
                 # Reenable console logging when there's an exception
                 enable_console_logging()
-                logger.warning("Switched to console logging due to exception.")
+                logger.warning('Switched to console logging due to exception.')
         finally:
             DispatcherMeta.__current_dispatcher__ = None
 
@@ -235,12 +236,12 @@ def on_logging(self, record: logging.LogRecord,
         elif record.levelno >= logging.DEBUG:
             log_level = protos.RpcLog.Debug
         else:
-            log_level = getattr(protos.RpcLog, "None")
+            log_level = getattr(protos.RpcLog, 'None')
 
         if is_system_log_category(record.name):
-            log_category = protos.RpcLog.RpcLogCategory.Value("System")
+            log_category = protos.RpcLog.RpcLogCategory.Value('System')
         else:  # customers using logging will yield 'root' in record.name
-            log_category = protos.RpcLog.RpcLogCategory.Value("User")
+            log_category = protos.RpcLog.RpcLogCategory.Value('User')
 
         log = dict(
             level=log_level,
@@ -251,7 +252,7 @@ def on_logging(self, record: logging.LogRecord,
 
         invocation_id = get_current_invocation_id()
         if invocation_id is not None:
-            log["invocation_id"] = invocation_id
+            log['invocation_id'] = invocation_id
 
         self._grpc_resp_queue.put_nowait(
             protos.StreamingMessage(
@@ -270,21 +271,21 @@ def worker_id(self) -> str:
     @staticmethod
     def _serialize_exception(exc: Exception):
         try:
-            message = f"{type(exc).__name__}: {exc}"
+            message = f'{type(exc).__name__}: {exc}'
         except Exception:
             message = ('Unhandled exception in function. '
                        'Could not serialize original exception message.')
 
         try:
             stack_trace = marshall_exception_trace(exc)
         except Exception:
-            stack_trace = ""
+            stack_trace = ''
 
         return protos.RpcException(message=message, stack_trace=stack_trace)
 
     async def _dispatch_grpc_request(self, request):
-        content_type = request.WhichOneof("content")
-        request_handler = getattr(self, f"_handle__{content_type}", None)
+        content_type = request.WhichOneof('content')
+        request_handler = getattr(self, f'_handle__{content_type}', None)
         if request_handler is None:
             # Don't crash on unknown messages.  Some of them can be ignored;
             # and if something goes really wrong the host can always just
@@ -356,16 +357,16 @@ def update_opentelemetry_status(self):
     async def _handle__worker_init_request(self, request):
         worker_init_request = request.worker_init_request
         config_manager.set_config(
-            os.path.join(worker_init_request.function_app_directory, "az-config.json")
+            os.path.join(worker_init_request.function_app_directory, 'az-config.json')
         )
         logger.info(
-            "Received WorkerInitRequest, "
-            "python version %s, "
-            "worker version %s, "
-            "request ID %s. "
-            "App Settings state: %s. "
-            "To enable debug level logging, please refer to "
-            "https://aka.ms/python-enable-debug-logging",
+            'Received WorkerInitRequest, '
+            'python version %s, '
+            'worker version %s, '
+            'request ID %s. '
+            'App Settings state: %s. '
+            'To enable debug level logging, please refer to '
+            'https://aka.ms/python-enable-debug-logging',
             sys.version,
             VERSION,
             self.request_id,
@@ -441,7 +442,7 @@ def load_function_metadata(self, function_app_directory, caller_info):
         """
         script_file_name = config_manager.get_app_setting(
             setting=PYTHON_SCRIPT_FILE_NAME,
-            default_value=f"{PYTHON_SCRIPT_FILE_NAME_DEFAULT}")
+            default_value=f'{PYTHON_SCRIPT_FILE_NAME_DEFAULT}')
 
         logger.debug(
             'Received load metadata request from %s, request ID %s, '
@@ -469,7 +470,7 @@ async def _handle__functions_metadata_request(self, request):
                                      script_file_name)
 
         logger.info(
-            "Received WorkerMetadataRequest, request ID %s, " "function_path: %s",
+            'Received WorkerMetadataRequest, request ID %s, ' 'function_path: %s',
             self.request_id,
             function_path,
         )
@@ -659,7 +660,7 @@ async def _handle__invocation_request(self, request):
             # for a customer's threads
             fi_context.thread_local_storage.invocation_id = invocation_id
             if fi.requires_context:
-                args["context"] = fi_context
+                args['context'] = fi_context
 
             if fi.output_types:
                 for name in fi.output_types:
@@ -767,7 +768,7 @@ async def _handle__function_environment_reload_request(self, request):
                 os.environ[var] = env_vars[var]
             config_manager.set_config(
                 os.path.join(
-                    func_env_reload_request.function_app_directory, "az-config.json"
+                    func_env_reload_request.function_app_directory, 'az-config.json'
                 )
             )
 
@@ -799,7 +800,8 @@ async def _handle__function_environment_reload_request(self, request):
             if config_manager.is_envvar_true(PYTHON_ENABLE_INIT_INDEXING):
                 try:
                     self.load_function_metadata(
-                        directory, caller_info="environment_reload_request")
+                        directory,
+                        caller_info="environment_reload_request")
 
                     if HttpV2Registry.http_v2_enabled():
                         capabilities[HTTP_URI] = \
@@ -819,7 +821,8 @@ async def _handle__function_environment_reload_request(self, request):
             success_response = protos.FunctionEnvironmentReloadResponse(
                 capabilities=capabilities,
                 worker_metadata=self.get_worker_metadata(),
-                result=protos.StatusResult(status=protos.StatusResult.Success))
+                result=protos.StatusResult(
+                    status=protos.StatusResult.Success))
 
             return protos.StreamingMessage(
                 request_id=self.request_id,
@@ -855,7 +858,7 @@ def index_functions(self, function_path: str, function_dir: str):
                 func_binding_logs = fx_bindings_logs.get(func)
                 for binding in func.get_bindings():
                     deferred_binding_info = func_binding_logs.get(
-                        binding.name) \
+                        binding.name)\
                         if func_binding_logs.get(binding.name) else ""
                     indexed_function_bindings_logs.append((
                         binding.type, binding.name, deferred_binding_info))

azure_functions_worker/logging.py

@@ -13,6 +13,7 @@
 SDK_LOG_PREFIX = "azure.functions"
 SYSTEM_ERROR_LOG_PREFIX = "azure_functions_worker_errors"
 
+
 logger: logging.Logger = logging.getLogger(SYSTEM_LOG_PREFIX)
 error_logger: logging.Logger = (
     logging.getLogger(SYSTEM_ERROR_LOG_PREFIX))

tests/unittests/test_dispatcher.py

@@ -598,6 +598,7 @@ class TestDispatcherStein(testutils.AsyncTestCase):
     def setUp(self):
         self._ctrl = testutils.start_mockhost(
             script_root=DISPATCHER_STEIN_FUNCTIONS_DIR)
+        config_manager.clear_config()
 
     async def test_dispatcher_functions_metadata_request(self):
         """Test if the functions metadata response will be sent correctly

tests/unittests/test_extension.py

@@ -80,6 +80,7 @@ def setUp(self):
 
     def tearDown(self) -> None:
         os.environ.pop(PYTHON_ENABLE_WORKER_EXTENSIONS)
+
         self.mock_sys_path.stop()
         self.mock_sys_module.stop()
         self.mock_environ.stop()

tests/unittests/test_script_file_name.py

@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 import os
+
 from tests.utils import testutils
 
 from azure_functions_worker.constants import (

tests/unittests/test_utilities_dependency.py

@@ -6,9 +6,10 @@
 import unittest
 from unittest.mock import patch
 
+from tests.utils import testutils
+
 from azure_functions_worker.utils.dependency import DependencyManager
 from azure_functions_worker.utils.config_manager import config_manager
-from tests.utils import testutils
 
 
 class TestDependencyManager(unittest.TestCase):