formatted and added 4 workers

App/Analytics/AnalyticsRoutes.py

@@ -5,6 +5,3 @@ from App.Users.Model import User
 from App.Post.Model import Post
 
 analytics_router = APIRouter(tags=["Analytics"])
-
-
-

App/Analytics/Model.py

@@ -12,10 +12,12 @@ class Analytics(orm.Model):
     registry = models
     fields = {
         "id": orm.Integer(primary_key=True),
-        "user": orm.ForeignKey(User, on_delete=orm.CASCADE,allow_null=True),  # Optional for unknown users.
+        "user": orm.ForeignKey(
+            User, on_delete=orm.CASCADE, allow_null=True
+        ),  # Optional for unknown users.
         "post": orm.ForeignKey(User, on_delete=orm.CASCADE),
         "ip": orm.String(max_length=100),
         "device": orm.String(max_length=100),
-        "country": orm.String(max_length=100),        
+        "country": orm.String(max_length=100),
         "createdAt": orm.DateTime(index=True, default=datetime.datetime.now),
     }

App/Analytics/Schemas.py

@@ -11,5 +11,3 @@ class BaseRequest(BaseModel):
 
 class editRequest(BaseRequest):
     updatedAt: datetime = datetime.now()
-
-    

App/Authentication.py

@@ -1,7 +1,2 @@
 from fastapi import APIRouter, status
-from fastapi.security import OAuth2PasswordBearer,OAuth2PasswordRequestForm 
-
-
-
-
-
+from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm

App/CommentLikes/CommentLikesRoutes.py

@@ -6,6 +6,7 @@ from App.Post.Model import Post
 
 commentLikes_router = APIRouter(tags=["CommentLikes"])
 
-@commentLikes_router.get('/likes/comments/add')
+
+@commentLikes_router.get("/likes/comments/add")
 async def add_commentLike():
-    pass
+    pass

App/CommentLikes/Model.py

@@ -7,6 +7,7 @@ from App.modelInit import database, models
 from App.Users.Model import User
 from App.Post.Model import Post
 
+
 class CommentLike(orm.Model):
     tablename = "commentlikes"
     registry = models

App/CommentLikes/Schemas.py

@@ -3,13 +3,12 @@ from pydantic import EmailStr, BaseModel
 from datetime import date, datetime, time, timedelta
 
 
-class addLike (BaseModel):
-    postId:int
-    userId:int
+class addLike(BaseModel):
+    postId: int
+    userId: int
 
-class deleteLike (BaseModel):
-    id:int
-    postId:int
-    userId:int
 
-    
+class deleteLike(BaseModel):
+    id: int
+    postId: int
+    userId: int

App/Comments/CommentRoutes.py

@@ -1,13 +1,12 @@
 from fastapi import APIRouter, status
-from .Schemas import createComment,editComment,deleteComment,allComments
+from .Schemas import createComment, editComment, deleteComment, allComments
 from .Model import Comment
 from App.Users.Model import User
 from App.Post.Model import Post
 from App.utils import get_user_and_post
 import asyncio
-postLike_router = APIRouter(tags=["PostLikes"])
-
 
+postLike_router = APIRouter(tags=["PostLikes"])
 
 
 comment_router = APIRouter(tags=["Comments"])
@@ -15,8 +14,8 @@ comment_router = APIRouter(tags=["Comments"])
 
 @comment_router.post("/comment/create")
 async def create_comment(comment: createComment):
-    user,_post = await get_user_and_post(comment)
-    data = await Comment.objects.create(user=user, content=comment.content,post=_post)
+    user, _post = await get_user_and_post(comment)
+    data = await Comment.objects.create(user=user, content=comment.content, post=_post)
     return {"code": 200, "message": "success", "payload": data.__dict__}
 
 
@@ -38,7 +37,6 @@ async def edit_comment(comment: editComment):
 
 @comment_router.post("/comment/all")
 async def all_comments(comment: allComments):
-
     user = await User.objects.filter(id=comment.userId).first()
     if not user:
         return {"code": 400, "message": "User does not exist", "payload": None}

App/Comments/Schemas.py

@@ -6,19 +6,21 @@ from datetime import date, datetime, time, timedelta
 class createComment(BaseModel):
     userId: int
     content: str
-    postId:int
+    postId: int
 
 
 class editComment(BaseModel):
-    id:int
+    id: int
     userId: int
     content: str
-    postId:int
+    postId: int
+
 
 class deleteComment(BaseModel):
