import sqlite3
def connect_db():
return sqlite3.connect('app.db')
def create_table():
with connect_db() as conn:
conn.execute('''CREATE TABLE IF NOT EXISTS records
(id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
age INTEGER NOT NULL);''')
conn.commit()
def insert_record(name, age):
with connect_db() as conn:
conn.execute('INSERT INTO records (name, age) VALUES (?, ?)', (name, age))
conn.commit()
def get_records():
with connect_db() as conn:
cursor = conn.execute('SELECT id, name, age FROM records')
return cursor.fetchall()
def update_record(record_id, name, age):
with connect_db() as conn:
conn.execute('UPDATE records SET name = ?, age = ? WHERE id = ?', (name,
age, record_id))
conn.commit()
def delete_record(record_id):
with connect_db() as conn:
conn.execute('DELETE FROM records WHERE id = ?', (record_id,))
conn.commit()
# Create table if not exists
create_table()