Update generated backend for blog_app with entities: posts, comments, tags, user

app/api/core/dependencies/dependencies.py

@@ -0,0 +1,8 @@
+from sqlalchemy.orm import Session
+from app.api.db.database import SessionLocal
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()

app/api/core/middleware/activity_tracker.py

@@ -0,0 +1,12 @@
+import time
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.requests import Request
+from starlette.responses import Response
+from loguru import logger
+class ActivityTrackerMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request: Request, call_next):
+        start_time = time.time()
+        response = await call_next(request)
+        process_time = time.time() - start_time
+        logger.info(f'{request.method} {request.url} - Process Time: {process_time:.6f} seconds')
+        return response

app/api/db/database.py

@@ -0,0 +1,7 @@
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+SQLALCHEMY_DATABASE_URL = 'sqlite:///./blog_app.db'
+engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={'check_same_thread': False})
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+Base = declarative_base()

app/api/v1/models/comments.py

@@ -0,0 +1,7 @@
+from sqlalchemy import Column, ForeignKey, Integer, String, Text
+from app.api.db.database import Base
+
+class Comments(Base):
+    __tablename__ = 'comments'
+
+    id = Column(Integer, primary_key=True, index=True)

app/api/v1/models/posts.py

@@ -0,0 +1,7 @@
+from sqlalchemy import Column, ForeignKey, Integer, String, Text
+from app.api.db.database import Base
+
+class Posts(Base):
+    __tablename__ = 'posts'
+
+    id = Column(Integer, primary_key=True, index=True)

app/api/v1/models/tags.py

@@ -0,0 +1,7 @@
+from sqlalchemy import Column, ForeignKey, Integer, String, Text
+from app.api.db.database import Base
+
+class Tags(Base):
+    __tablename__ = 'tags'
+
+    id = Column(Integer, primary_key=True, index=True)

app/api/v1/models/user.py

@@ -0,0 +1,7 @@
+from sqlalchemy import Column, ForeignKey, Integer, String, Text
+from app.api.db.database import Base
+
+class User(Base):
+    __tablename__ = 'user'
+
+    id = Column(Integer, primary_key=True, index=True)

app/api/v1/routes/__init__.py

@@ -0,0 +1,2 @@
+from fastapi import APIRouter
+router = APIRouter()

app/api/v1/routes/comments.py

@@ -0,0 +1,9 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
+from app.api.core.dependencies import get_db
+
+router = APIRouter()
+
+@router.get('/comments/')
+def read_comments(db: Session = Depends(get_db)):
+    return {'message': 'Read comments'}

app/api/v1/routes/posts.py

@@ -0,0 +1,9 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
+from app.api.core.dependencies import get_db
+
+router = APIRouter()
+
+@router.get('/posts/')
+def read_posts(db: Session = Depends(get_db)):
+    return {'message': 'Read posts'}

app/api/v1/routes/tags.py

@@ -0,0 +1,9 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
+from app.api.core.dependencies import get_db
+
+router = APIRouter()
+
+@router.get('/tags/')
+def read_tags(db: Session = Depends(get_db)):
+    return {'message': 'Read tags'}

app/api/v1/routes/user.py

@@ -0,0 +1,9 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
+from app.api.core.dependencies import get_db
+
+router = APIRouter()
+
+@router.get('/user/')
+def read_user(db: Session = Depends(get_db)):
+    return {'message': 'Read user'}

app/api/v1/schemas/comments.py

@@ -0,0 +1,10 @@
+from pydantic import BaseModel
+
+class CommentsBase(BaseModel):
+    pass
+
+class Comments(BaseModel):
+    id: int
+
+    class Config:
+        orm_mode = True

app/api/v1/schemas/posts.py

@@ -0,0 +1,10 @@
+from pydantic import BaseModel
+
+class PostsBase(BaseModel):
+    pass
+
+class Posts(BaseModel):
+    id: int
+
+    class Config:
+        orm_mode = True

app/api/v1/schemas/tags.py

@@ -0,0 +1,10 @@
+from pydantic import BaseModel
+
+class TagsBase(BaseModel):
+    pass
+
+class Tags(BaseModel):
+    id: int
+
+    class Config:
+        orm_mode = True

app/api/v1/schemas/user.py

@@ -0,0 +1,10 @@
+from pydantic import BaseModel
+
+class UserBase(BaseModel):
+    pass
+
+class User(BaseModel):
+    id: int
+
+    class Config:
+        orm_mode = True

main.py

@@ -1,7 +1,10 @@
 from fastapi import FastAPI
-
+from app.api.db.database import engine, Base
-app = FastAPI(title="Generated Backend")
+from app.api.v1.routes import router
-
+from app.api.core.middleware.activity_tracker import ActivityTrackerMiddleware
-@app.get("/")
+app = FastAPI()
-def read_root():
+app.add_middleware(ActivityTrackerMiddleware)
-    return {"message": "Welcome to the generated backend"}
+app.include_router(router, prefix='/v1')
+@app.on_event('startup')
+def startup():
+    Base.metadata.create_all(bind=engine)

requirements.txt

@@ -1,4 +1 @@
-fastapi
+# No code generated
-uvicorn
-sqlalchemy
-pydantic