diff options
| -rwxr-xr-x | auto-auto-complete.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/auto-auto-complete.py b/auto-auto-complete.py index 84acdd9..ac0993e 100755 --- a/auto-auto-complete.py +++ b/auto-auto-complete.py @@ -778,11 +778,11 @@ def main(shell, output, source): mane! ''' if __name__ == '__main__': - if len(sys.argv) != 6: + if len(sys.argv) == 0: print("USAGE: auto-auto-complete SHELL --output OUTPUT_FILE --source SOURCE_FILE") exit(2) - shell = sys.argv[1] + shell = None output = None source = None @@ -802,12 +802,14 @@ if __name__ == '__main__': if old is not None: abort('Duplicate option: ' + option) - for arg in sys.argv[2:]: + for arg in sys.argv[1:]: if option is not None: if option in aliases: option = aliases[option] useopt(option, arg) option = None + elif (shell is None) and not arg.startswith('-'): + shell = arg else: if '=' in arg: useopt(arg[:index('=')], arg[index('=') + 1:]) |