From f2fa89b64812a928ddfdd914e55435e3e31a296e Mon Sep 17 00:00:00 2001 From: thiagoluisbecker Date: Fri, 26 May 2023 16:52:52 -0300 Subject: [PATCH] Added context managers to lockdraw --- lib/matplotlib/widgets.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/lib/matplotlib/widgets.py b/lib/matplotlib/widgets.py index 13f7287fe375..7518c2f4700f 100644 --- a/lib/matplotlib/widgets.py +++ b/lib/matplotlib/widgets.py @@ -62,6 +62,12 @@ def locked(self): """Return whether the lock is currently held by an owner.""" return self._owner is not None + def __enter__(self): + return self + + def __exit__(self, *args): + self.release(self) + class Widget: """