Skip to content

Commit 91e845e

Browse files
committed
added function hooks for same naming with tucache
1 parent 9d6c0e2 commit 91e845e

2 files changed

Lines changed: 22 additions & 1 deletion

File tree

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@ web-cache/
55
pytest-web-cache/
66
*.sqlite
77
test.py
8+
*.db
9+
*.png
810

911
# --------------------------- General Python Template ----------------------------------------
1012

pytrackunit/sqlcache.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import sqlite3
44
import os.path
55
import os
6-
from datetime import datetime
6+
from datetime import datetime, timedelta
77
from pytrackunit.tucache import TuCache
88
from pytrackunit.tuiter import SqlIter, TuIter
99
from pytrackunit.helper import get_datetime
@@ -144,6 +144,7 @@ def __init__(self,auth=None,_dir=None,db_file="webdb.db",upstream_cache=None):
144144
self.web_db_path = db_file
145145
self._db = sqlite3.connect(self.web_db_path)
146146
self.cache = upstream_cache
147+
self.tdelta_end = None
147148
if self.cache is None:
148149
self.cache = TuCache(auth,_dir,verbose=True)
149150
if create_tables:
@@ -265,3 +266,21 @@ def get_errors_sql(self, veh_id, start_ts, end_ts, previter=None):
265266
(veh_id,start_ts,end_ts))),meta))
266267

267268
return previter, cnt
269+
def get_faults(self,veh_id,tdelta=None,previter=None):
270+
"""get_faults method"""
271+
if self.tdelta_end is None:
272+
end = datetime.now()
273+
else:
274+
end = self.tdelta_end
275+
end = end.replace(hour=0,minute=0,second=0,microsecond=0)
276+
if isinstance(tdelta,datetime):
277+
start = end+tdelta
278+
else:
279+
irange = int(tdelta)
280+
if irange <= 0:
281+
return []
282+
start = end-timedelta(days=irange)
283+
return self.get_errors(veh_id,start,end,previter)
284+
def get_faults_timedelta(self,veh_id,start,end,previter=None):
285+
"""get_faults_timedelta"""
286+
return self.get_errors(veh_id,start,end,previter)

0 commit comments

Comments
 (0)