add email to user registration

src/controllers/auth.py

@@ -36,7 +36,7 @@ def __init__(
         self.db_session = db_session
         self.redis_session = redis_session
 
-    async def register(self, password: str, username: str) -> UserOutRegister:
+    async def register(self, password: str, username: str, email: str) -> UserOutRegister:
         user = await self.user_crud.get_by_username(
             self.db_session,
             username=username,
@@ -50,6 +50,7 @@ async def register(self, password: str, username: str) -> UserOutRegister:
             db_session=self.db_session,
             username=username,
             password=password,
+            email=email,
             gauth=str(random_base32()),
         )
 
@@ -62,6 +63,7 @@ async def register(self, password: str, username: str) -> UserOutRegister:
             updated_at=user.updated_at,
             created_at=user.created_at,
             gauth=user.gauth,
+            email=user.email,
             qr_img=base64.b64encode(buffered.getvalue()).decode(),
         )
 
@@ -108,6 +110,7 @@ async def me(self, user_id) -> UserOut:
         return UserOut(
             id=user.id,
             username=user.username,
+            email=user.email,
             updated_at=user.updated_at,
             created_at=user.created_at,
         )

src/routers/auth.py

@@ -7,7 +7,7 @@
     get_current_user_from_db,
     get_current_user_with_refresh,
 )
-from src.schema.user import UserIn, UserOut, UserOutRegister
+from src.schema.user import UserRegister, UserOut, UserOutRegister, UserLogin
 
 router = APIRouter(
     prefix="/auth",
@@ -21,14 +21,14 @@
     "/register", description="register new user", response_model=UserOutRegister
 )
 async def register(
-    data: UserIn, db_session: DBManager = Depends(get_db)
+    data: UserRegister, db_session: DBManager = Depends(get_db)
 ) -> UserOutRegister:
     return await AuthController(db_session=db_session).register(**data.dict())
 
 
 @router.post("/login", description="Create access and refresh tokens for verified user")
 async def login(
-    data: UserIn,
+    data: UserLogin,
     request: Request,
     response: Response,
     db_session: DBManager = Depends(get_db),