[output/i3bar] Add flush method

flush() terminates a single iteration of widget drawing.

see #23

bumblebee-status

@@ -3,11 +3,14 @@
 import sys
 import bumblebee.engine
 import bumblebee.config
+import bumblebee.output
 
 def main():
     config = bumblebee.config.Config(sys.argv[1:])
+    output = bumblebee.output.I3BarOutput()
     engine = bumblebee.engine.Engine(
-        config=config
+        config=config,
+        output=output,
     )
 
     engine.run()

bumblebee/engine.py

@@ -52,6 +52,7 @@ class Engine(object):
             for module in self._modules:
                 widgets += module.widgets()
             self._output.draw(widgets)
+            self._output.flush()
 
         self._output.stop()
 

bumblebee/modules/test.py

@@ -8,4 +8,7 @@ class Module(bumblebee.engine.Module):
     def __init__(self, engine):
         super(Module, self).__init__(engine)
 
+    def widgets(self):
+        return []
+
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4

bumblebee/output.py

@@ -29,4 +29,9 @@ class I3BarOutput(object):
             })
         sys.stdout.write(json.dumps(result))
 
+    def flush(self):
+        """Flushes output"""
+        sys.stdout.write(",\n")
+        sys.stdout.flush()
+
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4

runlint.sh

@@ -1,3 +1,3 @@
 #!/bin/sh
 
-find . -name "*.py"|xargs pylint --disable=R0903
+find . -name "*.py"|xargs pylint --disable=R0903,R0201

tests/test_i3baroutput.py

@@ -41,4 +41,10 @@ class TestI3BarOutput(unittest.TestCase):
         for res in result:
             self.assertEquals(res["full_text"], self.someWidget.text())
 
+    @mock.patch("sys.stdout", new_callable=StringIO)
+    def test_flush(self, stdout):
+        self.output.flush()
+        self.assertEquals(",\n", stdout.getvalue())
+
+
 # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4