From d44f5c732e6b2bee7450f9489b915c307fa1c2e5 Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Thu, 3 Sep 2020 18:33:15 +0200
Subject: [PATCH] Fix pydantic schemas integer validation

---
 app/schemas.py | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/app/schemas.py b/app/schemas.py
index 941262f..6c4b8a4 100644
--- a/app/schemas.py
+++ b/app/schemas.py
@@ -1,28 +1,29 @@
 from fastapi import Query
 from pydantic import BaseModel, EmailStr
+from typing import Optional
 
 
 class RegisterSchema(BaseModel):
     full_name: str
     email: EmailStr
     password: str
-    gender: int = Query(None, le=1, ge=3)
+    gender: int = Query(None, ge=1, le=3)
     mobile: str = Query(None, min_length=8, max_length=13)
-    user_image: str = None
-    user_type: int = Query(None, le=1, ge=2)
-    lang_type: int = Query(None, le=1, ge=2)
-    device_type: int = Query(None, le=1, ge=2)
+    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
 
 
 class LoginSchema(BaseModel):
     email: EmailStr
     password: str
-    user_type: int = Query(None, le=1, ge=2)
-    lang_type: int = Query(None, le=1, ge=2)
+    user_type: int = Query(None, ge=1, le=2)
+    lang_type: int = Query(None, ge=1, le=2)
     device_id: str
 
 
 class OTPSchema(BaseModel):
-    mobile: str = Query(None, min_length=8, max_length=13)
-    otp: int = Query(None, le=6, ge=6)
+    access_key: str
+    otp: int = Query(None, ge=6, le=6)