diff options
| -rw-r--r-- | tools/free_space.py | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/tools/free_space.py b/tools/free_space.py index 4df8aec9d..b6847e8ea 100644 --- a/tools/free_space.py +++ b/tools/free_space.py @@ -19,13 +19,11 @@ def main(): print_bank = 'none' mapfile = 'pokecrystal.map' - if len(sys.argv) >= 2 and sys.argv[1].startswith('BANK='): - print_bank = sys.argv[1].split('=', 1)[-1] - if len(sys.argv) >= 3: - mapfile = sys.argv[2] - elif len(sys.argv) >= 3 and sys.argv[2].startswith('BANK='): - print_bank = sys.argv[2].split('=', 1)[-1] - mapfile = sys.argv[1] + for arg in sys.argv[1:]: + if arg.startswith('BANK='): + print_bank = arg.split('=', 1)[-1] + else: + mapfile = arg if print_bank not in {'all', 'none'}: try: |