Skip to content

Commit df10ddc

Browse files
Fix
1 parent d86a1e6 commit df10ddc

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

__main__.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@ def create_arg_parser():
1111
"""Create and initialize an argument parser object"""
1212
parser = ArgumentParser(description="Python to lua translator.")
1313
parser.add_argument("inputfilename", metavar="IF", type=str,
14-
help="A python script filename to translate it.")
14+
help="A python script filename to translate it.",
15+
nargs="?", default="")
1516

1617
parser.add_argument("--show-ast", help="Print python ast tree before code.",
1718
dest="show_ast", action="store_true")
@@ -28,6 +29,12 @@ def main():
2829
parser = create_arg_parser()
2930
argv = parser.parse_args()
3031

32+
if not argv.no_lua_init:
33+
print(Translator.get_luainit())
34+
35+
if argv.only_lua_init:
36+
return 0
37+
3138
input_filename = argv.inputfilename
3239
if not Path(input_filename).is_file():
3340
raise RuntimeError(
@@ -41,10 +48,7 @@ def main():
4148
raise RuntimeError("The input file is empty.")
4249

4350
translator = Translator(show_ast=argv.show_ast)
44-
lua_code = translator.translate(content)
45-
46-
if not argv.no_lua_init:
47-
print(Translator.get_luainit())
51+
lua_code = translator.translate(content)
4852

4953
if not argv.only_lua_init:
5054
print(lua_code)

0 commit comments

Comments
 (0)