From 708dc0e96d8554f660c7d984721d95378746f1a4 Mon Sep 17 00:00:00 2001
From: Felix Zett <fezett@gmail.com>
Date: Tue, 5 Aug 2025 22:23:00 +0200
Subject: [PATCH] Refactor database connection and add item management
 endpoints

---
 backend/database.py      | 16 +++++++----
 backend/main.py          | 58 +++++++++++++++++++++++++++++++++++++---
 backend/requirements.txt |  1 +
 3 files changed, 67 insertions(+), 8 deletions(-)

diff --git a/backend/database.py b/backend/database.py
index efab532..ff610a6 100644
--- a/backend/database.py
+++ b/backend/database.py
@@ -1,8 +1,14 @@
 from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
-from sqlalchemy.orm import sessionmaker
-import os
+from sqlalchemy.orm import sessionmaker, declarative_base
 
-DATABASE_URL = os.environ.get("DATABASE_URL")
+DATABASE_URL = "postgresql+asyncpg://postgres:postgres@db:5432/packlist"
 
-engine = create_async_engine(DATABASE_URL, echo=True)
-SessionLocal = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
+engine = create_async_engine(DATABASE_URL)
+SessionLocal = sessionmaker(
+    bind=engine, class_=AsyncSession, expire_on_commit=False
+)
+Base = declarative_base()
+
+async def get_db():
+    async with SessionLocal() as db:
+        yield db
diff --git a/backend/main.py b/backend/main.py
index db51244..d7112d6 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -1,9 +1,18 @@
-from fastapi import FastAPI
-from models import Base
-from database import engine
+from datetime import date
+from typing import List
+from uuid import UUID, uuid4
+from fastapi import Depends, FastAPI
+from pydantic import BaseModel
+from sqlalchemy import select
+from models import Base, Item, Tag
+from database import engine, get_db
+from sqlalchemy.ext.asyncio import AsyncSession
+
 
 app = FastAPI()
 
+FIXED_USER_ID = UUID("00000000-0000-0000-0000-000000000001")
+
 
 @app.on_event("startup")
 async def startup():
@@ -14,3 +23,46 @@ async def startup():
 @app.get("/")
 def read_root():
     return {"status": "running"}
+
+class TagCreate(BaseModel):
+    name: str
+
+
+class ItemCreate(BaseModel):
+    name: str
+    tag_names: List[str]
+
+
+class TripCreate(BaseModel):
+    name: str
+    start_date: date
+    end_date: date
+    selected_tags: List[str]
+    marked_tags: List[str]
+
+@app.post("/items/")
+async def create_item(item: ItemCreate, db: AsyncSession = Depends(get_db)):
+    user_id = FIXED_USER_ID
+
+    db_item = Item(id=uuid4(), name=item.name, user_id=user_id)
+
+    tags = []
+    for tag_name in item.tag_names:
+        result = await db.execute(select(Tag).where(Tag.name == tag_name, Tag.user_id == user_id))
+        tag = result.scalar_one_or_none()
+        if not tag:
+            tag = Tag(id=uuid4(), name=tag_name, user_id=user_id)
+            db.add(tag)
+        tags.append(tag)
+
+    db_item.tags = tags
+    db.add(db_item)
+    await db.commit()
+    return {"status": "item created", "item_id": str(db_item.id)}
+
+@app.get("/items/")
+async def read_items(db: AsyncSession = Depends(get_db)):
+    user_id = FIXED_USER_ID
+    result = await db.execute(select(Item).where(Item.user_id == user_id))
+    items = result.scalars().all()
+    return {"items" : [item.name for item in items]}
\ No newline at end of file
diff --git a/backend/requirements.txt b/backend/requirements.txt
index 4aceb6c..c30fe25 100644
--- a/backend/requirements.txt
+++ b/backend/requirements.txt
@@ -2,6 +2,7 @@ fastapi
 uvicorn[standard]
 sqlalchemy
 asyncpg
+psycopg2-binary
 pydantic
 python-dotenv