insforge-python/insforge/storage/models.py at ea26659f4b0955ce92259511f735ec642e69aa2f · InsForge/insforge-python · GitHub

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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
from __future__ import annotations

from datetime import datetime
from dataclasses import dataclass
from typing import Any

from pydantic import BaseModel, ConfigDict, Field


class StorageBucketListResponse(BaseModel):
    model_config = ConfigDict(extra="ignore")

    buckets: list[str] = Field(default_factory=list)


class StorageBucketResponse(BaseModel):
    model_config = ConfigDict(extra="ignore")

    message: str
    bucket_name: str = Field(alias="bucketName")


class StorageBucketUpdateResponse(BaseModel):
    model_config = ConfigDict(extra="ignore")

    message: str
    bucket: str
    is_public: bool = Field(alias="isPublic")


class StorageObjectResponse(BaseModel):
    model_config = ConfigDict(extra="ignore", populate_by_name=True)

    bucket: str
    key: str
    size: int
    mime_type: str | None = Field(default=None, alias="mimeType")
    uploaded_at: datetime = Field(alias="uploadedAt")
    url: str


class StorageDeleteObjectResponse(BaseModel):
    model_config = ConfigDict(extra="ignore")

    message: str


class StoragePagination(BaseModel):
    model_config = ConfigDict(extra="ignore")

    limit: int | None = None
    offset: int | None = None
    total: int | None = None


class StoredFileList(BaseModel):
    model_config = ConfigDict(extra="ignore", populate_by_name=True)

    data: list[StorageObjectResponse] = Field(default_factory=list)
    pagination: StoragePagination | None = None
    next_actions: str | None = Field(default=None, alias="nextActions")


class UploadStrategy(BaseModel):
    model_config = ConfigDict(extra="ignore", populate_by_name=True)

    method: str
    upload_url: str = Field(alias="uploadUrl")
    key: str
    confirm_required: bool = Field(alias="confirmRequired")
    confirm_url: str | None = Field(default=None, alias="confirmUrl")
    expires_at: datetime | None = Field(default=None, alias="expiresAt")
    fields: dict[str, Any] | None = None


class DownloadStrategy(BaseModel):
    model_config = ConfigDict(extra="ignore", populate_by_name=True)

    method: str
    url: str
    expires_at: datetime | None = Field(default=None, alias="expiresAt")
    headers: dict[str, str] | None = None


class StorageDeleteBucketResponse(BaseModel):
    model_config = ConfigDict(extra="ignore")

    message: str
    next_actions: str | None = Field(default=None, alias="nextActions")


@dataclass
class StorageDownloadResult:
    content: bytes
    content_type: str | None = None
    content_length: str | None = None