0ceeef31a6
- Add user model with relationship to tasks - Implement JWT token authentication - Create user registration and login endpoints - Update task endpoints to filter by current user - Add Alembic migration for user table - Update documentation with authentication details
62 lines
1.4 KiB
Python
62 lines
1.4 KiB
Python
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, EmailStr, Field, validator
|
|
|
|
|
|
class UserBase(BaseModel):
|
|
email: Optional[EmailStr] = None
|
|
username: Optional[str] = None
|
|
is_active: Optional[bool] = True
|
|
is_superuser: Optional[bool] = False
|
|
|
|
|
|
class UserCreate(UserBase):
|
|
email: EmailStr = Field(..., description="User email")
|
|
username: str = Field(..., min_length=3, max_length=50, description="Username")
|
|
password: str = Field(..., min_length=8, description="Password")
|
|
|
|
@validator("username")
|
|
def username_valid(cls, v):
|
|
# Additional validation for username
|
|
if not v.isalnum():
|
|
raise ValueError("Username must be alphanumeric")
|
|
return v
|
|
|
|
|
|
class UserUpdate(UserBase):
|
|
password: Optional[str] = Field(None, min_length=8, description="Password")
|
|
|
|
|
|
class UserInDBBase(UserBase):
|
|
id: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class User(UserInDBBase):
|
|
"""Return user information without sensitive data"""
|
|
|
|
pass
|
|
|
|
|
|
class UserInDB(UserInDBBase):
|
|
"""User model stored in DB, with hashed password"""
|
|
|
|
hashed_password: str
|
|
|
|
|
|
class Token(BaseModel):
|
|
"""OAuth2 compatible token"""
|
|
|
|
access_token: str
|
|
token_type: str
|
|
|
|
|
|
class TokenPayload(BaseModel):
|
|
"""JWT token payload"""
|
|
|
|
sub: Optional[int] = None
|