From 19ec0f061f0f30625e53e07448f5f044caa1a1d5 Mon Sep 17 00:00:00 2001 From: brettlangdon Date: Sun, 14 Feb 2021 09:46:48 -0500 Subject: [PATCH] Allow assigning persistent Span data --- sysaudit/__init__.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/sysaudit/__init__.py b/sysaudit/__init__.py index 989c6bf..ed93f27 100644 --- a/sysaudit/__init__.py +++ b/sysaudit/__init__.py @@ -63,7 +63,7 @@ def subscribe(event, hook): class Span: - __slots__ = ("name", "started", "ended") + __slots__ = ("name", "started", "ended", "data") class Message: __slots__ = ("type", "span", "data") @@ -78,10 +78,11 @@ class Span: self.__class__.__name__, self.type, self.span, self.data ) - def __init__(self, name): + def __init__(self, name, data=None): self.name = name self.started = False self.ended = False + self.data = data @property def id(self): @@ -111,6 +112,9 @@ class Span: self.end(data=dict(exc_type=exc_type, exc_val=exc_val, exc_tb=exc_tb)) def __str__(self): - return "{0}(name={1!r}, id={2!r})".format( - self.__class__.__name__, self.name, self.id + return "{0}(name={1!r}, id={2!r}, data={3!r})".format( + self.__class__.__name__, + self.name, + self.id, + self.data, )