50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
import re
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, EmailStr, Field, model_validator
|
|
|
|
|
|
class UserRegister(BaseModel):
|
|
username: str = Field(min_length=3, max_length=64)
|
|
email: EmailStr = Field()
|
|
phone_number: str = Field(min_length=2, max_length=16, pattern=r'^\+[1-9]\d{1,14}$')
|
|
password: str = Field(min_length=8, max_length=128, examples=["Abc123#!"],
|
|
description="Password must conform to this regex: \n```\n^(?=.*?[A-Z])(?=.*?[a-z])(?=.*?[0-9])(?=.*?[#?!@$ %^&*-]).{8,128}$\n```")
|
|
|
|
@model_validator(mode="after")
|
|
def validate_using_regex(self):
|
|
self.__validate_password()
|
|
return self
|
|
|
|
def __validate_password(self):
|
|
password_regex = r"^(?=.*?[A-Z])(?=.*?[a-z])(?=.*?[0-9])(?=.*?[#?!@$ %^&*-]).{8,128}$"
|
|
if not re.match(password_regex, self.password):
|
|
raise ValueError("Password is too weak")
|
|
|
|
class UserUpdate(BaseModel):
|
|
username: Optional[str] = Field(None, min_length=3, max_length=64)
|
|
first_name: Optional[str] = Field(None, max_length=64)
|
|
last_name: Optional[str] = Field(None, max_length=64)
|
|
email: Optional[EmailStr] = Field(None)
|
|
phone_number: Optional[str] = Field(None, min_length=2, max_length=16, pattern=r'^\+[1-9]\d{1,14}$')
|
|
password: Optional[str] = Field(None, min_length=8, max_length=128, examples=["Abc123#!"],
|
|
description="Password must conform to this regex: \n```\n^(?=.*?[A-Z])(?=.*?[a-z])(?=.*?[0-9])(?=.*?[#?!@$ %^&*-]).{8,128}$\n```")
|
|
|
|
@model_validator(mode="after")
|
|
def validate_using_regex(self):
|
|
self.__validate_password()
|
|
return self
|
|
|
|
def __validate_password(self):
|
|
password_regex = r"^(?=.*?[A-Z])(?=.*?[a-z])(?=.*?[0-9])(?=.*?[#?!@$ %^&*-]).{8,128}$"
|
|
if not re.match(password_regex, self.password):
|
|
raise ValueError("Password is too weak")
|
|
|
|
class Token(BaseModel):
|
|
access_token: str
|
|
token_type: str = "bearer"
|
|
|
|
|
|
class TokenPayload(BaseModel):
|
|
sub: str | None = None
|