File size: 2,370 Bytes
f30e75c 9e798a1 7c11a75 9e798a1 212bed8 9e798a1 f30e75c 7c11a75 6c582f0 f30e75c 7c11a75 f30e75c 9e798a1 41eef54 9e798a1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
from pydantic import BaseModel, Field, constr, ConfigDict
from typing import Optional
from datetime import datetime
from passlib.context import CryptContext
from decimal import Decimal
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
# Constants for phone and MAC address patterns
PHONE_PATTERN = r"^255\d{9}$"
MAC_PATTERN = r"^([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2})$"
class UserResponse(BaseModel):
id: str
name: str
phoneNumber: str
balance: Decimal
account_locked: bool
model_config = ConfigDict(from_attributes=True)
# class Config:
# orm_mode = True
# Register User Request
class RegisterUserRequest(BaseModel):
name: str = Field(..., max_length=100)
password: str = Field(..., max_length=100)
phoneNumber: str = Field(
...,
pattern=PHONE_PATTERN,
description="Tanzanian phone number starting with +255 or 0 followed by 9 digits",
)
mac_address: str = Field(
..., pattern=MAC_PATTERN, description="MAC address in standard format"
)
def hash_password(self):
self.password = pwd_context.hash(self.password)
class Config:
schema_extra = {
"example": {
"name": "John Doe",
"password": "StrongPassword1!",
"phoneNumber": "+255123456789",
"mac_address": "00:1A:2B:3C:4D:5E",
}
}
# Login User Request
class LoginUserRequest(BaseModel):
phoneNumber: str = Field(..., pattern=PHONE_PATTERN)
password: str
# grant_type: Optional[str] = None
mac_address: Optional[str] = Field(..., pattern=MAC_PATTERN)
# Access Token Response
class AccessTokenResponse(BaseModel):
access_token: str
token_type: str = "bearer"
# Base Response Schema
class BaseResponse(BaseModel):
code: int
message: str
payload: Optional[dict] = None
# Forgot Password Request
class ForgotPasswordRequest(BaseModel):
phoneNumber: str = Field(..., pattern=PHONE_PATTERN)
# Verify Reset Token Request
class VerifyResetTokenRequest(BaseModel):
phoneNumber: str = Field(..., pattern=PHONE_PATTERN)
reset_token: str = Field(..., max_length=6)
# Reset Password Request
class ResetPasswordRequest(BaseModel):
phoneNumber: str = Field(..., pattern=PHONE_PATTERN)
new_password: str = Field(..., max_length=100)
|