58 lines
3.2 KiB
Python
58 lines
3.2 KiB
Python
# Copyright (c) 2025 Patrick Motsch
|
|
# All rights reserved.
|
|
"""DataSource and ExternalEntry models for external data integration.
|
|
|
|
DataSource links a UserConnection to an external path (SharePoint folder,
|
|
Google Drive folder, FTP directory, etc.) for agent-accessible data containers.
|
|
"""
|
|
|
|
from typing import Dict, Any, Optional
|
|
from pydantic import BaseModel, Field
|
|
from modules.shared.attributeUtils import registerModelLabels
|
|
from modules.shared.timeUtils import getUtcTimestamp
|
|
import uuid
|
|
|
|
|
|
class DataSource(BaseModel):
|
|
"""Configured external data source linked to a UserConnection."""
|
|
id: str = Field(default_factory=lambda: str(uuid.uuid4()), description="Primary key")
|
|
connectionId: str = Field(description="FK to UserConnection")
|
|
sourceType: str = Field(description="sharepointFolder, googleDriveFolder, outlookFolder, ftpFolder")
|
|
path: str = Field(description="External path (e.g. '/sites/MySite/Documents/Reports')")
|
|
label: str = Field(description="User-visible label")
|
|
featureInstanceId: Optional[str] = Field(default=None, description="Scoped to feature instance")
|
|
mandateId: Optional[str] = Field(default=None, description="Mandate scope")
|
|
userId: str = Field(default="", description="Owner user ID")
|
|
autoSync: bool = Field(default=False, description="Automatically sync on schedule")
|
|
lastSynced: Optional[float] = Field(default=None, description="Last sync timestamp")
|
|
createdAt: float = Field(default_factory=getUtcTimestamp, description="Creation timestamp")
|
|
|
|
|
|
registerModelLabels(
|
|
"DataSource",
|
|
{"en": "Data Source", "de": "Datenquelle", "fr": "Source de données"},
|
|
{
|
|
"id": {"en": "ID", "de": "ID", "fr": "ID"},
|
|
"connectionId": {"en": "Connection ID", "de": "Verbindungs-ID", "fr": "ID de connexion"},
|
|
"sourceType": {"en": "Source Type", "de": "Quellentyp", "fr": "Type de source"},
|
|
"path": {"en": "Path", "de": "Pfad", "fr": "Chemin"},
|
|
"label": {"en": "Label", "de": "Bezeichnung", "fr": "Libellé"},
|
|
"featureInstanceId": {"en": "Feature Instance", "de": "Feature-Instanz", "fr": "Instance de fonctionnalité"},
|
|
"mandateId": {"en": "Mandate ID", "de": "Mandanten-ID", "fr": "ID du mandat"},
|
|
"userId": {"en": "User ID", "de": "Benutzer-ID", "fr": "ID utilisateur"},
|
|
"autoSync": {"en": "Auto Sync", "de": "Auto-Sync", "fr": "Synchro auto"},
|
|
"lastSynced": {"en": "Last Synced", "de": "Letzter Sync", "fr": "Dernier sync"},
|
|
"createdAt": {"en": "Created At", "de": "Erstellt am", "fr": "Créé le"},
|
|
},
|
|
)
|
|
|
|
|
|
class ExternalEntry(BaseModel):
|
|
"""An item (file or folder) from an external data source."""
|
|
name: str = Field(description="Item name")
|
|
path: str = Field(description="Full path within the source")
|
|
isFolder: bool = Field(default=False, description="True if directory/folder")
|
|
size: Optional[int] = Field(default=None, description="File size in bytes")
|
|
mimeType: Optional[str] = Field(default=None, description="MIME type (files only)")
|
|
lastModified: Optional[float] = Field(default=None, description="Last modification timestamp")
|
|
metadata: Dict[str, Any] = Field(default_factory=dict, description="Provider-specific metadata")
|