Mercurial Hosting > traffic-intelligence
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? |