Skip to content

Commit 1085490

Browse files
committed
Add context manager and deconstructor to machine
1 parent 85c9125 commit 1085490

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

pylabrobot/machine.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,3 +53,14 @@ async def setup(self):
5353
async def stop(self):
5454
await self.backend.stop()
5555
self._setup_finished = False
56+
57+
def __del__(self):
58+
if self.setup_finished:
59+
self.stop()
60+
61+
def __enter__(self):
62+
self.setup()
63+
return self
64+
65+
def __exit__(self, exc_type, exc_value, traceback):
66+
self.stop()

0 commit comments

Comments
 (0)