comparison 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
comparison
equal deleted inserted replaced
967:373e8ef6ee25 968:32a34a143c27
361 def initializeSites(session, directoryName): 361 def initializeSites(session, directoryName):
362 '''Initializes default site objects and Camera Views 362 '''Initializes default site objects and Camera Views
363 363
364 eg somedirectory/montreal/ contains intersection1, intersection2, etc. 364 eg somedirectory/montreal/ contains intersection1, intersection2, etc.
365 The site names would be somedirectory/montreal/intersection1, somedirectory/montreal/intersection2, etc.''' 365 The site names would be somedirectory/montreal/intersection1, somedirectory/montreal/intersection2, etc.'''
366 from os import listdir, path 366 from os import listdir, path, sep
367 sites = [] 367 sites = []
368 cameraViews = [] 368 cameraViews = []
369 names = listdir(directoryName) 369 names = listdir(directoryName)
370 cameraViewIdx = 1
370 for name in names: 371 for name in names:
371 if path.isdir(directoryName+'/'+name): 372 if path.isdir(directoryName+sep+name):
372 sites.append(Site(directoryName+'/'+name, None)) 373 sites.append(Site(directoryName+sep+name, None))
373 cameraViews.append(CameraView(-1, None, None, sites[-1], None)) 374 cameraViews.append(CameraView('view{}'.format(cameraViewIdx), None, sites[-1], None, None, None))
375 cameraViewIdx += 1
374 session.add_all(sites) 376 session.add_all(sites)
375 session.add_all(cameraViews) 377 session.add_all(cameraViews)
376 session.commit() 378 session.commit()
377 # TODO crawler for video files? 379 # TODO crawler for video files?