[IMP] Scripts with working stdin (#20)

* Pass script files without using rewriting stdin to allow working with stdin in scripts

* Bump version to 0.18.0

Author: fkantelberg

src/doblib/__init__.py

@@ -1 +1 @@
-VERSION = "0.17.2"
+VERSION = "0.18.0"

src/doblib/run.py

@@ -18,10 +18,19 @@ def load_shell_arguments(args):
         default=utils.get_config_file(),
         help="Configuration file to use. Default: %(default)s",
     )
-    parser.add_argument("file", nargs="?", help="File to execute")
+    parser.add_argument("file", default=None, nargs="?", help="File to execute")
     return parser.parse_known_args(args)
 
 
+def wrapped_console(script_file):
+    def console(local_vars):
+        local_vars["__name__"] = "__main__"
+        with open(script_file, "r", encoding="utf-8") as fp:
+            exec(fp.read(), local_vars)
+
+    return console
+
+
 class RunEnvironment(env.Environment):
     """Class to the environment"""
 
@@ -34,13 +43,12 @@ def shell(self, args=None):
         # pylint: disable=C0415,E0401
         from odoo.cli.shell import Shell
 
+        sys.argv = [args.file] + left if args.file else [""]
+        shell = Shell()
+
         if args.file:
-            sys.stdin = open(args.file, "r", encoding="utf-8")
-            sys.argv = [args.file] + left
-        else:
-            sys.argv = [""]
+            shell.console = wrapped_console(args.file)
 
-        shell = Shell()
         return shell.run(["-c", base.ODOO_CONFIG, "--no-http"])
 
     def start(self, args=None):

tests/test_run.py

@@ -40,9 +40,11 @@ def test_shell(env):
 
     with NamedTemporaryFile() as fp:
         shell.Shell.reset_mock()
+        console_mock = shell.Shell.return_value.console
         env.shell([fp.name])
         shell.Shell.assert_called_once()
         assert sys.argv == [fp.name]
+        assert console_mock != shell.Shell.return_value.console
 
 
 @mock.patch("doblib.utils.call", return_value=42)