httpdbg 1.2.1__tar.gz → 1.2.2__tar.gz

{httpdbg-1.2.1 → httpdbg-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: httpdbg
-Version: 1.2.1
+Version: 1.2.2
 Summary: A very simple tool to debug HTTP(S) client and server requests.
 Author-email: cle-b <cle@tictac.pm>
 License: Apache-2.0

{httpdbg-1.2.1 → httpdbg-1.2.2}/httpdbg/__init__.py

@@ -3,6 +3,6 @@ from httpdbg.hooks.all import httprecord
 from httpdbg.records import HTTPRecords
 
 
-__version__ = "1.2.1"
+__version__ = "1.2.2"
 
 __all__ = ["httprecord", "HTTPRecords"]

{httpdbg-1.2.1 → httpdbg-1.2.2}/httpdbg/hooks/generic.py

@@ -1,9 +1,10 @@
 # -*- coding: utf-8 -*-
+import builtins
+from collections.abc import Callable
 from contextlib import contextmanager
-import glob
+import functools
 import importlib
 import inspect
-from pathlib import Path
 import traceback
 from types import ModuleType
 from typing import Any
@@ -50,55 +51,83 @@ def set_hook_for_generic(records: HTTPRecords, method: Any):
 def hook_generic(
     records: HTTPRecords, initiators: Union[List[str], None] = None
 ) -> Generator[None, None, None]:
-    hooks = []
-
-    if not initiators:
-        initiators = []
-
-    for initiator in initiators:
-        logger().info(f"HOOK_GENERIC add initiator - {initiator}")
-        hooks += list_callables_from_package(records, initiator)
+    if initiators:
+        # we add a hook for a generic initiator only if the module is imported
+
+        hooks: List[Callable] = []
+        already_hooked: List[str] = []
+
+        original_builtin_import = builtins.__import__
+
+        def __hook__import__(
+            name,
+            globals=None,
+            locals=None,
+            fromlist=(),
+            level=0,
+            records=None,
+            initiators=None,
+            hooks=None,
+            already_hooked=None,
+        ):
+            if initiators is None:
+                raise Exception("initiators must not be None")
+            if hooks is None:
+                raise Exception("hooks must not be None")
+            if already_hooked is None:
+                raise Exception("already_hooked must not be None")
+
+            if (
+                name not in already_hooked
+            ):  # avoids parsing the same module multiple times
+                if (records is not None) and initiators:
+                    if (name in initiators) or (
+                        any(
+                            [
+                                name.startswith(f"{initiator}.")
+                                for initiator in initiators
+                            ]
+                        )
+                    ):
+                        logger().debug(f"HOOK IMPORT {name} - fromlist={fromlist}")
+                        already_hooked.append(name)
+                        hooks += list_callables_from_module(records, name)
+
+            # we temporarily restore the original built-in import function to avoid infinite recursion during import
+            __custom_import = builtins.__import__
+            builtins.__import__ = original_builtin_import
+            r = original_builtin_import(
+                name=name,
+                globals=globals,
+                locals=locals,
+                fromlist=fromlist,
+                level=level,
+            )
+            builtins.__import__ = __custom_import
+            return r
+
+        __hook__import_with_records_initiators_hooks_alreadyhooked__ = (
+            functools.partial(
+                __hook__import__,
+                records=records,
+                initiators=initiators,
+                hooks=hooks,
+                already_hooked=already_hooked,
+            )
+        )
 
-    try:
-        logger().info(
-            f"HOOK_GENERIC add initiator - hooks - {[hook.__httpdbg__.__name__ for hook in hooks]}"
+        builtins.__import__ = (
+            __hook__import_with_records_initiators_hooks_alreadyhooked__
         )
-    except Exception:
-        pass
 
     yield
 
-    if hooks:
-        for fnc in hooks:
-            fnc = undecorate(fnc)
+    if initiators:
+        builtins.__import__ == original_builtin_import
 
-
-def list_callables_from_package(records: HTTPRecords, package: str) -> List[Any]:
-    callables = []
-
-    try:
-        logger().info(f"LIST_CALLABLES_FROM_PACKAGE {package}")
-
-        callables += list_callables_from_module(records, package)
-
-        imported_package = importlib.import_module(package)
-        if hasattr(imported_package, "__path__"):
-            for p in glob.glob(f"{list(imported_package.__path__)[0]}/*"):
-                file_or_dir = Path(p)
-                if file_or_dir.is_file():
-                    if file_or_dir.name.endswith(".py"):
-                        callables += list_callables_from_module(
-                            records, f"{package}.{file_or_dir.name[:-3]}"
-                        )
-                elif file_or_dir.is_dir():
-                    if not file_or_dir.name.startswith("__"):  # __pycache__
-                        callables += list_callables_from_package(
-                            records, f"{package}.{file_or_dir.name}"
-                        )
-    except Exception as ex:
-        logger().info(f"LIST_CALLABLES_FROM_PACKAGE {package} - error - {str(ex)}")
-
-    return callables
+        if hooks:
+            for fnc in hooks:
+                fnc = undecorate(fnc)
 
 
 def list_callables_from_module(records: HTTPRecords, module: str) -> List[Any]:

{httpdbg-1.2.1 → httpdbg-1.2.2}/httpdbg/hooks/urllib3.py

@@ -52,7 +52,7 @@ def hook_urllib3(records: HTTPRecords) -> Generator[None, None, None]:
             )
         # v2
         if hasattr(urllib3, "request"):
-            urllib3.request = decorate(records, urllib3.request, set_hook_for_urllib3)
+            urllib3.request = decorate(records, urllib3.request, set_hook_for_urllib3)  # type: ignore[arg-type]
 
         hooks = True
     except ImportError:
@@ -70,4 +70,4 @@ def hook_urllib3(records: HTTPRecords) -> Generator[None, None, None]:
             )
         # v2
         if hasattr(urllib3, "request"):
-            urllib3.request = undecorate(urllib3.request)
+            urllib3.request = undecorate(urllib3.request)  # type: ignore[arg-type]

{httpdbg-1.2.1 → httpdbg-1.2.2}/httpdbg/hooks/utils.py

@@ -1,8 +1,13 @@
 # -*- coding: utf-8 -*-
+from collections.abc import Callable
 import inspect
+import typing
 
 from httpdbg.log import logger
 
+if typing.TYPE_CHECKING:
+    from httpdbg.records import HTTPRecords
+
 
 def getcallargs(original_method, *args, **kwargs):
     while hasattr(
@@ -28,12 +33,12 @@ def getcallargs(original_method, *args, **kwargs):
     return callargs
 
 
-def decorate(records, method, hook):
+def decorate(records: "HTTPRecords", method: Callable, hook: Callable):
     ori = method
     method = hook(records, method)
-    method.__httpdbg__ = ori
+    method.__httpdbg__ = ori  # type: ignore[attr-defined]
     return method
 
 
-def undecorate(method):
-    return method.__httpdbg__
+def undecorate(method: Callable):
+    return method.__httpdbg__  # type: ignore[attr-defined]

{httpdbg-1.2.1 → httpdbg-1.2.2}/httpdbg.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: httpdbg
-Version: 1.2.1
+Version: 1.2.2
 Summary: A very simple tool to debug HTTP(S) client and server requests.
 Author-email: cle-b <cle@tictac.pm>
 License: Apache-2.0