File size: 1,885 Bytes
6bed18e | 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 | from datetime import datetime
from typing import Optional
from sqlmodel import SQLModel, Field
from sqlalchemy import select
from sqlmodel import Session
from pydantic import ConfigDict
class TaskBase(SQLModel):
title: str = Field(min_length=1, max_length=255)
description: Optional[str] = Field(default=None, max_length=1000)
completed: bool = Field(default=False)
user_id: str = Field(max_length=255)
class Task(TaskBase, table=True):
id: Optional[int] = Field(default=None, primary_key=True)
created_at: datetime = Field(default_factory=datetime.utcnow)
updated_at: datetime = Field(default_factory=datetime.utcnow)
model_config = ConfigDict(from_attributes=True)
@classmethod
def get_by_user_id(cls, session: Session, user_id: str):
"""
Class method to get all tasks for a specific user
"""
statement = select(cls).where(cls.user_id == user_id)
result = session.exec(statement)
tasks = result.all()
return tasks
@classmethod
def get_by_id_and_user_id(cls, session: Session, task_id: int, user_id: str):
"""
Class method to get a specific task for a specific user (for data isolation)
"""
statement = select(cls).where(cls.id == task_id, cls.user_id == user_id)
return session.exec(statement).first()
class TaskCreate(TaskBase):
user_id: Optional[str] = Field(default=None, max_length=255) # Override to make optional for creation
class TaskUpdate(SQLModel):
title: Optional[str] = Field(default=None, min_length=1, max_length=255)
description: Optional[str] = Field(default=None, max_length=1000)
completed: Optional[bool] = None
from pydantic import ConfigDict
class TaskResponse(TaskBase):
id: int
created_at: datetime
updated_at: datetime
model_config = ConfigDict(from_attributes=True) |