From 14863301a08589ecf4e854e76d2b59084bc24818 Mon Sep 17 00:00:00 2001
From: brettlangdon <me@brett.is>
Date: Sun, 14 Feb 2021 12:19:11 -0500
Subject: [PATCH] Have Span.start return itself

---
 sysaudit/__init__.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/sysaudit/__init__.py b/sysaudit/__init__.py
index ed93f27..34a454d 100644
--- a/sysaudit/__init__.py
+++ b/sysaudit/__init__.py
@@ -96,6 +96,11 @@ class Span:
             self.message("start", data)
             self.started = True
 
+        # Return `self` so we can explicitly call `start` with data in a context manager:
+        #     with sysaudit.Span("my.event").start(dict(start_only="data")) as span:
+        #         pass
+        return self
+
     def end(self, data=None):
         if not self.ended:
             self.message("end", data)