mirror of
https://github.com/Ahp06/SUMO_Emissions.git
synced 2024-11-22 03:26:30 +00:00
Add cmdline argument parser
This commit is contained in:
parent
801d090305
commit
1be34f6031
@ -1,4 +1,5 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
import argparse
|
||||||
import os
|
import os
|
||||||
import shutil
|
import shutil
|
||||||
import subprocess
|
import subprocess
|
||||||
@ -143,9 +144,19 @@ def dict_to_list(d):
|
|||||||
return [item for k in d for item in (k, d[k])]
|
return [item for k in d for item in (k, d[k])]
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument('osmfile', type=str)
|
||||||
|
parser.add_argument('--path', type=str, help='Where to generate the files')
|
||||||
|
parser.add_argument('--name', type=str, required=True, help='Name of the scenario to generate')
|
||||||
|
args = parser.parse_args()
|
||||||
|
generate_all(args.osmfile, args.path, args.name)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
if os.path.isdir('/tmp/scenario/foo'):
|
if os.path.isdir('/tmp/scenario/foo'):
|
||||||
shutil.rmtree('/tmp/scenario/foo')
|
shutil.rmtree('/tmp/scenario/foo')
|
||||||
path = '/tmp/scenario/'
|
path = '/tmp/scenario/'
|
||||||
osm = '/tmp/scenario/map.osm'
|
osm = '/tmp/scenario/map.osm'
|
||||||
generate_all(osm, path, 'foo')
|
generate_all(osm, path, 'foo')
|
||||||
|
# main()
|
||||||
|
Loading…
Reference in New Issue
Block a user