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

app/api/core/dependencies/dependencies.py

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

app/api/core/middleware/activity_tracker.py

@@ -1,12 +1,13 @@
-import time
+from time 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()
+        start_time = time()
-        response = await call_next(request)
+        response: Response = await call_next(request)
-        process_time = time.time() - start_time
+        process_time = time() - start_time
-        logger.info(f'{request.method} {request.url} - Process Time: {process_time:.6f} seconds')
+        logger.info(f"Processed {request.method} {request.url} in {process_time:.4f} seconds")
-        return response
+        return response

app/api/v1/models/comments.py

@@ -1,7 +1,16 @@
 from sqlalchemy import Column, ForeignKey, Integer, String, Text
+from sqlalchemy.orm import relationship
 from app.api.db.database import Base
 
 class Comments(Base):
     __tablename__ = 'comments'
 
     id = Column(Integer, primary_key=True, index=True)
+    post_id = Column(Integer, ForeignKey('posts.id'))
+    user_id = Column(Integer, ForeignKey('users.id'))
+    comment_text = Column(Text)
+    created_at = Column(Integer)
+    updated_at = Column(Integer)
+
+    post = relationship('Post', back_populates='comments')
+    user = relationship('User', back_populates='comments')

app/api/v1/models/posts.py

@@ -1,7 +1,13 @@
 from sqlalchemy import Column, ForeignKey, Integer, String, Text
+from sqlalchemy.orm import relationship
 from app.api.db.database import Base
 
 class Posts(Base):
     __tablename__ = 'posts'
 
     id = Column(Integer, primary_key=True, index=True)
+    title = Column(String)
+    content = Column(Text)
+    user_id = Column(Integer, ForeignKey('users.id'))
+
+    user = relationship('User', back_populates='posts')

app/api/v1/models/tags.py

@@ -1,7 +1,12 @@
 from sqlalchemy import Column, ForeignKey, Integer, String, Text
+from sqlalchemy.orm import relationship
 from app.api.db.database import Base
 
 class Tags(Base):
     __tablename__ = 'tags'
 
     id = Column(Integer, primary_key=True, index=True)
+    name = Column(String, nullable=False, unique=True)
+    description = Column(Text)
+
+    posts = relationship("Post", back_populates="tags", secondary="post_tags")

app/api/v1/models/user.py

@@ -1,7 +1,14 @@
 from sqlalchemy import Column, ForeignKey, Integer, String, Text
+from sqlalchemy.orm import relationship
 from app.api.db.database import Base
 
 class User(Base):
     __tablename__ = 'user'
 
     id = Column(Integer, primary_key=True, index=True)
+    email = Column(String, unique=True, index=True)
+    hashed_password = Column(String)
+    is_active = Column(Integer, default=1)
+    full_name = Column(String)
+
+    posts = relationship("Post", back_populates="author")

app/api/v1/routes/__init__.py

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

app/api/v1/schemas/comments.py

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

app/api/v1/schemas/posts.py

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

app/api/v1/schemas/tags.py

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

app/api/v1/schemas/user.py

@@ -1,10 +1,11 @@
 from pydantic import BaseModel
 
 class UserBase(BaseModel):
-    pass
+    email: str
 
-class User(BaseModel):
+class User(UserBase):
     id: int
+    is_active: bool
 
     class Config:
-        orm_mode = True
+        orm_mode = True

main.py

@@ -2,9 +2,11 @@ from fastapi import FastAPI
 from app.api.db.database import engine, Base
 from app.api.v1.routes import router
 from app.api.core.middleware.activity_tracker import ActivityTrackerMiddleware
+
 app = FastAPI()
 app.add_middleware(ActivityTrackerMiddleware)
-app.include_router(router, prefix='/v1')
+app.include_router(router, prefix="/v1")
-@app.on_event('startup')
+
-def startup():
+@app.on_event("startup")
-    Base.metadata.create_all(bind=engine)
+def startup_event():
+    Base.metadata.create_all(bind=engine)

requirements.txt

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