Spaces:
Sleeping
Sleeping
| """ | |
| DataFlow Server — Pydantic Schemas | |
| """ | |
| from pydantic import BaseModel, EmailStr | |
| from typing import Optional, List | |
| from datetime import datetime | |
| # ─── Auth ─── | |
| class UserLogin(BaseModel): | |
| email: str | |
| password: str | |
| class UserRegister(BaseModel): | |
| full_name: str | |
| email: str | |
| password: str | |
| position: Optional[str] = None | |
| role: str = "developer" | |
| class Token(BaseModel): | |
| access_token: str | |
| token_type: str = "bearer" | |
| user_id: str | |
| full_name: str | |
| role: str | |
| class UserOut(BaseModel): | |
| id: str | |
| full_name: str | |
| email: str | |
| position: Optional[str] | |
| role: str | |
| language: str | |
| currency: str | |
| timezone: str | |
| two_factor_enabled: bool | |
| biometric_enabled: bool | |
| model_config = {"from_attributes": True} | |
| # ─── Project ─── | |
| class ProjectCreate(BaseModel): | |
| name: str | |
| description: str = "" | |
| start_date: datetime | |
| end_date: datetime | |
| budget: float | |
| status: str = "active" | |
| class ProjectUpdate(BaseModel): | |
| name: Optional[str] = None | |
| description: Optional[str] = None | |
| end_date: Optional[datetime] = None | |
| budget: Optional[float] = None | |
| spent: Optional[float] = None | |
| status: Optional[str] = None | |
| progress: Optional[float] = None | |
| risk_level: Optional[str] = None | |
| class ProjectOut(BaseModel): | |
| id: str | |
| name: str | |
| description: str | |
| start_date: datetime | |
| end_date: datetime | |
| budget: float | |
| spent: float | |
| status: str | |
| progress: float | |
| risk_level: str | |
| owner_id: Optional[str] | |
| created_at: datetime | |
| model_config = {"from_attributes": True} | |
| # ─── Task ─── | |
| class SubTaskOut(BaseModel): | |
| id: str | |
| title: str | |
| completed: bool | |
| model_config = {"from_attributes": True} | |
| class TaskCreate(BaseModel): | |
| title: str | |
| description: str = "" | |
| project_id: str | |
| status: str = "todo" | |
| priority: str = "medium" | |
| assignee_id: Optional[str] = None | |
| assignee_name: Optional[str] = None | |
| deadline: Optional[datetime] = None | |
| tags: str = "" | |
| class TaskUpdate(BaseModel): | |
| title: Optional[str] = None | |
| status: Optional[str] = None | |
| priority: Optional[str] = None | |
| assignee_id: Optional[str] = None | |
| assignee_name: Optional[str] = None | |
| deadline: Optional[datetime] = None | |
| class TaskOut(BaseModel): | |
| id: str | |
| title: str | |
| description: str | |
| project_id: str | |
| status: str | |
| priority: str | |
| assignee_id: Optional[str] | |
| assignee_name: Optional[str] | |
| deadline: Optional[datetime] | |
| created_at: datetime | |
| tags: str | |
| subtasks: List[SubTaskOut] = [] | |
| model_config = {"from_attributes": True} | |
| # ─── Expense ─── | |
| class ExpenseCreate(BaseModel): | |
| project_id: str | |
| title: str | |
| amount: float | |
| category: str | |
| date: Optional[datetime] = None | |
| description: Optional[str] = None | |
| class ExpenseUpdate(BaseModel): | |
| status: Optional[str] = None | |
| amount: Optional[float] = None | |
| title: Optional[str] = None | |
| class ExpenseOut(BaseModel): | |
| id: str | |
| project_id: str | |
| title: str | |
| amount: float | |
| category: str | |
| date: datetime | |
| status: str | |
| description: Optional[str] | |
| receipt_url: Optional[str] | |
| created_at: datetime | |
| model_config = {"from_attributes": True} | |
| # ─── Document ─── | |
| class DocumentOut(BaseModel): | |
| id: str | |
| name: str | |
| project_id: str | |
| type: str | |
| size_bytes: int | |
| is_favorite: bool | |
| extracted_text: Optional[str] | |
| recognition_confidence: Optional[float] | |
| detected_type: Optional[str] | |
| extracted_amount: Optional[float] | |
| extracted_date: Optional[datetime] | |
| extracted_supplier: Optional[str] | |
| uploaded_at: datetime | |
| model_config = {"from_attributes": True} | |
| # ─── Notification ─── | |
| class NotificationOut(BaseModel): | |
| id: str | |
| title: str | |
| body: str | |
| type: str | |
| is_read: bool | |
| created_at: datetime | |
| model_config = {"from_attributes": True} | |
| # ─── AI Forecast ─── | |
| class ForecastOut(BaseModel): | |
| id: str | |
| project_id: str | |
| current_spent: float | |
| planned_budget: float | |
| risk_level: str | |
| overrun_probability: float | |
| optimistic_forecast: float | |
| realistic_forecast: float | |
| pessimistic_forecast: float | |
| recommendations: str | |
| generated_at: datetime | |
| model_config = {"from_attributes": True} | |
| class ForecastRequest(BaseModel): | |
| project_id: str | |
| class OCRRequest(BaseModel): | |
| project_id: str | |