Team Ai
Apppublic

Dub973/nlp2sql-space

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes
db_builder.py14 linesDownload Raw Back to root
1import sqlite3, pathlib
2
3DB_PATH = pathlib.Path("company.db")
4SCHEMA_FILE = pathlib.Path("schema.sql")
5
6def build_db():
7    DB_PATH.unlink(missing_ok=True)
8    with sqlite3.connect(DB_PATH) as conn, open(SCHEMA_FILE) as f:
9        conn.executescript(f.read())
10    print("Database created")
11
12if __name__ == "__main__":
13    build_db()
14