Added context manager

2022-06-24 14:50:14 +02:00 · 2022-06-24 14:50:14 +02:00 · 69f0a5454c
commit 69f0a5454c
parent 3395ddc9e3
1 changed files with 12 additions and 0 deletions
--- a/cas_wrapper.py
+++ b/cas_wrapper.py
@ -30,6 +30,18 @@ class CasWrapper:
            self._cas = local['cas']
            self._cas['login']()

+    def __enter__(self):
+        with local.env(
+            CAS_API_KEY=self._cas_api_key,
+            SIGNER_ID=self._cas_signer_id,
+        ):
+            self._cas = local['cas']
+            self._cas['login']()
+        return self
+
+    def __exit__(self, exc_type, value, traceback):
+        self._cas['logout']()
+
    def notarize(
            self,
            local_path: str,