From 937b69ffcc08b8dd2bf8992f31c813ef121f70fd Mon Sep 17 00:00:00 2001 From: msramalho <19508417+msramalho@users.noreply.github.com> Date: Sat, 8 Feb 2025 23:33:02 +0000 Subject: [PATCH] new schemas --- src/db/schemas.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/src/db/schemas.py b/src/db/schemas.py index aff5730..b76e711 100644 --- a/src/db/schemas.py +++ b/src/db/schemas.py @@ -7,12 +7,12 @@ from datetime import datetime class SubmitSheet(BaseModel): sheet_id: str | None author_id: str | None = None - group_id: str | None + group_id: str = "default" tags: set[str] | None = set() columns: dict | None = {} # TODO: implement/remove -class SubmitManual(BaseModel): +class SubmitManual(BaseModel): # deprecated result: str # should be a Metadata.to_json() public: bool = False author_id: str | None = None @@ -54,7 +54,7 @@ class TaskDelete(Task): deleted: bool -class ActiveUser(BaseModel): +class ActiveUser(BaseModel): active: bool @@ -78,6 +78,7 @@ class ArchiveTrigger(BaseModel): group_id: Annotated[str, Len(min_length=1)] = "default" tags: set[str] | None = None + class ArchiveCreate(ArchiveTrigger): id: str | None = None result: dict | None = None @@ -85,6 +86,7 @@ class ArchiveCreate(ArchiveTrigger): urls: list | None = None store_until: datetime | None = None + class Archive(ArchiveCreate): created_at: datetime updated_at: datetime | None @@ -98,11 +100,18 @@ class Usage(BaseModel): monthly_mbs: int = 0 total_sheets: int = 0 + class UsageResponse(Usage): groups: dict[str, Usage] + class CelerySheetTask(BaseModel): success: bool sheet_id: str time: datetime - stats: dict \ No newline at end of file + stats: dict + + +class SubmitManualArchive(ArchiveTrigger): + url: None = None + result: str # should be a Metadata.to_json()