boto3-assist 0.1.10__tar.gz → 0.1.12__tar.gz

{boto3_assist-0.1.10 → boto3_assist-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: boto3_assist
-Version: 0.1.10
+Version: 0.1.12
 Summary: Additional boto3 wrappers to make your life a little easier
 Author-email: Eric Wilson <boto3-assist@geekcafe.com>
 License-File: LICENSE-EXPLAINED.txt

{boto3_assist-0.1.10 → boto3_assist-0.1.12}/pyproject.toml

@@ -7,7 +7,7 @@ packages = ["src/boto3_assist"]
 
 [project]
 name = "boto3_assist"
-version = "0.1.10"
+version = "0.1.12"
 authors = [
   { name="Eric Wilson", email="boto3-assist@geekcafe.com" }
 ]

{boto3_assist-0.1.10 → boto3_assist-0.1.12}/src/boto3_assist/connection_tracker.py

@@ -15,12 +15,19 @@ class ConnectionTracker:
     """
 
     def __init__(self, service_name: str) -> None:
-        self.__stack_trace_env_var: str = "ISSUE_STACK_TRACE"
+        self.__stack_trace_env_var: str = "BOTO3_ASSIST_CONNECTION_STACK_TRACE"
         self.__connection_couter: int = 0
         self.__service_name: str = service_name
-        self.__issue_stack_trace: bool = (
-            os.getenv(f"{self.__stack_trace_env_var}", "false") == "true"
-        )
+        self.__issue_stack_trace: bool | None = None
+
+    @property
+    def issue_stack_trace(self) -> bool:
+        """Returns True if the stack trace should be issued"""
+        if self.__issue_stack_trace is None:
+            self.__issue_stack_trace = (
+                os.getenv(self.__stack_trace_env_var, "").lower() == "true"
+            )
+        return self.__issue_stack_trace
 
     def increment_connection(self) -> None:
         """Increments the connection counter"""
@@ -32,11 +39,18 @@ class ConnectionTracker:
             if self.__service_name:
                 service_message = f"Your {self.__service_name} service has more than one connection.\n"
 
-            if not self.__issue_stack_trace:
+            if not self.issue_stack_trace:
                 stack_trace_message = (
                     f"\nTo add addtional information to the log and determine where additional connections are being created"
                     f", set the environment variable {self.__stack_trace_env_var} to true.\n"
                 )
+            else:
+                stack = "\n".join(traceback.format_stack())
+                stack_trace_message = (
+                    f"\nStack Trace Enabeld with {self.__stack_trace_env_var}"
+                    f"\n{stack}"
+                )
+
             self.__log_warning(
                 f"{service_message}"
                 f"Your dynamodb connection count is {self.connection_count}. "
@@ -48,12 +62,6 @@ class ConnectionTracker:
                 f"{stack_trace_message}"
             )
 
-            # do a stack trace
-            if self.__issue_stack_trace:
-                print("Stack Trace")
-                traceback.print_stack()
-                print("")
-
     def decrement_connection(self) -> None:
         """Decrements the connection counter"""
         self.__connection_couter -= 1

boto3_assist-0.1.12/src/boto3_assist/dynamodb/dynamodb_connection_tracker.py

@@ -0,0 +1,17 @@
+"""
+Geek Cafe, LLC
+Maintainers: Eric Wilson
+MIT License.  See Project Root for the license information.
+"""
+
+from boto3_assist.connection_tracker import ConnectionTracker
+
+
+class DynamoDBConnectionTracker(ConnectionTracker):
+    """
+    Tracks DynamoDB Connection Requests.
+    Useful in for performance tuning and debugging.
+    """
+
+    def __init__(self) -> None:
+        super().__init__("DynamoDB")

boto3_assist-0.1.12/src/boto3_assist/version.py

@@ -0,0 +1 @@
+__version__ = '0.1.12'

boto3_assist-0.1.10/src/boto3_assist/dynamodb/dynamodb_connection_tracker.py

@@ -1,46 +0,0 @@
-"""
-Geek Cafe, LLC
-Maintainers: Eric Wilson
-MIT License.  See Project Root for the license information.
-"""
-
-
-class DynamoDBConnectionTracker:
-    """
-    Tracks DynamoDB Connection Requests.
-    Useful in for performance tuning and debugging.
-    """
-
-    def __init__(self) -> None:
-        self.__connection_couter: int = 0
-
-    def increment_connection(self) -> None:
-        """Increments the connection counter"""
-        self.__connection_couter += 1
-
-        if self.connection_count > 1:
-            self.__log_warning(
-                f"Your dynamodb connection count is {self.connection_count}. "
-                "Under most circumstances you should be able to use the same connection "
-                "vs. creating a new one.  Connections are expensive in terms of time / latency. "
-                "If you are seeing perforance issues, check how and where you are creating your "
-                "connections.  You should be able to pass the .db connection to your other objects "
-                "and reuse your dynamodb boto connections."
-            )
-
-    def decrement_connection(self) -> None:
-        """Decrements the connection counter"""
-        self.__connection_couter -= 1
-
-    @property
-    def connection_count(self) -> int:
-        """Returns the current connection count"""
-        return self.__connection_couter
-
-    def reset(self) -> None:
-        """Resets the connection counter"""
-        self.__connection_couter = 0
-
-    def __log_warning(self, message: str) -> None:
-        """Logs a warning message"""
-        print(f"Warning: {message}")

boto3_assist-0.1.10/src/boto3_assist/version.py

@@ -1 +0,0 @@
-__version__ = '0.1.10'