WIPpython

lisa/trace.py

@@ -425,6 +425,9 @@ def __init__(self, path, events, **kwargs):
         self._parquet_dir = tempfile.TemporaryDirectory()
         self._metadata = self._make_parquets(events=events, path=path, parquet_dir=self._parquet_dir.name)
 
+    def __enter__(self):
+        return super().__enter__()
+
     def __exit__(self, *args, **kwargs):
         self._parquet_dir.__exit__(*args, **kwargs)
 
@@ -3893,8 +3896,7 @@ def _get_metadata(self, key, parser=None):
         if parser is None:
             parser = self._get_parser(needed_metadata={key})
 
-        with parser as parser:
-            return parser.get_metadata(key)
+        return parser.get_metadata(key)
 
     def _get_cacheable_metadata(self, key, parser=None):
         try:
@@ -4502,7 +4504,8 @@ def get_missing(df_map):
                         continue
                     else:
                         try:
-                            meta_event_df = parser.parse_event(event)
+                            with parser as parser:
+                                meta_event_df = parser.parse_event(event)
                         except MissingTraceEventError:
                             continue
                             # In case a meta-event is spread among multiple