diff --git a/obswebsocket/core.py b/obswebsocket/core.py index c309c5d..b8940c4 100644 --- a/obswebsocket/core.py +++ b/obswebsocket/core.py @@ -211,6 +211,23 @@ def unregister(self, func, event=None): self.eventmanager.unregister(func, event) + def __enter__(self): + """ + Entry for a with statement + + :return: This object + """ + self.connect() + return self + + def __exit__(self): + """ + Exit for a with statement + + :return: Nothing + """ + self.disconnect() + class RecvThread(threading.Thread): def __init__(self, core):