-    id:int
+    id: int
     userId: int
-    postId:int
+    postId: int
+
 
 class allComments(BaseModel):
     userId: int

App/Post/Model.py

@@ -12,7 +12,7 @@ class Post(orm.Model):
     registry = models
     fields = {
         "id": orm.Integer(primary_key=True),
-        "pageId": orm.Integer(index=True,default=0),
+        "pageId": orm.Integer(index=True, default=0),
         "content": orm.JSON(),
         "recommendations": orm.JSON(allow_null=True),
         "createdAt": orm.DateTime(index=True, default=datetime.datetime.now),

App/Post/PostRoutes.py

@@ -1,5 +1,5 @@
 from fastapi import APIRouter, status
-from .Schemas import createPost,editPost,getPost
+from .Schemas import createPost, editPost, getPost
 from .Model import Post
 
 post_router = APIRouter(tags=["Posts"])
@@ -7,18 +7,19 @@ post_router = APIRouter(tags=["Posts"])
 
 @post_router.post("/post/create")
 async def create_post(post: createPost):
-    data=await Post.objects.create(**post.dict())
+    data = await Post.objects.create(**post.dict())
     return {"code": 200, "message": "success", "payload": data.__dict__}
 
+
 @post_router.post("/post/update")
 async def create_post(post: editPost):
-    temp=await Post.objects.get(id= post.id)
-    data=await temp.update(recommendations=post.recommendations,content=post.content)
+    temp = await Post.objects.get(id=post.id)
+    data = await temp.update(recommendations=post.recommendations, content=post.content)
     # data=await Post.objects.update(**post.dict())
     return {"code": 200, "message": "success", "payload": temp.__dict__}
 
 
 @post_router.post("/post/get")
 async def create_post(post: getPost):
-    data=await Post.objects.get(id=post.id)
-    return {"code": 200, "message": "success", "payload": data.__dict__}
+    data = await Post.objects.get(id=post.id)
+    return {"code": 200, "message": "success", "payload": data.__dict__}

App/Post/Schemas.py

@@ -5,15 +5,14 @@ from datetime import date, datetime, time, timedelta
 
 class editPost(BaseModel):
     id: Optional[int]
-    content:Optional[dict]
-    recommendations:Optional[dict]
+    content: Optional[dict]
+    recommendations: Optional[dict]
 
 
 class createPost(BaseModel):
-    content:Optional[dict]
-    recommendations:Optional[dict]
+    content: Optional[dict]
+    recommendations: Optional[dict]
+
 
 class getPost(BaseModel):
     id: Optional[int]
-
-    

App/PostLikes/Model.py

@@ -15,6 +15,5 @@ class PostLike(orm.Model):
         "id": orm.Integer(primary_key=True),
         "user": orm.ForeignKey(User, on_delete=orm.CASCADE),
         "post": orm.ForeignKey(Post, on_delete=orm.CASCADE),
-        "createdAt": orm.DateTime(index=True, default=datetime.datetime.now)
+        "createdAt": orm.DateTime(index=True, default=datetime.datetime.now),
     }
-

App/PostLikes/PostLikesRoutes.py

@@ -1,29 +1,31 @@
 from fastapi import APIRouter, Depends
-from .Schemas import addLike,deleteLike
+from .Schemas import addLike, deleteLike
 from .Model import PostLike
 from App.Users.Model import User
 from App.Post.Model import Post
 from App.utils import get_user_and_post
 import asyncio
-postLike_router = APIRouter(tags=["PostLikes"])
-
 
+postLike_router = APIRouter(tags=["PostLikes"])
 
 
 @postLike_router.post("/postLike/add")
 async def add_postLike(post: addLike):
-    user,_post=await get_user_and_post(post) #validate if both the post and user exist
-    previos =await PostLike.objects.filter(user=user).filter(post=_post).first()
+    user, _post = await get_user_and_post(
+        post
+    )  # validate if both the post and user exist
+    previos = await PostLike.objects.filter(user=user).filter(post=_post).first()
     if previos:
         return {"code": 400, "message": "you already liked it", "payload": None}
-    
-    data=await PostLike.objects.create(post=_post,user=user)
+
+    data = await PostLike.objects.create(post=_post, user=user)
     return {"code": 200, "message": "success", "payload": data.__dict__}
 
+
 @postLike_router.post("/postLike/delete")
 async def create_post(post: deleteLike):
