Add DuckDB Schema and handling

src/lib/database.py

@@ -0,0 +1,89 @@
+from contextlib import contextmanager
+import duckdb
+import bcrypt
+from rich import print
+
+
+from lib import settings
+from models.user import UserIn, UserOut
+
+
+@contextmanager
+def get_db():
+    try:
+        conn = duckdb.connect(database=settings.db_url)
+        yield conn
+    finally:
+        conn.close()
+
+
+def db_run(sql):
+    with get_db() as conn:
+        conn.execute(sql)
+
+
+def init_db():
+    if settings.db_url.exists():
+        # TODO: Development feature. Remove this in production
+        from os import unlink
+
+        unlink(settings.db_url)
+        # return
+    sql = ""
+    with open("lib/database.sql") as f:
+        sql = f.read()
+
+    db_run(sql)
+
+    # Initialize default user
+
+    hashed_password = bcrypt.hashpw("password".encode("utf-8"), bcrypt.gensalt())
+
+    sql = f"INSERT INTO users (username, password, is_admin) VALUES ('admin', '{hashed_password.decode('utf-8')}', true)"
+    db_run(sql)
+
+    sql = f"INSERT INTO users (username, password, is_admin) VALUES ('test', '{hashed_password.decode('utf-8')}', false)"
+    db_run(sql)
+
+
+def get_user_by_username(username) -> UserIn | None:
+    sql = f"SELECT * FROM users WHERE username = '{username}'"
+    with get_db() as conn:
+        result = conn.execute(sql).fetchone()
+        if result is not None:
+            return UserIn(
+                id=str(result[0]),
+                username=result[1],
+                password=result[2],
+                email=result[3],
+                avatar_url=result[4],
+                is_active=result[5],
+                is_admin=result[6],
+                last_login=result[7],
+                created_at=result[8],
+                updated_at=result[9],
+            )
+
+        return None
+
+
+def get_all_users() -> list[UserOut]:
+    sql = "SELECT * FROM users"
+    with get_db() as conn:
+        result = conn.execute(sql).fetchall()
+        users = []
+        for row in result:
+            user = UserOut(
+                id=str(row[0]),
+                username=row[1],
+                email=row[3],
+                avatar_url=row[4],
+                is_active=row[5],
+                is_admin=row[6],
+                last_login=row[7],
+                created_at=row[8],
+                updated_at=row[9],
+            )
+            users.append(user)
+
+        return users