33 lines
1007 B
Python
33 lines
1007 B
Python
import sqlite3
|
|
|
|
class Database:
|
|
def __init__(self, database_file):
|
|
self.db = sqlite3.connect(database_file)
|
|
|
|
def build_database(self, sql_file_path):
|
|
sql_script_string = ''
|
|
with open(sql_file_path, 'r') as sql_file:
|
|
sql_script_string = sql_file.read()
|
|
|
|
cursor = self.db.cursor()
|
|
cursor.executescript(sql_script_string)
|
|
self.db.commit()
|
|
|
|
def select(self, index):
|
|
# Query the database for the specified index
|
|
cursor = self.db.cursor()
|
|
query = "SELECT name, address FROM people WHERE id = ?"
|
|
cursor.execute(query, (index,))
|
|
result = cursor.fetchone()
|
|
if result:
|
|
return result
|
|
else:
|
|
return None
|
|
|
|
def insert(self, name, address):
|
|
# Insert new entry into the database
|
|
cursor = self.db.cursor()
|
|
query = "INSERT INTO people (name, address) VALUES (?, ?)"
|
|
cursor.execute(query, (name, address))
|
|
self.db.commit()
|