diff --git a/libs/fingerprint_index.py b/libs/fingerprint_index.py index 5b0d03e..4d96b01 100644 --- a/libs/fingerprint_index.py +++ b/libs/fingerprint_index.py @@ -52,12 +52,14 @@ class FingerprintIndex: CREATE TABLE md5_all ( md5_db TEXT PRIMARY KEY, md5_list TEXT, - fp_list TEXT); + fp_list TEXT, + fp_count INTEGER); ''') self.db_conn.execute(''' CREATE TABLE md5_tables ( md5_table TEXT, - fp_list TEXT); + fp_list TEXT, + fp_count INTEGER); ''') logging.info("Successfully created index table") self.__populateIndex(fp_dir) @@ -102,8 +104,8 @@ class FingerprintIndex: # logging.info("INSERT INTO md5_index VALUES(?, ?, ?): {}; {}; {}".format(md5_all, str(md5_list), filename)) self.db_conn.execute( ''' - INSERT INTO md5_all VALUES(?, ?, ?) - ''', [md5_db, ','.join(md5_list), filename]) + INSERT INTO md5_all VALUES(?, ?, ?, ?) + ''', [md5_db, ','.join(md5_list), filename, 1]) except sql.IntegrityError: try: fp_list = self.__selectFileList(md5_db)