Mercurial > hg > nsaunier > traffic-intelligence

comparison trafficintelligence/metadata.py @ 1073:8ab92ee3cbef

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
minor
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 20 Jul 2018 16:23:56 -0400
parents 9d4a06f49cb8
children 8734742c08c0
comparison
equal deleted inserted replaced
1071:58994b08be42 1073:8ab92ee3cbef
1 from datetime import datetime, timedelta 1 from datetime import datetime, timedelta
2 from pathlib import Path
2 from os import path, listdir, sep 3 from os import path, listdir, sep
3 from math import floor 4 from math import floor
4 5
5 from numpy import zeros, loadtxt, array 6 from numpy import zeros, loadtxt, array
6 7
337 # eg bounding box depends on camera view, tracking configuration depends on camera view 338 # eg bounding box depends on camera view, tracking configuration depends on camera view
338 # results: sqlite 339 # results: sqlite
339 340
340 def createDatabase(filename): 341 def createDatabase(filename):
341 'creates a session to query the filename' 342 'creates a session to query the filename'
342 engine = create_engine('sqlite:///'+filename) 343 if Path(filename).is_file():
343 Base.metadata.create_all(engine) 344 engine = create_engine('sqlite:///'+filename)
344 Session = sessionmaker(bind=engine) 345 Base.metadata.create_all(engine)
345 return Session() 346 Session = sessionmaker(bind=engine)
347 return Session()
348 else:
349 print('The file '+filename+' does not exist')
350 return None
346 351
347 def connectDatabase(filename): 352 def connectDatabase(filename):
348 'creates a session to query the filename' 353 'creates a session to query the filename'
349 engine = create_engine('sqlite:///'+filename) 354 if Path(filename).is_file():
350 Session = sessionmaker(bind=engine) 355 engine = create_engine('sqlite:///'+filename)
351 return Session() 356 Session = sessionmaker(bind=engine)
357 return Session()
358 else:
359 print('The file '+filename+' does not exist')
360 return None
352 361
353 def getSite(session, siteId = None, name = None, description = None): 362 def getSite(session, siteId = None, name = None, description = None):
354 'Returns the site(s) matching the index or the name' 363 'Returns the site(s) matching the index or the name'
355 if siteId is not None: 364 if siteId is not None:
356 return session.query(Site).filter(Site.idx == int(siteId)).all() 365 return session.query(Site).filter(Site.idx == int(siteId)).all()