#!/usr/bin/env python3
import argparse
from db import connect, init_db

def main():
    ap = argparse.ArgumentParser(description="DB CLI")
    ap.add_argument("cmd", choices=["init"])
    ap.add_argument("--db", default="data/newsletter.db")
    ap.add_argument("--schema", default="scripts/schema.sql")
    args = ap.parse_args()

    con = connect(args.db)
    if args.cmd == "init":
        init_db(con, args.schema)
        print(f"Initialized schema in {args.db}")

if __name__ == "__main__":
    main()