"""
Linux BenchTools - Document Schemas
"""

from pydantic import BaseModel
from typing import List


class DocumentResponse(BaseModel):
    """Document response"""
    id: int
    device_id: int
    doc_type: str
    filename: str
    mime_type: str
    size_bytes: int
    uploaded_at: str

    class Config:
        from_attributes = True


class DocumentListResponse(BaseModel):
    """List of documents"""
    items: List[DocumentResponse] = []