diff options
-rw-r--r-- | quick_lineup.py | 31 |
1 files changed, 18 insertions, 13 deletions
diff --git a/quick_lineup.py b/quick_lineup.py index 237bc19..965922e 100644 --- a/quick_lineup.py +++ b/quick_lineup.py @@ -2,18 +2,23 @@ import sys from ql.console import Console -if len(sys.argv) < 3 or len(sys.argv) > 4: - print('Give correct parameters: round, segment and (optionally) table') - sys.exit(1) +def main(): + if len(sys.argv) < 3 or len(sys.argv) > 4: + print('Give correct parameters: round, segment and (optionally) table') + sys.exit(1) -round = int(sys.argv[1]) -segment = int(sys.argv[2]) -if len(sys.argv) == 4: - table = int(sys.argv[3]) -else: - table = None + round = int(sys.argv[1]) + segment = int(sys.argv[2]) + if len(sys.argv) == 4: + table = int(sys.argv[3]) + else: + table = None -try: - Console(round, segment, table).run() -except: - print('ERROR: %s' % sys.exc_info()[1]) + try: + Console(round, segment, table).run() + except: + print('ERROR: %s' % sys.exc_info()[1]) + + +if __name__ == '__main__': + main() |