amsdal 0.5.19__cp312-cp312-macosx_10_13_universal2.whl → 0.5.21__cp312-cp312-macosx_10_13_universal2.whl

amsdal/services/external_connections.py

@@ -0,0 +1,262 @@
+"""
+External Connection Manager for accessing external services and databases.
+
+This module provides a high-level interface for working with external connections
+such as read-only databases, email services, storage services, etc.
+"""
+
+from typing import Any
+from typing import TypeVar
+
+from amsdal_data.application import AsyncDataApplication
+from amsdal_data.application import DataApplication
+from amsdal_utils.utils.singleton import Singleton
+
+T = TypeVar('T')
+
+
+class ExternalConnectionManager(metaclass=Singleton):
+    """
+    Manager for accessing external service connections.
+
+    Provides a convenient interface to access external connections configured
+    in the application, such as read-only databases, email services, etc.
+
+    Example usage:
+        manager = ExternalConnectionManager()
+
+        # Get read-only database connection
+        external_db = manager.get_connection('external_users_db')
+        rows = external_db.fetch_all('SELECT * FROM users WHERE active = 1')
+
+        # Get email service
+        email = manager.get_connection('email_service')
+        email.send_email(...)
+    """
+
+    def __init__(self) -> None:
+        self._data_application: DataApplication | None = None
+        self._async_data_application: AsyncDataApplication | None = None
+
+    def setup(
+        self,
+        data_application: DataApplication | None = None,
+        async_data_application: AsyncDataApplication | None = None,
+    ) -> None:
+        """
+        Set up the manager with the data application instance.
+
+        Args:
+            data_application: Sync DataApplication instance
+            async_data_application: Async DataApplication instance
+        """
+        self._data_application = data_application
+        self._async_data_application = async_data_application
+
+    def get_connection(self, name: str) -> Any:
+        """
+        Get an external service connection by name.
+
+        Args:
+            name: Name of the external connection (as configured in resources)
+
+        Returns:
+            The external connection object
+
+        Raises:
+            RuntimeError: If manager is not set up
+            KeyError: If connection not found
+        """
+        if self._data_application is None and self._async_data_application is None:
+            msg = 'ExternalConnectionManager not set up. Call setup() first.'
+            raise RuntimeError(msg)
+
+        app = self._data_application or self._async_data_application
+        if app is None:  # Shouldn't happen due to check above, but satisfy mypy
+            msg = 'No data application available'
+            raise RuntimeError(msg)
+        return app.get_external_service_connection(name)
+
+    def has_connection(self, name: str) -> bool:
+        """
+        Check if an external connection exists.
+
+        Args:
+            name: Name of the external connection
+
+        Returns:
+            bool: True if connection exists, False otherwise
+        """
+        if self._data_application is None and self._async_data_application is None:
+            return False
+
+        try:
+            self.get_connection(name)
+            return True
+        except KeyError:
+            return False
+
+    def list_connections(self) -> list[str]:
+        """
+        List all available external connection names.
+
+        Returns:
+            list[str]: List of connection names
+        """
+        if self._data_application:
+            return list(self._data_application._external_service_connections.keys())  # noqa: SLF001
+        if self._async_data_application:
+            return list(self._async_data_application._external_service_connections.keys())  # noqa: SLF001
+        return []
+
+
+class ExternalDatabaseReader:
+    """
+    Helper class for reading from external read-only databases.
+
+    Provides a convenient interface for querying external databases
+    with common patterns like filtering, mapping results, etc.
+
+    Example usage:
+        reader = ExternalDatabaseReader('external_users_db')
+
+        # Fetch all users
+        users = reader.fetch_all('SELECT * FROM users')
+
+        # Fetch with parameters
+        active_users = reader.fetch_all(
+            'SELECT * FROM users WHERE active = ?',
+            (1,)
+        )
+
+        # Fetch one record
+        user = reader.fetch_one('SELECT * FROM users WHERE id = ?', (user_id,))
+
+        # Get as dictionaries
+        user_dicts = reader.fetch_all_as_dicts('SELECT * FROM users LIMIT 10')
+    """
+
+    def __init__(self, connection_name: str):
+        """
+        Initialize the reader with a connection name.
+
+        Args:
+            connection_name: Name of the external database connection
+        """
+        self.connection_name = connection_name
+        self._manager = ExternalConnectionManager()
+
+    @property
+    def connection(self) -> Any:
+        """Get the underlying connection object."""
+        return self._manager.get_connection(self.connection_name)
+
+    def fetch_all(self, query: str, parameters: tuple[Any, ...] | None = None) -> list[Any]:
+        """
+        Execute query and fetch all results.
+
+        Args:
+            query: SQL query to execute
+            parameters: Query parameters (optional)
+
+        Returns:
+            list: List of result rows
+        """
+        return self.connection.fetch_all(query, parameters)
+
+    def fetch_one(self, query: str, parameters: tuple[Any, ...] | None = None) -> Any | None:
+        """
+        Execute query and fetch one result.
+
+        Args:
+            query: SQL query to execute
+            parameters: Query parameters (optional)
+
+        Returns:
+            Single result row or None
+        """
+        return self.connection.fetch_one(query, parameters)
+
+    def fetch_all_as_dicts(self, query: str, parameters: tuple[Any, ...] | None = None) -> list[dict[str, Any]]:
+        """
+        Execute query and fetch all results as dictionaries.
+
+        Args:
+            query: SQL query to execute
+            parameters: Query parameters (optional)
+
+        Returns:
+            list[dict]: List of result dictionaries
+        """
+        rows = self.fetch_all(query, parameters)
+        return [dict(row) for row in rows]
+
+    def fetch_one_as_dict(self, query: str, parameters: tuple[Any, ...] | None = None) -> dict[str, Any] | None:
+        """
+        Execute query and fetch one result as dictionary.
+
+        Args:
+            query: SQL query to execute
+            parameters: Query parameters (optional)
+
+        Returns:
+            dict | None: Result dictionary or None
+        """
+        row = self.fetch_one(query, parameters)
+        return dict(row) if row else None
+
+    def get_table_names(self) -> list[str]:
+        """
+        Get list of all tables in the database.
+
+        Returns:
+            list[str]: List of table names
+        """
+        return self.connection.get_table_names()
+
+    def get_table_schema(self, table_name: str) -> list[dict[str, Any]]:
+        """
+        Get schema information for a table.
+
+        Args:
+            table_name: Name of the table
+
+        Returns:
+            list[dict]: List of column information dictionaries
+        """
+        return self.connection.get_table_schema(table_name)
+
+    def count(self, table: str, where_clause: str = '', parameters: tuple[Any, ...] | None = None) -> int:
+        """
+        Count rows in a table.
+
+        Args:
+            table: Table name
+            where_clause: Optional WHERE clause (without WHERE keyword)
+            parameters: Query parameters for WHERE clause
+
+        Returns:
+            int: Number of rows
+        """
+        query = f'SELECT COUNT(*) as count FROM {table}'  # noqa: S608
+        if where_clause:
+            query += f' WHERE {where_clause}'
+
+        result = self.fetch_one(query, parameters)
+        return result['count'] if result else 0
+
+    def exists(self, table: str, where_clause: str, parameters: tuple[Any, ...]) -> bool:
+        """
+        Check if a record exists.
+
+        Args:
+            table: Table name
+            where_clause: WHERE clause (without WHERE keyword)
+            parameters: Query parameters
+
+        Returns:
+            bool: True if record exists, False otherwise
+        """
+        query = f'SELECT 1 FROM {table} WHERE {where_clause} LIMIT 1'  # noqa: S608
+        result = self.fetch_one(query, parameters)
+        return result is not None

