diff --git a/app/mongo_worker.py b/app/mongo_worker.py index 69ac46f..7f6c80d 100644 --- a/app/mongo_worker.py +++ b/app/mongo_worker.py @@ -29,12 +29,12 @@ class MongoWorker: else: return False - def add_user(self, user_id: int, username: str, first_name: str, last_name: str, photo_path: str) -> User: + def add_user(self, user_id: int, username: str, first_name: str, last_name: str, photo_id: str) -> User: new_user = User(user_id=user_id, username=username, first_name=first_name, last_name=last_name, - photo_path=photo_path, + photo_id=photo_id, registration_date=datetime.now().isoformat()) try: res = self.users_data.insert_one(new_user.model_dump()) @@ -51,5 +51,5 @@ if __name__ == "__main__": mongo = MongoWorker() mongo.get_mongodb_info() print(mongo.check_user(123)) - # print(mongo.add_user(123, "VolochayIgor", "Igor", "Volochay", os.path.join("/path/to", "photo.jpg"))) + # print(mongo.add_user(123, "VolochayIgor", "Igor", "Volochay", "photo_0.jpg")) print(mongo.get_user(123)) diff --git a/app/schemas.py b/app/schemas.py index aa68d8b..9e2b63a 100644 --- a/app/schemas.py +++ b/app/schemas.py @@ -1,3 +1,5 @@ +import typing + from pydantic import BaseModel @@ -7,7 +9,7 @@ class User(BaseModel): first_name: str last_name: str - photo_path: str + photo_id: str activity: int = 0 liked_post_ids: list[int] = list() @@ -15,3 +17,7 @@ class User(BaseModel): comments_ids: list[int] = list() registration_date: str + +class BaseResponse(BaseModel): + result: typing.Any + error: bool = False