diff --git a/module/db/README.md b/module/db/README.md new file mode 100644 index 0000000..6419f4d --- /dev/null +++ b/module/db/README.md @@ -0,0 +1 @@ +# Agent Collaboration Test diff --git a/module/db/migrate.py b/module/db/migrate.py new file mode 100644 index 0000000..9459c78 --- /dev/null +++ b/module/db/migrate.py @@ -0,0 +1,21 @@ +import sqlite3 +import os + +DB_PATH = 'app.db' + +def migrate(): + conn = sqlite3.connect(DB_PATH) + cursor = conn.cursor() + + with open('schema.sql', 'r') as f: + cursor.executescript(f.read()) + + with open('seed.sql', 'r') as f: + cursor.executescript(f.read()) + + conn.commit() + conn.close() + print(f"Database migrated: {DB_PATH}") + +if __name__ == '__main__': + migrate() diff --git a/module/db/schema.sql b/module/db/schema.sql new file mode 100644 index 0000000..6b9aa9d --- /dev/null +++ b/module/db/schema.sql @@ -0,0 +1,13 @@ +CREATE TABLE IF NOT EXISTS users ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + name TEXT NOT NULL, + email TEXT UNIQUE, + created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP +); + +CREATE TABLE IF NOT EXISTS sessions ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + user_id INTEGER REFERENCES users(id), + token TEXT NOT NULL, + expires_at TIMESTAMP NOT NULL +); diff --git a/module/db/seed.sql b/module/db/seed.sql new file mode 100644 index 0000000..0e6696e --- /dev/null +++ b/module/db/seed.sql @@ -0,0 +1,3 @@ +INSERT INTO users (name, email) VALUES ('Alice', 'alice@example.com'); +INSERT INTO users (name, email) VALUES ('Bob', 'bob@example.com'); +INSERT INTO users (name, email) VALUES ('Charlie', 'charlie@example.com');