File size: 961 Bytes
08b1fbc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | import sqlite3
import json
DB = "data/ekyc.db"
def init_db():
conn = sqlite3.connect(DB)
c = conn.cursor()
c.execute("""
CREATE TABLE IF NOT EXISTS users (
id TEXT PRIMARY KEY,
name TEXT,
father TEXT,
dob TEXT,
type TEXT,
embedding TEXT
)
""")
conn.commit()
conn.close()
def insert_record(data):
conn = sqlite3.connect(DB)
c = conn.cursor()
c.execute("""
INSERT INTO users VALUES (?, ?, ?, ?, ?, ?)
""", (
data["ID"],
data["Name"],
data["Father's Name"],
data["DOB"],
data["ID Type"],
json.dumps(data["Embedding"])
))
conn.commit()
conn.close()
def check_duplicate(user_id):
conn = sqlite3.connect(DB)
c = conn.cursor()
c.execute("SELECT * FROM users WHERE id=?", (user_id,))
exists = c.fetchone() is not None
conn.close()
return exists
|