mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-26 01:06:23 +02:00
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
"""Schemas for Google Drive connector."""
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class DriveItem(BaseModel):
|
|
"""Represents a Google Drive file or folder."""
|
|
|
|
id: str = Field(..., description="Google Drive item ID")
|
|
name: str = Field(..., description="Item display name")
|
|
|
|
|
|
class GoogleDriveIndexRequest(BaseModel):
|
|
"""Request body for indexing Google Drive content."""
|
|
|
|
folders: list[DriveItem] = Field(
|
|
default_factory=list, description="List of folders to index"
|
|
)
|
|
files: list[DriveItem] = Field(
|
|
default_factory=list, description="List of specific files to index"
|
|
)
|
|
|
|
def has_items(self) -> bool:
|
|
"""Check if any items are selected."""
|
|
return len(self.folders) > 0 or len(self.files) > 0
|
|
|
|
def get_folder_ids(self) -> list[str]:
|
|
"""Get list of folder IDs."""
|
|
return [folder.id for folder in self.folders]
|
|
|
|
def get_folder_names(self) -> list[str]:
|
|
"""Get list of folder names."""
|
|
return [folder.name for folder in self.folders]
|
|
|
|
def get_file_ids(self) -> list[str]:
|
|
"""Get list of file IDs."""
|
|
return [file.id for file in self.files]
|
|
|
|
def get_file_names(self) -> list[str]:
|
|
"""Get list of file names."""
|
|
return [file.name for file in self.files]
|