def create_list(list_name):
connection = connect()
sql = f"INSERT INTO list (name) VALUES ('{list_name}');"
insert(connection, sql)
close_connection(connection)
def delete_list(list_name):
connection = connect()
sql = f"DELETE FROM list WHERE list.name = '{list_name}';"
delete(connection, sql)
close_connection(connection)
def create_contact(name, email, list_id):
connection = connect()
sql = f"INSERT INTO contact (name, email, list_id) \
VALUES ('{name}','{email}', {list_id});"
insert(connection, sql)
close_connection(connection)
def search_contacts(list_name):
connection = connect()
sql = f"SELECT contact.name, contact.email from contact, list \
where contact.list_id = list.id and list.name = '{list_name}';"
contacts = select(connection, sql)
close_connection(connection)
return contacts
def search_list():
connection = connect()
sql = "SELECT * FROM list;"
lists = select(connection, sql)
if not lists:
create_list("Padrão")
lists = [(1, "Padrão")]
close_connection(connection)
return lists
def initialize():
connection = connect()
sql_create_table_lists = """ CREATE TABLE IF NOT EXISTS list (
id integer PRIMARY KEY AUTOINCREMENT,
name text NOT NULL
); """
# Criando a tabela
sql_create_table_contact = """ CREATE TABLE IF NOT EXISTS contact (
id integer PRIMARY KEY AUTOINCREMENT,
name text NOT NULL,
email text NOT NULL,
list_id integer NOT NULL,
FOREIGN KEY(list_id) REFERENCES list(id)
ON DELETE CASCADE
); """
create(connection, sql_create_table_lists)
create(connection, sql_create_table_contact)
close_connection(connection)