Avoid failing on an empty script. (#66)

zooba · web-flow · commit 84790eae15c2 · 2025-05-02T22:24:29.000+01:00
Fixes #65
diff --git a/src/manage/scriptutils.py b/src/manage/scriptutils.py
@@ -142,7 +142,7 @@ def _read_script(cmd, script, encoding):
     except OSError as ex:
         raise LookupError(script) from ex
     with f:
-        first_line = next(f).rstrip()
+        first_line = next(f, "").rstrip()
         if first_line.startswith("#!"):
             try:
                 return _parse_shebang(cmd, first_line)
diff --git a/tests/test_scriptutils.py b/tests/test_scriptutils.py
@@ -33,6 +33,7 @@ def _fake_install(v, **kwargs):
 ]
 
 @pytest.mark.parametrize("script, expect", [
+    ("", None),
     ("#! /usr/bin/test1.0\n#! /usr/bin/test2.0\n", "1.0"),
     ("#! /usr/bin/test2.0\n#! /usr/bin/test1.0\n", "2.0"),
     ("#! /usr/bin/test1.0.exe\n#! /usr/bin/test2.0\n", "1.0"),

Original file line number	Diff line number	Diff line change
`@@ -33,6 +33,7 @@ def _fake_install(v, **kwargs):`
`33`	`33`	`]`
`34`	`34`
`35`	`35`	`@pytest.mark.parametrize("script, expect", [`
	`36`	`+ ("", None),`
`36`	`37`	`("#! /usr/bin/test1.0\n#! /usr/bin/test2.0\n", "1.0"),`
`37`	`38`	`("#! /usr/bin/test2.0\n#! /usr/bin/test1.0\n", "2.0"),`
`38`	`39`	`("#! /usr/bin/test1.0.exe\n#! /usr/bin/test2.0\n", "1.0"),`