Making the locking code a little cleaner

[doneit] / src / gtk_toolbox.py

diff --git a/src/gtk_toolbox.py b/src/gtk_toolbox.py
index 52d29c9..0780ad1 100644 (file)
--- a/src/gtk_toolbox.py
+++ b/src/gtk_toolbox.py
@@ -1,11 +1,21 @@
 #!/usr/bin/python
 
 import warnings
+import contextlib
 
 import gobject
 import gtk
 
 
+@contextlib.contextmanager
+def gtk_lock():
+       gtk.gdk.threads_enter()
+       try:
+               yield
+       finally:
+               gtk.gdk.threads_leave()
+
+
 class LoginWindow(object):
 
        def __init__(self, widgetTree):