yield remaining

Author: snopoke

taskbadger/process.py

@@ -34,6 +34,9 @@ def run(self):
                 else:
                     yield
 
+        if self.capture_output and (stdout or stderr):
+            yield {"stdout": stdout.read(), "stderr": stderr.read()}
+
         self.returncode = process.returncode
 
 
@@ -67,6 +70,9 @@ def read(self):
         self._lock.release()
         return outdata
 
+    def __bool__(self):
+        return bool(self.data)
+
 
 def _should_update(last_update: datetime, update_frequency_seconds):
     return (datetime.utcnow() - last_update).total_seconds() >= update_frequency_seconds