Try and fix method capture

Author: Val Brodsky

libs/labelbox/src/labelbox/client.py

@@ -156,6 +156,12 @@ def enable_experimental(self) -> bool:
     def app_url(self) -> str:
         return self._request_client.app_url
 
+    def set_sdk_method(self, sdk_method: str):
+        self._request_client.sdk_method = sdk_method
+
+    def unset_sdk_method(self):
+        self._request_client.sdk_method = None
+
     def execute(
         self,
         query=None,

libs/labelbox/src/labelbox/pagination.py

@@ -1,16 +1,25 @@
 # Size of a single page in a paginated query.
 from abc import ABC, abstractmethod
-from typing import Any, Callable, Dict, List, Optional, Tuple, Type, Union
-
-from typing import TYPE_CHECKING
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Callable,
+    Dict,
+    List,
+    Optional,
+    Tuple,
+    Type,
+    Union,
+)
+
+from lbox.call_info import call_info_as_str
 
 if TYPE_CHECKING:
     from labelbox import Client
     from labelbox.orm.db_object import DbObject
 
 _PAGE_SIZE = 100
 
-
 class PaginatedCollection:
     """An iterable collection of database objects (Projects, Labels, etc...).
 
@@ -49,9 +58,11 @@ def __init__(
         self._fetched_all = False
         self._data: List[Dict[str, Any]] = []
         self._data_ind = 0
+        self._client = client
+        self._client.set_sdk_method(call_info_as_str())
 
         pagination_kwargs = {
-            "client": client,
+            "client": self._client,
             "obj_class": obj_class,
             "dereferencing": dereferencing,
             "experimental": experimental,
@@ -72,11 +83,13 @@ def __iter__(self):
     def __next__(self):
         if len(self._data) <= self._data_ind:
             if self._fetched_all:
+                self._client.unset_sdk_method()
                 raise StopIteration()
 
             page_data, self._fetched_all = self.paginator.get_next_page()
             self._data.extend(page_data)
             if len(page_data) == 0:
+                self._client.unset_sdk_method()
                 raise StopIteration()
 
         rval = self._data[self._data_ind]

libs/lbox-clients/src/lbox/request_client.py

@@ -39,7 +39,7 @@ def call_info():
     method_name = "Unknown"
     prefix = ""
     class_name = ""
-    skip_methods = ["wrapper", "__init__"]
+    skip_methods = ["wrapper", "__init__", "execute"]
     skip_classes = ["PaginatedCollection", "_CursorPagination", "_OffsetPagination"]
 
     try:
@@ -51,14 +51,16 @@ def call_info():
                 class_name = call_info.frame.f_locals.get(
                     "self", None
                 ).__class__.__name__
-
-                if method_name not in skip_methods and class_name not in skip_classes:
-                    if TEST_FILE_PATTERN.search(call_info.filename):
-                        prefix = "test:"
-                    else:
-                        if class_name == "NoneType":
-                            class_name = ""
-                        break
+                print(call_info.frame.f_locals)
+                import pdb; pdb.set_trace()
+                if method_name not in skip_methods:
+                    if class_name not in skip_classes:
+                        if TEST_FILE_PATTERN.search(call_info.filename):
+                            prefix = "test:"
+                        else:
+                            if class_name == "NoneType":
+                                class_name = ""
+                            break
 
     except Exception:
         pass