Merge pull request #10 from JacobCallahan/main

Add basic context manager functionality
SatelliteQE · Aug 15, 2022 · c06b552 · c06b552
2 parents f344a97 + 951bfbe
commit c06b552
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/manifester/manifester.py b/manifester/manifester.py
@@ -323,3 +323,13 @@ def get_manifest(self):
                 subscription_data=sub,
             )
         return self.trigger_manifest_export()
+
+    def __enter__(self):
+        try:
+            return self.get_manifest()
+        except:
+            self.delete_subscription_allocation()
+            raise
+
+    def __exit__(self, *tb_args):
+        self.delete_subscription_allocation()