Using Tiny DB

access_controller.py

@@ -5,11 +5,12 @@
 import firebase_admin
 from firebase_admin import credentials
 from firebase_admin import db
-from pymongo import MongoClient
 
-from database_access import FirebaseAccess, MongoAccess
+from database_access import FirebaseAccess, TinyDbAccess
 from fingerprint_reader import FingerprintReader
 
+from tinydb import TinyDB
+
 GPIO.setmode(GPIO.BCM) # GPIO Numbers instead of board numbers
 
 
@@ -142,19 +143,15 @@ def execute():
 
 
 def get_database_access():
-    if validate_connection():
-        return firebase_access
-    else:
-        return mongo_access
+    return tiny_db_access
 
 
-client = MongoClient('localhost', 27020)
-local_db = client['cda']
-
 authenticate()
 reader = FingerprintReader()
 
 firebase_access = FirebaseAccess(db)
-mongo_access = MongoAccess(local_db)
+
+tiny_db = TinyDB('./cda.json')
+tiny_db_access = TinyDbAccess(tiny_db)
 
 execute()

database_access.py

@@ -2,6 +2,7 @@
 from datetime import datetime
 
 import pytz
+from tinydb import Query
 
 
 class DatabaseAccess:
@@ -52,7 +53,7 @@ def create_log_data(hash_data, user, log_id):
         "datetime": str(time),
         "hash": hash_data,
         "user": user,
-        "_id": log_id
+        "_id": str(log_id)
     }
     return log_data
 
@@ -70,3 +71,25 @@ def add_log(self, hash, user_id):
 
         logs_db = self.db.logs
         logs_db.insert_one(log_data)
+
+
+class TinyDbAccess(DatabaseAccess):
+
+    def get_fingerprint(self, fingerprint):
+        print('Searching fingerprint in local db')
+        finger_db = self.db.table('fingerprints')
+        fingerprint_query = Query()
+
+        user = finger_db.get(fingerprint_query.fingerprint == fingerprint)
+
+        if user:
+            return user['user']
+        else:
+            return None
+
+    def add_log(self, hash, user_id):
+        print('Adding local log')
+        log_data = create_log_data(hash, user_id, uuid.uuid1())
+
+        logs_db = self.db.table('logs')
+        logs_db.insert(log_data)

fingerprint_sync.py

@@ -1,9 +1,9 @@
 import firebase_admin
 from firebase_admin import credentials
 from firebase_admin import db
-from pymongo import MongoClient
 import datetime
 from os import system
+from tinydb import TinyDB
 
 
 def authenticate():
@@ -15,20 +15,20 @@ def authenticate():
 
 def sync():
 
-    local_db.fingerprints.drop()
+    local_db.table('fingerprints').truncate()
     fingerprints = db.reference('fingerprints/').get()
     for i in fingerprints:
         add_fingerprint(i, fingerprints[i]['user'])
     print('Finish fingerprint sync')
 
-    local_db.users.drop()
+    local_db.table('users').truncate()
     users = db.reference('users/').get()
     for i in users:
         add_user(i, users[i]['name'], users[i]['lastname'], users[i]['company'], users[i]['status'])
     print('Finish user sync')
 
-    data = {'date': datetime.datetime.now()}
-    local_db.syncs.insert_one(data)
+    data = {'date': datetime.datetime.now().strftime('%Y-%m-%dT%H:%M:%S')}
+    local_db.table('syncs').insert(data)
     print('Change sync date')
 
 
@@ -42,19 +42,18 @@ def add_user(user_id, name, lastname, company, status):
         'status': status,
         'company': company
     }
-    local_db.users.insert_one(data)
+    local_db.table('users').insert(data)
 
 
 def add_fingerprint(fingerprint_id, user_id):
     data = {
         'fingerprint': fingerprint_id,
         'user': user_id
     }
-    local_db.fingerprints.insert_one(data)
+    local_db.table('fingerprints').insert(data)
 
 
-client = MongoClient('localhost', 27017)
-local_db = client['cda']
+local_db = TinyDB('./cda.json')
 
 
 system('clear')

requirements.txt

@@ -3,4 +3,5 @@ pymongo==3.4.0
 pytz==2020.1
 urllib3==1.24.3
 
-pyfingerprint~=1.5
+pyfingerprint~=1.5
+tinydb==4.4.0