Configure and use black

.vscode/settings.json

@@ -1,3 +1,10 @@
 {
-    "files.trimTrailingWhitespace": true
+    "files.trimTrailingWhitespace": true,
+    "[python]": {
+        "editor.defaultFormatter": "ms-python.black-formatter"
+    },
+    "editor.formatOnSave": true,
+    "black-formatter.args": [
+        "--line-length=160"
+    ]
 }

executiontime/__init__.py

@@ -1,6 +1,7 @@
 """
 Defines the printexecutiontime decorator
 """
+
 from datetime import datetime
 from functools import wraps
 from typing import Any, Callable, Optional
@@ -24,20 +25,23 @@
 WHITE = Fore.WHITE
 YELLOW = Fore.YELLOW
 
-def printexecutiontime(message: str, output: Callable[..., None] =print, color:Optional[str]=None)->Any:
-    '''
+
+def printexecutiontime(message: str, output: Callable[..., None] = print, color: Optional[str] = None) -> Any:
+    """
     This function returns a decorator. This allows to have a decorator that accepts parameters.
     message: A string with a '{0}' placeholder for the time that will be sent to the console.
-    '''
-    def decorator(function: Callable[..., Any])->Any:
-        '''
+    """
+
+    def decorator(function: Callable[..., Any]) -> Any:
+        """
         The decorator itself returns a wrapper function that will replace the original one.
-        '''
+        """
+
         @wraps(function)
-        def wrapper(*args: Any, **kwargs: Any)->Any:
-            '''
+        def wrapper(*args: Any, **kwargs: Any) -> Any:
+            """
             This wrapper calculates and displays the execution time of the function.
-            '''
+            """
             start = datetime.utcnow()
             value = function(*args, **kwargs)
             elapsed = datetime.utcnow() - start
@@ -46,5 +50,7 @@ def wrapper(*args: Any, **kwargs: Any)->Any:
                 msg = color + msg + Fore.RESET
             output(msg)
             return value
+
         return wrapper
+
     return decorator