amsdal/services/external_model_generator.py

@@ -0,0 +1,350 @@
+"""
+External Model Generator Service.
+
+Generates ExternalModel classes from external connection schemas.
+This allows runtime model generation from external databases without
+manual model definition.
+"""
+
+from typing import Any
+from typing import cast
+
+from amsdal_data.connections.external.base import SchemaIntrospectionProtocol
+from amsdal_models.classes.external_model import ExternalModel
+from amsdal_models.utils.schema_converter import ExternalSchemaConverter
+from amsdal_utils.schemas.schema import ObjectSchema
+
+from amsdal.services.external_connections import ExternalConnectionManager
+
+
+class ExternalModelGenerator:
+    """
+    Service for generating ExternalModel classes from external connections.
+
+    This service introspects external database schemas and generates
+    corresponding ExternalModel classes that can be used immediately
+    for querying the external data.
+
+    Features:
+    - Automatic schema introspection
+    - Type mapping (SQL types -> Python types)
+    - Primary key detection
+    - In-memory model class generation
+    - No lakehouse schema creation
+
+    Example usage:
+        # Generate models for a single table
+        generator = ExternalModelGenerator()
+        User = generator.generate_model('external_db', 'users')
+
+        # Now use the generated model
+        users = User.objects.filter(active=True).execute()
+
+        # Generate models for all tables
+        models = generator.generate_models_for_connection('external_db')
+        User = models['User']
+        Post = models['Post']
+    """
+
+    def __init__(self) -> None:
+        self._connection_manager = ExternalConnectionManager()
+        self._schema_converter = ExternalSchemaConverter()
+
+    def generate_model(
+        self,
+        connection_name: str,
+        table_name: str,
+        model_name: str | None = None,
+    ) -> type[ExternalModel]:
+        """
+        Generate an ExternalModel class for a specific table.
+
+        Args:
+            connection_name: Name of the external connection
+            table_name: Name of the table to generate model for
+            model_name: Optional custom model name (defaults to classified table name)
+
+        Returns:
+            type[ExternalModel]: Generated model class ready to use
+
+        Raises:
+            ValueError: If connection doesn't support schema introspection
+            ConnectionError: If connection is not available
+            RuntimeError: If model generation fails
+
+        Example:
+            generator = ExternalModelGenerator()
+            User = generator.generate_model('external_db', 'users')
+
+            # Query using the generated model
+            active_users = User.objects.filter(active=True).execute()
+        """
+        # Get the connection
+        connection = self._connection_manager.get_connection(connection_name)
+
+        # Check if connection supports schema introspection
+        if not isinstance(connection, SchemaIntrospectionProtocol):  # type: ignore[misc]
+            msg = (
+                f"Connection '{connection_name}' does not support schema introspection. "
+                f'Connection type: {type(connection).__name__}'
+            )
+            raise ValueError(msg)
+
+        # Get table schema
+        table_schema = connection.get_table_schema(table_name)
+
+        # Convert to ObjectSchema
+        # Detect connection type and use appropriate converter
+        object_schema = self._convert_schema(
+            connection=connection,
+            table_name=table_name,
+            table_schema=table_schema,
+            connection_name=connection_name,
+        )
+
+        # Generate model class from ObjectSchema
+        model_class = self._create_model_class(object_schema, model_name)
+
+        return model_class
+
+    def generate_models_for_connection(
+        self,
+        connection_name: str,
+        table_names: list[str] | None = None,
+    ) -> dict[str, type[ExternalModel]]:
+        """
+        Generate ExternalModel classes for all tables in a connection.
+
+        Args:
+            connection_name: Name of the external connection
+            table_names: Optional list of specific tables to generate models for.
+                        If None, generates models for all tables.
+
+        Returns:
+            dict[str, type[ExternalModel]]: Dictionary mapping model names to model classes
+
+        Raises:
+            ValueError: If connection doesn't support schema introspection
+            ConnectionError: If connection is not available
+
+        Example:
+            generator = ExternalModelGenerator()
+            models = generator.generate_models_for_connection('external_db')
+
+            # Access generated models
+            User = models['User']
+            Post = models['Post']
+            Comment = models['Comment']
+
+            # Or generate only specific tables
+            models = generator.generate_models_for_connection(
+                'external_db',
+                table_names=['users', 'posts']
+            )
+        """
+        # Get the connection
+        connection = self._connection_manager.get_connection(connection_name)
+
+        # Check if connection supports schema introspection
+        if not isinstance(connection, SchemaIntrospectionProtocol):  # type: ignore[misc]
+            msg = (
+                f"Connection '{connection_name}' does not support schema introspection. "
+                f'Connection type: {type(connection).__name__}'
+            )
+            raise ValueError(msg)
+
+        # Get list of tables
+        if table_names is None:
+            table_names = connection.get_table_names()
+
+        # Generate models for each table
+        models: dict[str, type[ExternalModel]] = {}
+        for table_name in table_names:
+            try:
+                model = self.generate_model(connection_name, table_name)
+                models[model.__name__] = model
+            except Exception as e:
+                # Log error but continue with other tables
+                print(f"Warning: Failed to generate model for table '{table_name}': {e}")
+                continue
+
+        return models
+
+    def _convert_schema(
+        self,
+        connection: Any,
+        table_name: str,
+        table_schema: list[dict[str, Any]],
+        connection_name: str,
+    ) -> ObjectSchema:
+        """
+        Convert raw table schema to ObjectSchema based on connection type.
+
+        Args:
+            connection: The connection object
+            table_name: Name of the table
+            table_schema: Raw schema data from connection
+            connection_name: Name of the connection
+
+        Returns:
+            ObjectSchema: Converted schema
+        """
+        # Detect connection type and use appropriate converter
+        connection_type = type(connection).__name__
+
+        if 'sqlite' in connection_type.lower():
+            return self._schema_converter.sqlite_schema_to_object_schema(
+                table_name=table_name,
+                columns=table_schema,
+                connection_name=connection_name,
+            )
+
+        # For other connection types, try to use generic converter
+        # First, try to detect the schema format
+        if table_schema and isinstance(table_schema[0], dict):
+            # Check if it's SQLite format (has 'cid', 'name', 'type', 'pk', etc.)
+            if all(key in table_schema[0] for key in ('cid', 'name', 'type')):
+                return self._schema_converter.sqlite_schema_to_object_schema(
+                    table_name=table_name,
+                    columns=table_schema,
+                    connection_name=connection_name,
+                )
+
+            # Check if it's PostgreSQL format (has 'column_name', 'data_type', etc.)
+            if 'column_name' in table_schema[0] and 'data_type' in table_schema[0]:
+                return self._schema_converter.postgres_schema_to_object_schema(
+                    table_name=table_name,
+                    columns=table_schema,
+                    connection_name=connection_name,
+                )
+
+            # Try generic converter with format normalization
+            normalized_columns = self._normalize_schema_format(table_schema)
+            return self._schema_converter.generic_schema_to_object_schema(
+                table_name=table_name,
+                columns=normalized_columns,
+                connection_name=connection_name,
+            )
+
+        msg = f'Unknown schema format for connection type: {connection_type}'
+        raise ValueError(msg)
+
+    def _normalize_schema_format(self, table_schema: list[dict[str, Any]]) -> list[dict[str, Any]]:
+        """
+        Normalize various schema formats to generic format.
+
+        Converts various schema formats to the format expected by generic_schema_to_object_schema:
+        {'name': str, 'type': str, 'nullable': bool, 'primary_key': bool, 'default': Any}
+        """
+        normalized = []
+
+        for column in table_schema:
+            # Try to extract name
+            name = column.get('name') or column.get('column_name') or column.get('field')
+
+            # Try to extract type
+            col_type = column.get('type') or column.get('data_type') or column.get('field_type') or 'TEXT'
+
+            # Try to extract nullable
+            nullable = True
+            if 'nullable' in column:
+                nullable = column['nullable']
+            elif 'is_nullable' in column:
+                nullable = column['is_nullable'] in (True, 'YES', 'yes', 1)
+            elif 'notnull' in column:
+                nullable = column['notnull'] in (False, 0)
+
+            # Try to extract primary key
+            pk = column.get('primary_key') or column.get('pk') or False
+            if isinstance(pk, int):
+                pk = pk > 0
+
+            # Try to extract default
+            default = column.get('default') or column.get('dflt_value') or column.get('column_default')
+
+            normalized.append(
+                {
+                    'name': name,
+                    'type': col_type,
+                    'nullable': nullable,
+                    'primary_key': pk,
+                    'default': default,
+                }
+            )
+
+        return normalized
+
+    def _create_model_class(
+        self,
+        object_schema: ObjectSchema,
+        custom_name: str | None = None,
+    ) -> type[ExternalModel]:
+        """
+        Create an ExternalModel class from ObjectSchema.
+
+        Args:
+            object_schema: The schema to create model from
+            custom_name: Optional custom model name
+
+        Returns:
+            type[ExternalModel]: Generated model class
+        """
+        # Extract model metadata from schema
+        model_name = custom_name or object_schema.title
+        table_name = cast(str, object_schema.__table_name__)  # type: ignore[attr-defined]
+        connection_name = cast(str, object_schema.__connection__)  # type: ignore[attr-defined]
+        pk_fields = getattr(object_schema, '__primary_key__', None)
+
+        # Build class attributes
+        class_attrs: dict[str, Any] = {
+            '__table_name__': table_name,
+            '__connection__': connection_name,
+            '__module__': __name__,
+        }
+
+        # Add primary key if present
+        if pk_fields:
+            # For composite keys, use list; for single key, use string
+            if len(pk_fields) == 1:
+                class_attrs['__primary_key__'] = pk_fields[0]
+            else:
+                class_attrs['__primary_key__'] = pk_fields
+
+        # Add field annotations from schema properties
+        annotations: dict[str, type] = {}
+        if object_schema.properties:
+            for field_name, field_def in object_schema.properties.items():
+                # Map CoreTypes to Python types for annotations
+                field_type = self._core_type_to_python_type(getattr(field_def, 'type', 'string'))
+                annotations[field_name] = field_type
+
+        class_attrs['__annotations__'] = annotations
+
+        # Create the model class dynamically
+        model_class = type(model_name, (ExternalModel,), class_attrs)
+
+        return cast(type[ExternalModel], model_class)
+
+    @staticmethod
+    def _core_type_to_python_type(core_type: str) -> type:
+        """
+        Convert CoreType string to Python type for annotations.
+
+        Args:
+            core_type: CoreType value (e.g., 'string', 'integer')
+
+        Returns:
+            type: Corresponding Python type
+        """
+        type_mapping = {
+            'string': str,
+            'integer': int,
+            'number': float,
+            'boolean': bool,
+            'date': str,  # Will be string representation
+            'datetime': str,  # Will be string representation
+            'binary': bytes,
+            'array': list,
+            'dictionary': dict,
+        }
+        return type_mapping.get(core_type, str)

{amsdal-0.5.19.dist-info → amsdal-0.5.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amsdal
-Version: 0.5.19
+Version: 0.5.21
 Summary: AMSDAL
 License: AMSDAL End User License Agreement