163 lines
3.7 KiB
Python
163 lines
3.7 KiB
Python
from pydantic import BaseModel, EmailStr, Field, ConfigDict
|
|
from typing import Optional
|
|
from datetime import datetime, date
|
|
from ..models.models import UserRole, MembershipStatus, PaymentStatus, PaymentMethod
|
|
|
|
|
|
# User Schemas
|
|
class UserBase(BaseModel):
|
|
email: EmailStr
|
|
first_name: str = Field(..., min_length=1, max_length=100)
|
|
last_name: str = Field(..., min_length=1, max_length=100)
|
|
phone: Optional[str] = None
|
|
address: Optional[str] = None
|
|
|
|
|
|
class UserCreate(UserBase):
|
|
password: str = Field(..., min_length=8)
|
|
|
|
|
|
class UserUpdate(BaseModel):
|
|
first_name: Optional[str] = Field(None, min_length=1, max_length=100)
|
|
last_name: Optional[str] = Field(None, min_length=1, max_length=100)
|
|
phone: Optional[str] = None
|
|
address: Optional[str] = None
|
|
|
|
|
|
class UserResponse(UserBase):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
role: UserRole
|
|
is_active: bool
|
|
created_at: datetime
|
|
last_login: Optional[datetime] = None
|
|
|
|
|
|
class UserInDB(UserResponse):
|
|
hashed_password: str
|
|
|
|
|
|
# Authentication Schemas
|
|
class Token(BaseModel):
|
|
access_token: str
|
|
token_type: str = "bearer"
|
|
|
|
|
|
class TokenData(BaseModel):
|
|
user_id: Optional[int] = None
|
|
|
|
|
|
class LoginRequest(BaseModel):
|
|
email: EmailStr
|
|
password: str
|
|
|
|
|
|
# Password Reset Schemas
|
|
class ForgotPasswordRequest(BaseModel):
|
|
email: EmailStr
|
|
|
|
|
|
class ResetPasswordRequest(BaseModel):
|
|
token: str = Field(..., min_length=1)
|
|
new_password: str = Field(..., min_length=8)
|
|
|
|
|
|
# Membership Tier Schemas
|
|
class MembershipTierBase(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=100)
|
|
description: Optional[str] = None
|
|
annual_fee: float = Field(..., ge=0)
|
|
benefits: Optional[str] = None
|
|
|
|
|
|
class MembershipTierCreate(MembershipTierBase):
|
|
pass
|
|
|
|
|
|
class MembershipTierUpdate(BaseModel):
|
|
name: Optional[str] = Field(None, min_length=1, max_length=100)
|
|
description: Optional[str] = None
|
|
annual_fee: Optional[float] = Field(None, ge=0)
|
|
benefits: Optional[str] = None
|
|
is_active: Optional[bool] = None
|
|
|
|
|
|
class MembershipTierResponse(MembershipTierBase):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
is_active: bool
|
|
created_at: datetime
|
|
|
|
|
|
# Membership Schemas
|
|
class MembershipBase(BaseModel):
|
|
tier_id: int
|
|
auto_renew: bool = False
|
|
|
|
|
|
class MembershipCreate(MembershipBase):
|
|
start_date: date
|
|
end_date: date
|
|
|
|
|
|
class MembershipUpdate(BaseModel):
|
|
tier_id: Optional[int] = None
|
|
status: Optional[MembershipStatus] = None
|
|
end_date: Optional[date] = None
|
|
auto_renew: Optional[bool] = None
|
|
|
|
|
|
class MembershipResponse(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
user_id: int
|
|
tier_id: int
|
|
status: MembershipStatus
|
|
start_date: date
|
|
end_date: date
|
|
auto_renew: bool
|
|
created_at: datetime
|
|
tier: MembershipTierResponse
|
|
|
|
|
|
# Payment Schemas
|
|
class PaymentBase(BaseModel):
|
|
amount: float = Field(..., gt=0)
|
|
payment_method: PaymentMethod
|
|
notes: Optional[str] = None
|
|
|
|
|
|
class PaymentCreate(PaymentBase):
|
|
membership_id: Optional[int] = None
|
|
|
|
|
|
class PaymentUpdate(BaseModel):
|
|
status: Optional[PaymentStatus] = None
|
|
transaction_id: Optional[str] = None
|
|
payment_date: Optional[datetime] = None
|
|
notes: Optional[str] = None
|
|
|
|
|
|
class PaymentResponse(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
user_id: int
|
|
membership_id: Optional[int] = None
|
|
amount: float
|
|
payment_method: PaymentMethod
|
|
status: PaymentStatus
|
|
transaction_id: Optional[str] = None
|
|
payment_date: Optional[datetime] = None
|
|
notes: Optional[str] = None
|
|
created_at: datetime
|
|
|
|
|
|
# Message Response
|
|
class MessageResponse(BaseModel):
|
|
message: str
|
|
detail: Optional[str] = None
|