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()