mirror of
				https://github.com/Ahp06/SUMO_Emissions.git
				synced 2025-11-04 03:59:19 +00:00 
			
		
		
		
	Add cmdline argument parser
This commit is contained in:
		@@ -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()
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user