add an endpoint that writes to db

Author: JimmyDore

Taskfile.yml

@@ -3,6 +3,8 @@ version: '3'
 vars:
   FRONTEND_DIR: frontend
   BACKEND_DIR: backend
+  PROD_HOST: vpsjim
+  PROD_DB_PATH: ~/pingmaster/data/sql_app.db
 
 tasks:
   install:
@@ -96,4 +98,22 @@ tasks:
     desc: Arrête les conteneurs Docker
     dir: '{{.BACKEND_DIR}}'
     cmds:
-      - docker-compose down
+      - docker-compose down
+
+  db:download:
+    desc: Download the production database
+    cmds:
+      - mkdir -p {{.BACKEND_DIR}}/data
+      - scp {{.PROD_HOST}}:{{.PROD_DB_PATH}} {{.BACKEND_DIR}}/data/prod_db.db
+      - echo "Production database downloaded to {{.BACKEND_DIR}}/data/prod_db.db"
+
+  db:view:
+    desc: Open the downloaded production database in VS Code
+    cmds:
+      - code {{.BACKEND_DIR}}/data/prod_db.db
+
+  db:manage:
+    desc: Download and view the production database
+    cmds:
+      - task: db:download
+      - task: db:view

backend/app/api/endpoints/messages.py

@@ -0,0 +1,28 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+from app.db.session import get_db
+from app.db.models import Message
+from pydantic import BaseModel
+
+router = APIRouter()
+
+class MessageCreate(BaseModel):
+    content: str
+
+class MessageResponse(BaseModel):
+    id: int
+    content: str
+
+    class Config:
+        from_attributes = True
+
+@router.post("/messages/", response_model=MessageResponse, status_code=201)
+def create_message(
+    message: MessageCreate,
+    db: Session = Depends(get_db)
+):
+    db_message = Message(content=message.content)
+    db.add(db_message)
+    db.commit()
+    db.refresh(db_message)
+    return db_message

backend/app/db/models.py

@@ -0,0 +1,10 @@
+from sqlalchemy import Column, Integer, String, DateTime
+from sqlalchemy.sql import func
+from .session import Base
+
+class Message(Base):
+    __tablename__ = "messages"
+
+    id = Column(Integer, primary_key=True, index=True)
+    content = Column(String, nullable=False)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())

backend/app/db/session.py

@@ -0,0 +1,18 @@
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker, declarative_base
+
+SQLITE_URL = "sqlite:///./sql_app.db"
+
+engine = create_engine(SQLITE_URL)
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+Base = declarative_base()
+
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+def init_db():
+    Base.metadata.create_all(bind=engine)

backend/app/main.py

@@ -1,11 +1,16 @@
 from fastapi import FastAPI
-from app.api.endpoints import hello
+from app.api.endpoints import hello, messages
+from app.db.session import init_db
 
 app = FastAPI(
     title="MonitoringDashboard API",
     description="Backend API for Monitoring Dashboard",
     version="0.1.0"
 )
 
-# Include routers
-app.include_router(hello.router, prefix="/api")
+@app.on_event("startup")
+def startup_event():
+    init_db()
+
+app.include_router(hello.router, prefix="/api")
+app.include_router(messages.router, prefix="/api")

backend/app/tests/api/test_messages.py

@@ -0,0 +1,11 @@
+from fastapi.testclient import TestClient
+
+def test_create_message(client: TestClient):
+    response = client.post(
+        "/api/messages/",
+        json={"content": "Test message"}
+    )
+    assert response.status_code == 201
+    data = response.json()
+    assert "id" in data
+    assert data["content"] == "Test message" 

backend/app/tests/conftest.py

@@ -1,7 +1,29 @@
 import pytest
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
 from fastapi.testclient import TestClient
+from app.db.session import Base, get_db
 from app.main import app
 
+SQLITE_TEST_URL = "sqlite:///./test.db"
+
+@pytest.fixture
+def test_db():
+    engine = create_engine(SQLITE_TEST_URL, connect_args={"check_same_thread": False})
+    Base.metadata.create_all(bind=engine)
+    TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+    
+    def override_get_db():
+        try:
+            db = TestingSessionLocal()
+            yield db
+        finally:
+            db.close()
+            
+    app.dependency_overrides[get_db] = override_get_db
+    yield TestingSessionLocal
+    Base.metadata.drop_all(bind=engine)
+
 @pytest.fixture
-def client():
+def client(test_db):
     return TestClient(app)

backend/pytest.ini

@@ -0,0 +1,3 @@
+[pytest]
+markers =
+    asyncio: mark a test as an asyncio test.

backend/requirements.txt

@@ -1,7 +1,24 @@
-fastapi>=0.104.0
-uvicorn>=0.24.0
-pytest>=7.4.3
-httpx>=0.25.1
-sqlalchemy>=2.0.23
-aiosqlite>=0.19.0
-python-dotenv>=1.0.0
+aiosqlite==0.20.0
+annotated-types==0.7.0
+anyio==4.6.2.post1
+certifi==2024.8.30
+click==8.1.7
+fastapi==0.115.6
+greenlet==3.1.1
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.0
+idna==3.10
+iniconfig==2.0.0
+packaging==24.2
+pluggy==1.5.0
+pydantic==2.10.3
+pydantic_core==2.27.1
+pytest==8.3.4
+pytest-asyncio==0.24.0
+python-dotenv==1.0.1
+sniffio==1.3.1
+SQLAlchemy==2.0.36
+starlette==0.41.3
+typing_extensions==4.12.2
+uvicorn==0.32.1