Merge pull request #14 from Querent-ai/connector_base

Added Connector base

Author: saraswatpuneet

querent/connectors/base_connector.py renamed to querent/config/connector_config.py

@@ -0,0 +1,35 @@
+from typing import List, Dict
+from pathlib import Path
+from enum import Enum
+
+class ConnectorErrorKind(Enum):
+    NotFound = "not_found"
+    Unauthorized = "unauthorized"
+    Service = "noincompatible"
+
+
+
+class ConnectorResolverError(Exception):
+    def __init__(self, kind: ConnectorErrorKind, message: str):
+        super().__init__(message)
+        self.kind = kind
+
+
+class ConnectorError(Exception):
+    def __init__(self, kind: ConnectorErrorKind, source: any):
+        self.kind = kind
+        self.source = source
+
+
+class NotFoundError(ConnectorError):
+    def __init__(self, message: str):
+        super().__init__(ConnectorErrorKind.NotFound, message)
+
+class unauthorizedError(ConnectorError):
+    def __init__(self, message: str):
+        super().__init__(ConnectorErrorKind.unauthorized, message)
+
+
+class incompatibleError(ConnectorError):
+    def __init__(self, message: str):
+        super().__init__(ConnectorErrorKind.incompatible, message)

querent/connectors/connecror.errors.py

@@ -0,0 +1,27 @@
+from typing import Dict, Optional, Union
+from querent.config.connector_config import ConnectorBackend
+from querent.connectors.connector_base import Connector
+from querent.connectors.connector_errors import ConnectorResolverError, ConnectorErrorKind
+from querent.common.uri import Protocol, Uri
+from querent.connectors.connector_factory import ConnectorFactory
+class ConnectorResolver:
+    def __init__(self):
+        pass
+
+    def resolve(self, src_uri: Uri, dest_uri: Uri) -> Optional[Connector]:
+        src_backend = self._determine_backend(src_uri.protocol)
+        dest_backend = self._determine_backend(src_dest.protocol)
+        if backend in self.connector_factories:
+            return self.connector_factories[src_backend].resolve(uri), 
+        else:
+            raise ConnectorResolverError(
+                ConnectorErrorKind.NotSupported, backend, "Unsupported backend"
+            )
+
+    def _determine_backend(self, protocol: Protocol) -> ConnectorBackend:
+        if protocol.is_file_Connector():
+            return ConnectorBackend.LocalFile
+        else:
+            raise ConnectorResolverError(
+                ConnectorErrorKind.NotSupported, "Unknown backend", "Unknown backend"
+            )