Mercurial Hosting > traffic-intelligence
comparison trafficintelligence/metadata.py @ 1073:8ab92ee3cbef
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() |