FIX: improved the compareFingerprintDir, split find fingerprint functions, bug fixes

dbfp.py

@@ -63,13 +63,24 @@ def indexFingerprints(fp_dir):
 
 #
 def compareFingerprintDir(file_in, fp_dir):
+  try:
     db = FingerprintDB()
     db.scanDBFile(file_in)
-  md5_db = db.getMD5DB()
-  md5_tables = db.getMD5Tables()
     fp = FingerprintIndex()
     fp.openIndex(fp_dir)
-  fp_list = fp.findFP(md5_db, md5_tables)
+
+    # search for fingerprints with exact database match
+    fp_ret = fp.findFP(db.getMD5DB())
+    if (fp_ret):
+      print "Database matche(s) found"
+      print "RESULTS:"
+      for fp_list in fp_ret:
+        for fp in fp_list[1].split(','):
+          print fp
+      print
+    # search for fingerprints with similar tables
+    else:
+      fp_list = fp.findFPTables(db.getMD5Tables())
       results = []
       for fp in fp_list:
         fq_fp = fp_dir + os.path.sep + fp
@@ -77,9 +88,13 @@ def compareFingerprintDir(file_in, fp_dir):
         percent = db.compareDB(fq_fp)
         results.append(percent)
 
+      print "Table matche(s) found"
       print "RESULTS: {}".format(results)
       results.sort()
       print "RESULTS: {}".format(results)
+  except Exception as ex:
+    print "Error comparing fingerprint"
+    print ex
 
 #
 def androidPull():

libs/fingerprint.py

@@ -148,6 +148,9 @@ class FingerprintDB:
   def getMD5DB(self):
     return self.db_hash
 
+  def getMD5Tables(self):
+    return self.table_hashes
+
   #
   # def getMD5Tables(self):
   #   if (self.table_hashes):

libs/fingerprint_index.py

@@ -72,12 +72,16 @@ class FingerprintIndex:
       raise FingerprintIndexWrite("Error creating an index file\n")
     return retVal
 
-  #
+  # looking for databases with exact schema match
-  def findFP(self, md5_db, md5_tables):
+  # RETURN: (fingerprint list, json_filename list, fingerprint count)
+  def findFP(self, md5_db):
     rows = self.__qDatabaseMD5(md5_db)    
     if len(rows) > 0:
       return rows
+    return None
 
+  # return an a list of json fingerprint files to open
+  def findFPTables(self, md5_tables):
     for md5_table in md5_tables:
       retval = {}
       rows = self.__qTableMD5(md5_table)
@@ -86,7 +90,6 @@ class FingerprintIndex:
         fps = fp_list.split(',')
         for fp in fps:
           retval[fp] = 1
-    #print "***** __qTableMD5 *****\n{}\n".format(retval.keys())
     return retval.keys()
 
   #