added initialization of sqlite

Author: Marius Schlueter

pytrackunit/tucache.py

@@ -1,12 +1,33 @@
 """tucache module"""
 
+import sqlite3
+import os.path
 from .webcache import WebCache
 
 class TuCache:
     """tucache class"""
-    def __init__(self,auth=None,_dir=None,verbose=False):
+    def __init__(self,auth=None,_dir=None,use_sqlite=False,verbose=False):
         self.cache = WebCache(auth=auth,_dir=_dir,verbose=verbose)
-        self._verbose = verbose
+        self.use_sqlite = use_sqlite
+        if self.use_sqlite:
+            self.web_db_path = os.path.join(_dir,"webdb.db")
+            self._db = sqlite3.connect(self.web_db_path)
+            cur = self._db.cursor()
+            _res = list(cur.execute(
+                '''
+                SELECT name FROM sqlite_master WHERE type="table" AND name="history"
+                '''))
+            if len(_res) == 0:
+                cur.execute('''CREATE TABLE history
+                (unit text not null, time text not null, event int, keyId text, location text, address text, heading int, speed real, 
+                km real, run1 real,run2 real,run3 real,run4 real,runOdo real,temperature1 real,temperature2 real,
+                input1 int,input2 int,input3 int,input4 int,input5 int,input6 int,input7 int,input8 int,input9 int,input10 int,
+                output1 int,output2 int,output3 int,output4 int,output5 int,
+                analogInput1 real,analogInput2 real,analogInput4 real,
+                Input1ChangeCounter INT,Input2ChangeCounter INT,Input3ChangeCounter INT,Input4ChangeCounter INT,
+                batteryLevel real,externalPower real,
+                primary key (unit,time))''')
+                self._db.commit()
     def clean(self):
         """deletes all cached data"""
         self.cache.clean()
@@ -19,4 +40,7 @@ def get(self,url):
             return []
         if self.cache.dont_read_files and len(data) == 0:
             return []
+        # if self.use_sqlite and "api.trackunit.com/public/Report/UnitHistory" in url:
+        #     cur = self.db.cursor()
+        #     _data = list(map(lambda x: (x["time"])))
         return data.get('list')

tests/test_tucache.py

@@ -57,3 +57,6 @@ def test_Get_return_hashes():
 	data = cache.get(DUMMY_URL)
 	assert len(data) == 1
 	assert data[0] == "70551c7a431274e4617c94ad307346d2.json"
+def test_db():
+	cache = TuCache(_dir="pytest-web-cache",use_sqlite=True)
+

tests/test_webcache.py

@@ -5,8 +5,8 @@
 from os.path import isfile, join
 
 DUMMY_URL = 'https://pokeapi.co/api/v2/pokemon/ditto'
-DUMMY_RESPONSE_HASH = "3b28bef695a954335cadf285a6220b374bf5e947ef55102bea38823df53d2865"
-DATA_LEN = 21779
+DUMMY_RESPONSE_HASH = "493d9c6b65e19a64514d98a0515418f577c3a035300dc11423e2747591109764"
+DATA_LEN = 23291
 
 def get_hash(x):
 	return hashlib.sha256(x.encode('ascii')).hexdigest()