26 lines
1 KiB
Python
26 lines
1 KiB
Python
|
from datetime import datetime, timedelta
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
from pydantic.json import timedelta_isoformat
|
||
|
|
|
||
|
|
|
||
|
|
class ProjectNew(BaseModel):
|
||
|
|
"""Model used when a user creates a new project."""
|
||
|
|
|
||
|
|
name: str = Field(description="Name of the project", example="Learning")
|
||
|
|
|
||
|
|
|
||
|
|
class ProjectInfo(BaseModel):
|
||
|
|
"""Model used when querying information about a module."""
|
||
|
|
|
||
|
|
name: str = Field(description="Name of the project", example="Learning")
|
||
|
|
start_date: datetime = Field(description="Project creation date", example=datetime.now())
|
||
|
|
duration: timedelta = Field(description="Total tracked duration", example=timedelta(days=3, hours=5, minutes=47))
|
||
|
|
records: int = Field(description="Total number of trackings/records", example=42)
|
||
|
|
|
||
|
|
class Config: # TODO: Adding this config may be done with a decorator
|
||
|
|
"""pydantic config"""
|
||
|
|
json_encoders = {
|
||
|
|
# Serialize timedeltas as ISO 8601 and not as float seconds, which is the default
|
||
|
|
timedelta: timedelta_isoformat,
|
||
|
|
}
|