printer as context manager

bwhmather · Jan 23, 2015 · c01bf60 · c01bf60
1 parent 65900c5
commit c01bf60
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/linemode/base.py b/linemode/base.py
@@ -20,3 +20,9 @@ def run_commands(self, commands):
 
     def shutdown(self):
         pass
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_value, exc_traceback):
+        self.shutdown()