From a1e4b1948642738b571a24e4354257e296f50d20 Mon Sep 17 00:00:00 2001 From: Kovid Goyal Date: Thu, 10 Mar 2022 06:34:50 +0530 Subject: [PATCH] ... --- kitty/shm.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/kitty/shm.py b/kitty/shm.py index 6fafa9457..279de1eb8 100644 --- a/kitty/shm.py +++ b/kitty/shm.py @@ -63,10 +63,20 @@ class SharedMemory: except OSError: pass + def __enter__(self) -> 'SharedMemory': + return self + + def __exit__(self, *a: object) -> None: + self.close() + @property def name(self) -> str: return self._name + @property + def fileno(self) -> int: + return self._fd + @property def buf(self) -> memoryview: ans = self._buf