input_handler can be specified from config

1 files changed, 13 insertions, 6 deletions

diff --git a/modules/input_handlers/stdin_input.py b/modules/input_handlers/stdin_input.py
index 044ac0d..5c50511 100644
--- a/modules/input_handlers/stdin_input.py
+++ b/modules/input_handlers/stdin_input.py
@@ -1,9 +1,16 @@
 import sys
 
-def get_input_stdin():
-    for line in sys.stdin:
-        line = line.strip()
-        if not line:
-            continue
+from modules.input_handlers.input_handler import InputHandler
 
-        yield line
\ No newline at end of file
+
+class StdinInput(InputHandler):
+    def cleanup(self):
+        pass
+
+    def get_input(self):
+        for line in sys.stdin:
+            line = line.strip()
+            if not line:
+                continue
+
+            yield line
\ No newline at end of file