Create base schemas for common of attributes

app/schemas.py

@@ -3,27 +3,63 @@ from pydantic import BaseModel, EmailStr
 from typing import Optional
 
 
-class RegisterSchema(BaseModel):
-    full_name: str
+class UserBase(BaseModel):
     email: EmailStr
+    device_id: str
+    lang_type: int = Query(None, ge=1, le=2)
+    user_type: int = Query(None, ge=1, le=2)
+
+
+class UserCreate(UserBase):
+    full_name: str
     password: str
     gender: int = Query(None, ge=1, le=3)
     mobile: str = Query(None, min_length=8, max_length=13)
     user_image: Optional[str] = None
-    user_type: int = Query(None, ge=1, le=2)
-    lang_type: int = Query(None, ge=1, le=2)
     device_type: int = Query(None, ge=1, le=2)
-    device_id: str
+    city_id: int
+    access_key: str
+    badge: int
+
+    class Config:
+        orm_mode = True
 
 
-class LoginSchema(BaseModel):
-    email: EmailStr
+class UserLogin(UserBase):
     password: str
-    user_type: int = Query(None, ge=1, le=2)
-    lang_type: int = Query(None, ge=1, le=2)
-    device_id: str
+
+    class Config:
+        orm_mode = True
 
 
-class OTPSchema(BaseModel):
+class SocialLogin(UserBase):
+    type: int = Query(None, ge=1, le=2)
+    social_id: str
+    device_type: int = Query(None, ge=1, le=2)
+    mobile: str = Query(None, min_length=8, max_length=13)
+    full_name: str
+
+    class Config:
+        orm_mode = True
+
+
+class CreateResponse(UserBase):
+    id: int
+    full_name: str
+    password: str
+    gender: int = Query(None, ge=1, le=3)
+    mobile: str = Query(None, min_length=8, max_length=13)
+    user_image: Optional[str] = None
+    device_type: int = Query(None, ge=1, le=2)
+    otp: int = Query(None, ge=6, le=6)
+
+    class Config:
+        orm_mode = True
+
+
+class OTPVerify(BaseModel):
     access_key: str
     otp: int = Query(None, ge=6, le=6)
+
+    class Config:
+        orm_mode = True