Mercurial Hosting > traffic-intelligence
diff python/metadata.py @ 968:32a34a143c27
work on sumo and metadata
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 07 Dec 2017 15:13:06 -0500 |
parents | 5d9899504977 |
children | 5d788d2e8ffc |
line wrap: on
line diff
--- a/python/metadata.py Tue Dec 05 16:15:44 2017 -0500 +++ b/python/metadata.py Thu Dec 07 15:13:06 2017 -0500 @@ -363,14 +363,16 @@ eg somedirectory/montreal/ contains intersection1, intersection2, etc. The site names would be somedirectory/montreal/intersection1, somedirectory/montreal/intersection2, etc.''' - from os import listdir, path + from os import listdir, path, sep sites = [] cameraViews = [] names = listdir(directoryName) + cameraViewIdx = 1 for name in names: - if path.isdir(directoryName+'/'+name): - sites.append(Site(directoryName+'/'+name, None)) - cameraViews.append(CameraView(-1, None, None, sites[-1], None)) + if path.isdir(directoryName+sep+name): + sites.append(Site(directoryName+sep+name, None)) + cameraViews.append(CameraView('view{}'.format(cameraViewIdx), None, sites[-1], None, None, None)) + cameraViewIdx += 1 session.add_all(sites) session.add_all(cameraViews) session.commit()