diff options
-rw-r--r-- | virtual_table.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/virtual_table.py b/virtual_table.py index dd0139a..74a99b6 100644 --- a/virtual_table.py +++ b/virtual_table.py @@ -13,7 +13,7 @@ from bs4.element import NavigableString class JFRVirtualTable: def __parse_filepaths(self, prefix): - file_path = path.realpath(prefix + '.html') + file_path = path.realpath(prefix) tournament_path = path.dirname(file_path) tournament_prefix = path.splitext(path.basename(file_path))[0] @@ -300,8 +300,17 @@ if __name__ == '__main__': argument_parser = argparse.ArgumentParser( description='Fix display for virtual tables in JFR Pary result pages') + + def file_path(filepath): + filepath = unicode(filepath, sys.getfilesystemencoding()) + if path.isfile(filepath): + return filepath + else: + argument_parser.error('File %s does not exist' % filepath) + argument_parser.add_argument('path', metavar='PATH', - help='tournament path with JFR prefix') + help='tournament path with JFR prefix', + type=file_path) argument_parser.add_argument('-t', '--text', metavar='HEADER', default='Wirtualny stolik:', help='traveller header for virtual score') |