database.py

import sqlite3


database_name = "task.db"

conn = sqlite3.connect("task.db")
cursor = conn.cursor()
cursor.execute("""
CREATE TABLE IF NOT EXISTS task(
    id INTEGER PRIMARY KEY AUTOINCREMENT, 
    description TEXT NOT NULL, status TEXT DEFAULT 'Pending')""")
conn.commit()
conn.close

def add(description):
    try:
        conn = sqlite3.connect("task.db")
        cursor = conn.cursor()
        cursor.execute("""INSERT INTO task(description) 
                            VALUES(?) """, (description,))
        conn.commit()
        print("task added sucessfully")
    except sqlite3.Error as e:
            print(f"Database error: {e}")
    finally:
            conn.close()

def list():
    try:
        conn = sqlite3.connect("task.db")
        cursor = conn.cursor()
        cursor.execute("""SELECT * FROM task""")
        rows= cursor.fetchall()
        return rows
    except sqlite3.Error as e:
            print(f"Database error: {e}")
    finally:
            conn.close()

def done(id):
    try:
        conn = sqlite3.connect("task.db")
        cursor = conn.cursor()
        cursor.execute("""UPDATE task SET status='Done' 
                    WHERE id=? """,(id,))
        conn.commit()
        
        
        print("task completed")
    except sqlite3.Error as e:
            print(f"Database error: {e}")
    finally:
            conn.close()

def delete(id):
    try:
        conn = sqlite3.connect("task.db")
        cursor = conn.cursor()
        cursor.execute("""DELETE FROM task WHERE id=?""", (id,))
        conn.commit()
        
        
        print("task deleted")
    except sqlite3.Error as e:
            print(f"Database error: {e}")
    finally:
            conn.close()

def edit(id, edited_description):
    try:
        conn = sqlite3.connect("task.db")
        cursor = conn.cursor()
        cursor.execute("""UPDATE task SET description=? WHERE id=?""", 
                    (edited_description, id))
        conn.commit()
        
        
        print("task updated")
    except sqlite3.Error as e:
        print(f"Database error: {e}")
    finally:
        conn.close()

def reset_table():
      try:
        conn = sqlite3.connect("task.db")
        cursor = conn.cursor()
        cursor.execute("DELETE FROM task;")
        cursor.execute("""DELETE FROM sqlite_sequence 
                       WHERE name='task'""")
        conn.commit()
        
        
        print("database reset successfully")
      except sqlite3.Error as e:
        print(f"Database error: {e}")
      finally:
        conn.close()