From a1585bd5a0bd769d37705a400183a9c2c6a2cbb1 Mon Sep 17 00:00:00 2001
From: IgorVolochay <spore.editor@gmail.com>
Date: Mon, 17 Feb 2025 14:54:10 +0300
Subject: [PATCH] Rework add_user rout

---
 app/main.py                | 12 ++++++------
 app/schemas/api_schemas.py |  8 ++++++++
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/app/main.py b/app/main.py
index 4687ec5..49d45fd 100644
--- a/app/main.py
+++ b/app/main.py
@@ -24,13 +24,13 @@ async def get_user(user_id: int,
     return BaseResponse(result=result)
 
 @app.post("/add_user")
-async def add_user(user_id: int,
-                   username: str,
-                   first_name: str,
-                   last_name: str,
-                   photo_url: str,
+async def add_user(new_user: AddUserBody,
                    mongo: MongoWorker = Depends(MongoWorker)) -> BaseResponse:
-    result = mongo.add_user(user_id, username, first_name, last_name, photo_url)
+    result = mongo.add_user(new_user.user_id,
+                            new_user.username,
+                            new_user.first_name,
+                            new_user.last_name,
+                            new_user.photo_url)
     return BaseResponse(result=result)
 
 
diff --git a/app/schemas/api_schemas.py b/app/schemas/api_schemas.py
index 562312c..386a51a 100644
--- a/app/schemas/api_schemas.py
+++ b/app/schemas/api_schemas.py
@@ -6,3 +6,11 @@ from pydantic import BaseModel
 class BaseResponse(BaseModel):
     result: typing.Any
     error: bool = False
+
+class AddUserBody(BaseModel):
+    user_id: int
+    username: str
+
+    first_name: str
+    last_name: str
+    photo_url: str
\ No newline at end of file