Skip to content

Commit 78570cd

Browse files
committed
add debug
1 parent e54d98e commit 78570cd

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

src/iterative_telemetry/__init__.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,26 +35,37 @@ def __init__(
3535
enabled: Union[bool, Callable] = True,
3636
url=URL,
3737
token=TOKEN,
38+
debug: bool = False,
3839
):
3940
self.tool_name = tool_name
4041
self.tool_version = tool_version
4142
self.enabled = enabled
4243
self.url = url
4344
self.token = token
45+
self.debug = debug
46+
if self.debug:
47+
logger.setLevel(logging.DEBUG)
48+
logger.debug("IterativeTelemetryLogger is in debug mode")
4449

45-
def send_cli_call(self, cmd_name: str, **kwargs):
46-
self.send_event("cli", cmd_name, **kwargs)
50+
def send_cli_call(self, cmd_name: str, error: str = None, **kwargs):
51+
self.send_event("cli", cmd_name, error=error, **kwargs)
4752

4853
def send_event(
4954
self,
5055
interface: str,
5156
action: str,
57+
error: str = None,
5258
use_thread: bool = False,
5359
use_daemon: bool = True,
5460
**kwargs,
5561
):
5662
self.send(
57-
{"interface": interface, "action": action, "extra": kwargs},
63+
{
64+
"interface": interface,
65+
"action": action,
66+
"error": error,
67+
"extra": kwargs,
68+
},
5869
use_thread=use_thread,
5970
use_daemon=use_daemon,
6071
)
@@ -80,6 +91,7 @@ def send(
8091
raise ValueError(
8192
"use_thread and use_daemon cannot be true at the same time"
8293
)
94+
logger.debug("Sending payload %s", payload)
8395
impl = self._send
8496
if use_daemon:
8597
impl = self._send_daemon

0 commit comments

Comments
 (0)