Commit 0baf1624addfef18c0c8b5f2d65017d803c0ad35
1 parent
37c010cba0
Exists in
master
Better main organisation of the volia.data module
Showing 1 changed file with 22 additions and 19 deletions Side-by-side Diff
volia/data.py
| ... | ... | @@ -5,29 +5,10 @@ |
| 5 | 5 | |
| 6 | 6 | import core.data |
| 7 | 7 | |
| 8 | -# Main parser | |
| 9 | -parser = argparse.ArgumentParser(description="manage files") | |
| 10 | -subparsers = parser.add_subparsers(title="actions") | |
| 11 | 8 | |
| 12 | -# Filter ids | |
| 13 | -parser_filter = subparsers.add_parser("filter", help="filter a file") | |
| 14 | -parser_filter.add_argument("--file", type=str, help="path of the file to filter", required=True) | |
| 15 | -parser_filter.add_argument("--filter", type=str, help="filter file [id_from] [id_to]", required=True) | |
| 16 | -parser_filter.add_argument("--outfile", type=str, help="output file", required=True) | |
| 17 | -parser_filter.set_defaults(which="filter") | |
| 18 | 9 | |
| 19 | -# Convert | |
| 20 | -parser_convert = subparsers.add_parser("convert", help="convert a file") | |
| 21 | -parser_convert.add_argument("--file", type=str, help="...", required=True) | |
| 22 | -parser_convert.add_argument("--type-from", type=str, choices=["old-masseffect", "new-masseffect"], help="...", required=True) | |
| 23 | -parser_convert.add_argument("--type", choices=["old-masseffect", "new-masseffect"], required=True) | |
| 24 | -parser_convert.set_defaults(which="convert") | |
| 25 | 10 | |
| 26 | -# Parse | |
| 27 | -args = parser.parse_args() | |
| 28 | 11 | |
| 29 | - | |
| 30 | - | |
| 31 | 12 | def filter_file(file, filter, outfile): |
| 32 | 13 | file_path = file |
| 33 | 14 | filter_path = filter |
| ... | ... | @@ -47,6 +28,28 @@ |
| 47 | 28 | |
| 48 | 29 | |
| 49 | 30 | if __name__ == "__main__": |
| 31 | + # Main parser | |
| 32 | + parser = argparse.ArgumentParser(description="manage files") | |
| 33 | + subparsers = parser.add_subparsers(title="actions") | |
| 34 | + | |
| 35 | + # Filter ids | |
| 36 | + parser_filter = subparsers.add_parser("filter", help="filter a file") | |
| 37 | + parser_filter.add_argument("--file", type=str, help="path of the file to filter", required=True) | |
| 38 | + parser_filter.add_argument("--filter", type=str, help="filter file [id_from] [id_to]", required=True) | |
| 39 | + parser_filter.add_argument("--outfile", type=str, help="output file", required=True) | |
| 40 | + parser_filter.set_defaults(which="filter") | |
| 41 | + | |
| 42 | + # Convert | |
| 43 | + parser_convert = subparsers.add_parser("convert", help="convert a file") | |
| 44 | + parser_convert.add_argument("--file", type=str, help="...", required=True) | |
| 45 | + parser_convert.add_argument("--type-from", type=str, choices=["old-masseffect", "new-masseffect"], help="...", required=True) | |
| 46 | + parser_convert.add_argument("--type", choices=["old-masseffect", "new-masseffect"], required=True) | |
| 47 | + parser_convert.set_defaults(which="convert") | |
| 48 | + | |
| 49 | + # Parse | |
| 50 | + args = parser.parse_args() | |
| 51 | + | |
| 52 | + # Run commands | |
| 50 | 53 | runner = SubCommandRunner({ |
| 51 | 54 | "convert" : convert, |
| 52 | 55 | "filter": filter_file, |