-    user,_post=await get_user_and_post(post)
-    data=await PostLike.objects.filter(id=post.id).first()
+    user, _post = await get_user_and_post(post)
+    data = await PostLike.objects.filter(id=post.id).first()
     if not data:
         return {"code": 400, "message": "Does not exist", "payload": None}
     if user.id == data.user.id and _post.id == data.post.id:
@@ -33,8 +35,8 @@ async def create_post(post: deleteLike):
 
 @postLike_router.post("/postLike/get")
 async def create_post(post: deleteLike):
-    user,_post=await get_user_and_post(post)
-    data=await PostLike.objects.filter(id=post.id).first()
+    user, _post = await get_user_and_post(post)
+    data = await PostLike.objects.filter(id=post.id).first()
     if not data:
         return {"code": 400, "message": "Does not exist", "payload": None}
     if user.id == data.user.id and _post.id == data.post.user.id:

App/PostLikes/Schemas.py

@@ -3,14 +3,12 @@ from pydantic import EmailStr, BaseModel
 from datetime import date, datetime, time, timedelta
 
 
-class addLike (BaseModel):
-    postId:int
-    userId:int
+class addLike(BaseModel):
+    postId: int
+    userId: int
 
-class deleteLike (BaseModel):
-    id:int
-    postId:int
-    userId:int
 
-
-    
+class deleteLike(BaseModel):
+    id: int
+    postId: int
+    userId: int

App/Settings.py

@@ -1,3 +1,3 @@
 class Settings:
     ALGORITHM = "HS256"
-    HASH="86c5ceb27e1bf441130299c0209e5f35b88089f62c06b2b09d65772274f12057"
+    HASH = "86c5ceb27e1bf441130299c0209e5f35b88089f62c06b2b09d65772274f12057"

App/Users/UserRoutes.py

@@ -2,9 +2,9 @@ from fastapi import APIRouter, status
 from .Schemas import BaseRequest
 from .Model import User
 from sqlalchemy import and_
-from fastapi.security import OAuth2PasswordBearer,OAuth2PasswordRequestForm 
+from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
 
-oauth_scheme=OAuth2PasswordBearer(tokenUrl='/user/login')
+oauth_scheme = OAuth2PasswordBearer(tokenUrl="/user/login")
 user_router = APIRouter(tags=["User"])
 
 

App/utils.py

@@ -3,12 +3,15 @@ from App.Post.Model import Post
 import asyncio
 from fastapi import HTTPException
 
+
 async def get_user_and_post(content):
     try:
-    #    user = None
-    #    post = await Post.objects.get(id=content.postId)
-    #    print(post.id)
-        user,post = await asyncio.gather(*[User.objects.get(id=content.userId), Post.objects.get(id=content.postId)])
+        #    user = None
+        #    post = await Post.objects.get(id=content.postId)
+        #    print(post.id)
+        user, post = await asyncio.gather(
+            *[User.objects.get(id=content.userId), Post.objects.get(id=content.postId)]
+        )
     except:
         raise HTTPException(status_code=400, detail="Invalid data")
-    return user,post
+    return user, post

Dockerfile

@@ -3,5 +3,5 @@ WORKDIR /usr/src/nginx
 COPY . .
 
 RUN python3 -m pip install -r requirements.txt 
-CMD  uvicorn App.app:app --host 0.0.0.0
+CMD  gunicorn main:app --workers 4   --host 0.0.0.0
 EXPOSE 8000

requirements.txt

@@ -14,4 +14,5 @@ Werkzeug==2.2.2
 passlib                # for password hashing
 pydantic[email]
 uvicorn==0.21.1
+gunicorn
 python-multipart

stress_test.py

@@ -0,0 +1,23 @@
+import asyncio
+import aiohttp
+
+
+async def send_post_request(session,number=0):
+    async with session.post(
+        url="https://fastapi-mbonea-mjema.cloud.okteto.net/comment/create",
+        json={"userId": 1, "content": f"comment {number}", "postId": 1},
+    ) as response:
+        response_text = await response.text()
+        print(f"Response from {number}: {response_text}")
+
+
+async def send_post_requests_async():
+    async with aiohttp.ClientSession() as session:
+        tasks = []
+        for number in range(1000):
+            tasks.append(asyncio.ensure_future(send_post_request(session,number)))
+        await asyncio.gather(*tasks)
+
+
+if __name__ == "__main__":
+    asyncio.run(send_post_requests_async())