Eliminare 'src/LFDB.py'
This commit is contained in:
parent
aeafcfe219
commit
ae47cc326f
55
src/LFDB.py
55
src/LFDB.py
|
|
@ -1,55 +0,0 @@
|
|||
import os
|
||||
import sqlite3
|
||||
from sqlite3 import Error
|
||||
from werkzeug.datastructures import FileStorage
|
||||
|
||||
|
||||
class LFDB:
|
||||
|
||||
def __init__(self, db_path):
|
||||
# self.lf = LFUtilities.load(settings.DATASET_BEBLID)
|
||||
# self.ids = np.loadtxt(settings.DATASET_IDS, dtype=str).tolist()
|
||||
# self.bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
|
||||
self.conn = sqlite3.connect(db_path, check_same_thread=False)
|
||||
|
||||
def close(self):
|
||||
if self.conn:
|
||||
self.conn.close()
|
||||
|
||||
def put(self, docId, features):
|
||||
try:
|
||||
self.conn.text_factory = str
|
||||
#print("[INFO] : Successful connection!")
|
||||
cur = self.conn.cursor()
|
||||
insert_file = '''INSERT INTO lf(docId, features) VALUES(?, ?)'''
|
||||
cur = self.conn.cursor()
|
||||
cur.execute(insert_file, (docId, features,))
|
||||
#print("[INFO] : The blob for ", docId, " is in the database.")
|
||||
except Error as e:
|
||||
print(e)
|
||||
|
||||
def commit(self):
|
||||
try:
|
||||
if self.conn:
|
||||
self.conn.commit()
|
||||
print("committing...")
|
||||
except Error as e:
|
||||
print(e)
|
||||
|
||||
def get(self, docId):
|
||||
try:
|
||||
self.conn.text_factory = str
|
||||
cur = self.conn.cursor()
|
||||
# print("[INFO] : Connected to SQLite to read_blob_data")
|
||||
sql_fetch_blob_query = """SELECT * from lf where docId = ?"""
|
||||
cur.execute(sql_fetch_blob_query, (docId,))
|
||||
record = cur.fetchall()
|
||||
for row in record:
|
||||
converted_file_name = row[1]
|
||||
blob = row[2]
|
||||
# parse out the file name from converted_file_name
|
||||
cur.close()
|
||||
except sqlite3.Error as error:
|
||||
print("[INFO] : Failed to read blob data from sqlite table", error)
|
||||
return blob
|
||||
|
||||
Loading…
Reference in New Issue