QueryPal/backend/models/data_documents.py at 40a760ac8bbdf0dc8ee630bd08d1ae2417e01cb7 · ChingEnLin/QueryPal · 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
from pydantic import BaseModel
from typing import Optional, List, Dict, Any


class DataDocumentsFilter(BaseModel):
    key: str
    value: Any = ""  # Can be str, int, etc. depending on the field type
    operator: Optional[str] = "equals"  # 'equals', 'exists', 'not_exists'


class DataDocumentsRequest(BaseModel):
    account_id: str
    database_name: str
    collection_name: str
    page: int = 1
    limit: int = 20
    filter: Optional[DataDocumentsFilter] = None
    filters: Optional[List[DataDocumentsFilter]] = None


class DataDocumentsResponse(BaseModel):
    documents: List[Dict[str, Any]]
    currentPage: int
    totalPages: int
    totalDocuments: int


class FindByIdRequest(BaseModel):
    account_id: str
    database_name: str
    collection_names: List[str]
    document_id: str
    key_context: Optional[str] = None


class FindByIdResponse(BaseModel):
    document: Dict[str, Any]
    collectionName: str


class UpdateDocumentRequest(BaseModel):
    account_id: str
    database_name: str
    collection: str
    id: str
    content: Dict[str, Any]


class SingleDocumentRequest(BaseModel):
    account_id: str
    database_name: str
    collection_name: str
    document_id: str


class InsertDocumentRequest(BaseModel):
    account_id: str
    database_name: str
    collection_name: str
    document: Dict[str, Any]


class DeleteDocumentRequest(BaseModel):
    account_id: str
    database_name: str
    collection_name: str
    document_id: str


# Document History Models
class DocumentHistoryEntry(BaseModel):
    id: str
    user_email: str
    operation: str  # 'insert', 'update', 'delete'
    timestamp_utc: str
    diff_data: Any
    database_name: str
    collection_name: str


class DocumentHistoryRequest(BaseModel):
    account_id: str
    database_name: str
    collection_name: str
    document_id: str


class DocumentHistoryResponse(BaseModel):
    document_id: str
    history_entries: List[DocumentHistoryEntry]
    total_entries: int