import sqlite3
from collections.abc import Iterator

DB_PATH = "tugas.db"

SKEMA = """
CREATE TABLE IF NOT EXISTS tugas (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    judul TEXT NOT NULL,
    selesai INTEGER NOT NULL DEFAULT 0
)
"""


def dapatkan_koneksi() -> Iterator[sqlite3.Connection]:
    koneksi = sqlite3.connect(DB_PATH)
    koneksi.row_factory = sqlite3.Row
    koneksi.execute(SKEMA)
    try:
        yield koneksi
    finally:
        koneksi.close()