Move screen stop/start logic into context manager.

As well as reducing duplication and adding screen size detection to
:pipeto and :pyshell, this ensures that the screen is always restarted,
resulting in cleaner error handling if an error occurs while the screen
is stopped.

1 files changed, 4 insertions, 10 deletions

diff --git a/alot/commands/globals.py b/alot/commands/globals.py
index e8721758..4e0ed506 100644
--- a/alot/commands/globals.py
+++ b/alot/commands/globals.py
@@ -284,13 +284,8 @@ class ExternalCommand(Command):
             d = threads.deferToThread(thread_code)
             d.addCallback(afterwards)
         else:
-            ui.mainloop.stop()
-            ret = thread_code()
-            ui.mainloop.start()
-
-            # make sure urwid renders its canvas at the correct size
-            ui.mainloop.screen_size = None
-            ui.mainloop.draw_screen()
+            with ui.paused():
+                ret = thread_code()
 
             afterwards(ret)
 
@@ -348,9 +343,8 @@ class PythonShellCommand(Command):
     repeatable = True
 
     def apply(self, ui):
-        ui.mainloop.stop()
-        code.interact(local=locals())
-        ui.mainloop.start()
+        with ui.paused():
+            code.interact(local=locals())
 
 
 @registerCommand(MODE, 'repeat')