amsdal 0.0.41__cp310-cp310-win_amd64.whl → 0.1.1__cp310-cp310-win_amd64.whl

amsdal/contrib/frontend_configs/migrations/0000_initial.py

@@ -0,0 +1,255 @@
+from amsdal_utils.models.enums import SchemaTypes
+
+from amsdal.migration import migrations
+
+
+class Migration(migrations.Migration):
+    operations: list[migrations.Operation] = [
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendConfigSkipNoneBase',
+            new_schema={
+                'title': 'FrontendConfigSkipNoneBase',
+                'properties': {},
+                'meta_class': 'TypeMeta',
+                'custom_code': "from typing import Any\n\n\ndef model_dump(self, **kwargs: Any) -> dict[str, Any]:  # type: ignore[no-untyped-def]  # noqa: ARG001\n    kwargs['exclude_none'] = True\n\n    return super().model_dump(**kwargs)  # type: ignore[misc]\n\n\ndef model_dump_json(self, **kwargs: Any) -> str:  # type: ignore[no-untyped-def]  # noqa: ARG001\n    kwargs['exclude_none'] = True\n\n    return super().model_dump_json(**kwargs)  # type: ignore[misc]",
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendConfigGroupValidator',
+            new_schema={
+                'title': 'FrontendConfigGroupValidator',
+                'type': 'FrontendConfigSkipNoneBase',
+                'properties': {
+                    'mainControl': {'type': 'string', 'title': 'Main Control'},
+                    'dependentControls': {'type': 'array', 'items': {'type': 'string'}, 'title': 'Dependent Controls'},
+                    'condition': {
+                        'type': 'string',
+                        'title': 'Condition',
+                        'options': [
+                            {'key': 'Less Than', 'value': 'lt'},
+                            {'key': 'Less Than or Equal', 'value': 'lte'},
+                            {'key': 'Greater Than', 'value': 'gt'},
+                            {'key': 'Greater Than or Equal', 'value': 'gte'},
+                            {'key': 'Equal', 'value': 'eq'},
+                            {'key': 'Not Equal', 'value': 'neq'},
+                            {'key': 'Exist', 'value': 'exist'},
+                        ],
+                    },
+                },
+                'meta_class': 'TypeMeta',
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendActivatorConfig',
+            new_schema={
+                'title': 'FrontendActivatorConfig',
+                'type': 'FrontendConfigGroupValidator',
+                'properties': {'value': {'type': 'anything', 'title': 'Value'}},
+                'meta_class': 'TypeMeta',
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendConfigTextMask',
+            new_schema={
+                'title': 'FrontendConfigTextMask',
+                'type': 'FrontendConfigSkipNoneBase',
+                'required': ['mask_string'],
+                'properties': {
+                    'mask_string': {'type': 'string', 'title': 'Mask String'},
+                    'prefix': {'type': 'string', 'title': 'Prefix'},
+                    'suffix': {'type': 'string', 'title': 'Suffix'},
+                    'thousands_separator': {'type': 'string', 'title': 'Thousands Separator'},
+                },
+                'meta_class': 'TypeMeta',
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendConfigSliderOption',
+            new_schema={
+                'title': 'FrontendConfigSliderOption',
+                'type': 'FrontendConfigSkipNoneBase',
+                'properties': {
+                    'min': {'type': 'number', 'title': 'Minimum'},
+                    'max': {'type': 'number', 'title': 'Maximum'},
+                    'range': {'type': 'boolean', 'title': 'Range'},
+                },
+                'meta_class': 'TypeMeta',
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendConfigValidator',
+            new_schema={
+                'title': 'FrontendConfigValidator',
+                'type': 'FrontendConfigGroupValidator',
+                'properties': {
+                    'function': {
+                        'type': 'string',
+                        'title': 'Function',
+                        'options': [
+                            {'key': 'Minimum', 'value': 'min'},
+                            {'key': 'Maximum', 'value': 'max'},
+                            {'key': 'Minimum Length', 'value': 'minLength'},
+                            {'key': 'Maximum Length', 'value': 'maxLength'},
+                            {'key': 'Required', 'value': 'required'},
+                            {'key': 'Pattern', 'value': 'pattern'},
+                        ],
+                    },
+                    'value': {'type': 'string', 'title': 'Value'},
+                },
+                'meta_class': 'TypeMeta',
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendConfigAsyncValidator',
+            new_schema={
+                'title': 'FrontendConfigAsyncValidator',
+                'type': 'FrontendConfigSkipNoneBase',
+                'properties': {'endpoint': {'type': 'string', 'title': 'Endpoint'}},
+                'meta_class': 'TypeMeta',
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendConfigControlAction',
+            new_schema={
+                'title': 'FrontendConfigControlAction',
+                'type': 'FrontendConfigSkipNoneBase',
+                'required': ['action', 'text', 'type'],
+                'properties': {
+                    'action': {'type': 'string', 'title': 'Action'},
+                    'text': {'type': 'string', 'title': 'Text'},
+                    'type': {
+                        'type': 'string',
+                        'title': 'Type',
+                        'options': [
+                            {'key': 'Arrow Next', 'value': 'arrow-next'},
+                            {'key': 'Arrow Prev', 'value': 'arrow-prev'},
+                            {'key': 'Action Button', 'value': 'action-button'},
+                            {'key': 'Text Next', 'value': 'text-next'},
+                            {'key': 'Text Prev', 'value': 'text-prev'},
+                        ],
+                    },
+                    'dataLayerEvent': {'type': 'string', 'title': 'Data Layer Event'},
+                    'activator': {'type': 'string', 'title': 'Activator'},
+                    'icon': {'type': 'string', 'title': 'Icon'},
+                },
+                'meta_class': 'TypeMeta',
+                'custom_code': "from pydantic import field_validator\n\n\n@field_validator('action', mode='after')  # type: ignore[misc]\n@classmethod\ndef validate_action(cls, v: str) -> str:  # type: ignore[no-untyped-def] # noqa: ARG001\n    if not v.startswith('navigate::') and v not in [\n        'goPrev',\n        'goNext',\n        'goNextWithSubmit',\n        'submit',\n        'submitWithDataLayer',\n    ]:\n        msg = 'Action must be one of: goPrev, goNext, goNextWithSubmit, submit, submitWithDataLayer, navigate::{string}'\n        raise ValueError(msg)\n\n    return v",
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendConfigOption',
+            new_schema={
+                'title': 'FrontendConfigOption',
+                'type': 'FrontendConfigSkipNoneBase',
+                'properties': {
+                    'label': {'type': 'string', 'title': 'Label'},
+                    'value': {'type': 'string', 'title': 'Value'},
+                },
+                'meta_class': 'TypeMeta',
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendControlConfig',
+            new_schema={
+                'title': 'FrontendControlConfig',
+                'type': 'FrontendConfigSkipNoneBase',
+                'required': ['type', 'name'],
+                'properties': {
+                    'type': {
+                        'type': 'string',
+                        'title': 'Type',
+                        'options': [
+                            {'key': 'Infoscreen', 'value': 'infoscreen'},
+                            {'key': 'Info Group', 'value': 'info-group'},
+                            {'key': 'Text', 'value': 'text'},
+                            {'key': 'Number', 'value': 'number'},
+                            {'key': 'Select', 'value': 'select'},
+                            {'key': 'Object Group', 'value': 'object_group'},
+                            {'key': 'Group', 'value': 'group'},
+                            {'key': 'Group Toggle', 'value': 'group_toggle'},
+                            {'key': 'Group Switch', 'value': 'group_switch'},
+                            {'key': 'Password', 'value': 'password'},
+                            {'key': 'Email', 'value': 'email'},
+                            {'key': 'Phone', 'value': 'phone'},
+                            {'key': 'textarea', 'value': 'textarea'},
+                            {'key': 'Multiselect', 'value': 'multiselect'},
+                            {'key': 'number_initial', 'value': 'number_initial'},
+                            {'key': 'number_plus', 'value': 'number_plus'},
+                            {'key': 'number_minus', 'value': 'number_minus'},
+                            {'key': 'number_equals', 'value': 'number_equals'},
+                            {'key': 'toggle', 'value': 'toggle'},
+                            {'key': 'radio', 'value': 'radio'},
+                            {'key': 'checkbox', 'value': 'checkbox'},
+                            {'key': 'number-slider', 'value': 'number-slider'},
+                            {'key': 'number-operations', 'value': 'number-operations'},
+                            {'key': 'date', 'value': 'date'},
+                            {'key': 'dateTriplet', 'value': 'dateTriplet'},
+                            {'key': 'array', 'value': 'array'},
+                            {'key': 'object', 'value': 'object'},
+                            {'key': 'file', 'value': 'file'},
+                            {'key': 'dropzone', 'value': 'dropzone'},
+                            {'key': 'object_latest', 'value': 'object_latest'},
+                            {'key': 'Dictionary', 'value': 'dict'},
+                            {'key': 'Time', 'value': 'time'},
+                            {'key': 'Datetime', 'value': 'datetime'},
+                            {'key': 'Bytes', 'value': 'Bytes'},
+                        ],
+                    },
+                    'name': {'type': 'string', 'title': 'Name'},
+                    'label': {'type': 'string', 'title': 'Label'},
+                    'required': {'type': 'boolean', 'title': 'Required'},
+                    'hideLabel': {'type': 'boolean', 'title': 'Hide Label'},
+                    'actions': {'type': 'array', 'items': {'type': 'FrontendConfigControlAction'}, 'title': 'Actions'},
+                    'validators': {
+                        'type': 'array',
+                        'items': {'type': 'FrontendConfigValidator'},
+                        'title': 'Validators',
+                    },
+                    'asyncValidators': {
+                        'type': 'array',
+                        'items': {'type': 'FrontendConfigAsyncValidator'},
+                        'title': 'Async Validators',
+                    },
+                    'activators': {
+                        'type': 'array',
+                        'items': {'type': 'FrontendActivatorConfig'},
+                        'title': 'Activators',
+                    },
+                    'additionalText': {'type': 'string', 'title': 'Additional Text'},
+                    'value': {'type': 'anything', 'title': 'Value'},
+                    'placeholder': {'type': 'string', 'title': 'Placeholder'},
+                    'options': {'type': 'array', 'items': {'type': 'FrontendConfigOption'}, 'title': 'Options'},
+                    'mask': {'type': 'FrontendConfigTextMask', 'title': 'Mask'},
+                    'controls': {'type': 'array', 'items': {'type': 'FrontendControlConfig'}, 'title': 'Controls'},
+                    'showSearch': {'type': 'boolean', 'title': 'Show Search'},
+                    'sliderOptions': {'type': 'FrontendConfigSliderOption', 'title': 'Slider Option'},
+                    'customLabel': {'type': 'array', 'items': {'type': 'string'}, 'title': 'Custom Label'},
+                    'control': {'type': 'FrontendControlConfig', 'title': 'Control'},
+                    'entityType': {'type': 'string', 'title': 'Entity Type'},
+                },
+                'meta_class': 'TypeMeta',
+            },
+        ),
+        migrations.CreateClass(
+            schema_type=SchemaTypes.CONTRIB,
+            class_name='FrontendModelConfig',
+            new_schema={
+                'title': 'FrontendModelConfig',
+                'required': ['class_name'],
+                'properties': {
+                    'class_name': {'type': 'string', 'title': 'Class Name'},
+                    'control': {'type': 'FrontendControlConfig', 'title': 'Control'},
+                },
+            },
+        ),
+    ]

amsdal/manager.pyi

@@ -5,7 +5,6 @@ from amsdal.cloud.services.auth.signup_service import SignupService as SignupSer
 from amsdal.configs.main import settings as settings
 from amsdal.errors import AmsdalAuthenticationError as AmsdalAuthenticationError, AmsdalMissingCredentialsError as AmsdalMissingCredentialsError, AmsdalRuntimeError as AmsdalRuntimeError, AmsdalSignupError as AmsdalSignupError
 from amsdal.fixtures.manager import FixturesManager as FixturesManager
-from amsdal.migration.manager import MigrationManager as MigrationManager
 from amsdal.mixins.build_mixin import BuildMixin as BuildMixin
 from amsdal.mixins.class_versions_mixin import ClassVersionsMixin as ClassVersionsMixin
 from amsdal.operations.manager import OperationsManager as OperationsManager
@@ -31,6 +30,8 @@ class AmsdalManager(BuildMixin, ClassVersionsMixin, metaclass=Singleton):
         Initializes all sub managers. Reads the configuration.
         """
     @property
+    def is_setup(self) -> bool: ...
+    @property
     def is_authenticated(self) -> bool:
         """
         Indicates if you have passed the AMSDAL license authentication process.
@@ -43,7 +44,7 @@ class AmsdalManager(BuildMixin, ClassVersionsMixin, metaclass=Singleton):
         """
         Initiates models root path and the connections
         """
-    def build(self, source_models_path: Path, source_transactions_path: Path, source_static_files_path: Path, source_fixtures_path: Path) -> None:
+    def build(self, source_models_path: Path, source_transactions_path: Path, source_static_files_path: Path, source_fixtures_path: Path, source_migrations_path: Path) -> None:
         """
         Build method
 
@@ -51,6 +52,7 @@ class AmsdalManager(BuildMixin, ClassVersionsMixin, metaclass=Singleton):
         :param source_transactions_path: Path to the directory where the source transactions are located.
         :param source_static_files_path: Path to the directory where the source static files are located.
         :param source_fixtures_path: Path to the directory where the source fixtures are located.
+        :param source_migrations_path: Path to the directory where the source migrations are located.
         :return: None
 
         This method is used to build the necessary components for the Amsdal framework.
@@ -67,9 +69,13 @@ class AmsdalManager(BuildMixin, ClassVersionsMixin, metaclass=Singleton):
         and `ClassVersionsMixin`. It is intended to be used in the Amsdal framework for managing
         and building components.
         """
+    def post_setup(self) -> None:
+        """
+        Registers internal classes and prepares connections (creates internal tables).
+        """
     def migrate(self) -> None:
         """
-        Check changes in the models and apply them to the database.
+        DEPRECATED: Check changes in the models and apply them to the database.
         """
     def _check_auth(self) -> None: ...
     @property

amsdal/migration/base_migration_schemas.pyi

@@ -0,0 +1,44 @@
+import abc
+from _typeshed import Incomplete
+from abc import ABC, abstractmethod
+from amsdal.configs.constants import BASE_DIR as BASE_DIR
+from amsdal_models.classes.model import Model
+from amsdal_models.schemas.data_models.core import DictSchema, LegacyDictSchema, TypeData
+from amsdal_models.schemas.data_models.schema import ObjectSchema, PropertyData
+from amsdal_utils.models.enums import SchemaTypes as SchemaTypes
+from pathlib import Path
+
+class BaseMigrationSchemas(ABC, metaclass=abc.ABCMeta):
+    _classes: Incomplete
+    _classes_versions: Incomplete
+    _buffered_classes: Incomplete
+    def __init__(self) -> None: ...
+    def get_model(self, name: str) -> type[Model]: ...
+    @abstractmethod
+    def register_model(self, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    @abstractmethod
+    def unregister_model(self, class_name: str) -> None: ...
+    @abstractmethod
+    def compile_buffered_classes(self) -> None: ...
+    @staticmethod
+    def register_model_version(class_name: str, class_version: str) -> None: ...
+
+class DefaultMigrationSchemas(BaseMigrationSchemas):
+    model_class_template_layout: Path
+    model_class_template: Path
+    dict_validator_template: Path
+    options_validator_template: Path
+    def register_model(self, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    def compile_buffered_classes(self) -> None: ...
+    def unregister_model(self, class_name: str) -> None: ...
+    def _compile_buffered_classes(self) -> list[tuple[str, type[Model]]]: ...
+    def _build_class_source(self, class_name: str, schema: ObjectSchema, schema_type: SchemaTypes) -> tuple[str, str]: ...
+    @staticmethod
+    def _resolve_class_inheritance(schema: ObjectSchema) -> str: ...
+    @staticmethod
+    def _process_custom_code(custom_code: str | None) -> tuple[str, str]: ...
+    def _render_property(self, name: str, property_schema: PropertyData, required: list[str]) -> str: ...
+    def _render_type_annotation(self, type_: str, items: TypeData | DictSchema | LegacyDictSchema | None) -> str: ...
+    def _render_validators(self, schema: ObjectSchema) -> list[str]: ...
+    def _get_all_types(self, type_: str, items: TypeData | DictSchema | LegacyDictSchema | None) -> set[str]: ...
+    def _is_reference(self, type_: str) -> bool: ...

amsdal/migration/data_classes.pyi

@@ -0,0 +1,78 @@
+from amsdal_models.classes.model import Model
+from amsdal_models.schemas.data_models.schema import ObjectSchema, PropertyData
+from amsdal_utils.models.data_models.address import Address as Address
+from amsdal_utils.models.enums import SchemaTypes as SchemaTypes
+from dataclasses import dataclass
+from enum import Enum
+from pathlib import Path
+
+class Action(str, Enum):
+    CREATED: str
+    UPDATED: str
+    NO_ACTION: str
+
+@dataclass
+class ClassSaveResult:
+    action: Action
+    instance: Model
+    def __init__(self, action, instance) -> None: ...
+
+@dataclass
+class ClassUpdateResult:
+    is_updated: bool
+    class_instance: Model
+    def __init__(self, is_updated, class_instance) -> None: ...
+
+@dataclass
+class MigrateResult:
+    class_instance: Model
+    is_table_created: bool
+    is_data_migrated: bool
+    def __init__(self, class_instance, is_table_created, is_data_migrated) -> None: ...
+
+class ModuleTypes(str, Enum):
+    APP: str
+    CORE: str
+    CONTRIB: str
+
+@dataclass
+class MigrationFile:
+    path: Path
+    type: ModuleTypes
+    number: int
+    module: str | None = ...
+    applied_at: float | None = ...
+    stored_address: Address | None = ...
+    @property
+    def is_initial(self) -> bool: ...
+    def __init__(self, path, type, number, module, applied_at, stored_address) -> None: ...
+
+@dataclass
+class ClassSchema:
+    object_schema: ObjectSchema
+    type: ModuleTypes
+    def __init__(self, object_schema, type) -> None: ...
+
+class OperationTypes(str, Enum):
+    CREATE_CLASS: str
+    UPDATE_CLASS: str
+    DELETE_CLASS: str
+
+@dataclass
+class MigrateOperation:
+    type: OperationTypes
+    class_name: str
+    schema_type: SchemaTypes
+    old_schema: ObjectSchema | PropertyData | None = ...
+    new_schema: ObjectSchema | PropertyData | None = ...
+    def __init__(self, type, class_name, schema_type, old_schema, new_schema) -> None: ...
+
+class MigrationDirection(str, Enum):
+    FORWARD: str
+    BACKWARD: str
+
+@dataclass
+class MigrationResult:
+    direction: MigrationDirection
+    migration: MigrationFile
+    def __init__(self, direction, migration) -> None: ...

amsdal/migration/executors/base.pyi

@@ -0,0 +1,21 @@
+import abc
+from _typeshed import Incomplete
+from abc import ABC, abstractmethod
+from amsdal.migration.base_migration_schemas import BaseMigrationSchemas as BaseMigrationSchemas
+from amsdal_models.schemas.data_models.schema import ObjectSchema
+from amsdal_utils.models.enums import SchemaTypes as SchemaTypes
+
+class BaseMigrationExecutor(ABC, metaclass=abc.ABCMeta):
+    schemas: BaseMigrationSchemas
+    _buffer: Incomplete
+    def __init__(self) -> None: ...
+    @abstractmethod
+    def create_class(self, schemas: BaseMigrationSchemas, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    @abstractmethod
+    def update_class(self, schemas: BaseMigrationSchemas, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    @abstractmethod
+    def delete_class(self, schemas: BaseMigrationSchemas, class_name: str, schema_type: SchemaTypes) -> None: ...
+    @staticmethod
+    def _resolve_base_class_name(class_name: str, meta_class: str) -> str: ...
+    def buffer_class_migration(self, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    def flush_buffer(self) -> None: ...

amsdal/migration/executors/default_executor.pyi

@@ -0,0 +1,20 @@
+from _typeshed import Incomplete
+from amsdal.migration.base_migration_schemas import BaseMigrationSchemas as BaseMigrationSchemas
+from amsdal.migration.data_classes import Action as Action, ClassSaveResult as ClassSaveResult
+from amsdal.migration.executors.base import BaseMigrationExecutor as BaseMigrationExecutor
+from amsdal.migration.utils import object_schema_to_table_schema as object_schema_to_table_schema
+from amsdal_models.classes.model import Model
+from amsdal_models.schemas.data_models.schema import ObjectSchema
+from amsdal_utils.models.enums import SchemaTypes
+
+class DefaultMigrationExecutor(BaseMigrationExecutor):
+    schemas: Incomplete
+    _table_schemas_manager: Incomplete
+    def __init__(self, schemas: BaseMigrationSchemas) -> None: ...
+    def create_class(self, schemas: BaseMigrationSchemas, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    def update_class(self, schemas: BaseMigrationSchemas, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    def delete_class(self, schemas: BaseMigrationSchemas, class_name: str, schema_type: SchemaTypes) -> None: ...
+    def flush_buffer(self) -> None: ...
+    def _save_class(self, base_class: type[Model], object_schema: ObjectSchema, schema_type: SchemaTypes) -> ClassSaveResult: ...
+    def _create_table(self, object_schema: ObjectSchema, class_version: str) -> None: ...
+    def _migrate_historical_data(self, schemas: BaseMigrationSchemas, class_name: str, prior_version: str, new_version: str) -> None: ...

amsdal/migration/executors/state_executor.pyi

@@ -0,0 +1,14 @@
+from _typeshed import Incomplete
+from amsdal.migration.base_migration_schemas import BaseMigrationSchemas as BaseMigrationSchemas
+from amsdal.migration.executors.base import BaseMigrationExecutor as BaseMigrationExecutor
+from amsdal_models.schemas.data_models.schema import ObjectSchema
+from amsdal_utils.models.enums import SchemaTypes
+
+class StateMigrationExecutor(BaseMigrationExecutor):
+    schemas: Incomplete
+    do_fetch_latest_version: Incomplete
+    def __init__(self, schemas: BaseMigrationSchemas, *, do_fetch_latest_version: bool = True) -> None: ...
+    def create_class(self, schemas: BaseMigrationSchemas, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    def update_class(self, schemas: BaseMigrationSchemas, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    def delete_class(self, schemas: BaseMigrationSchemas, class_name: str, schema_type: SchemaTypes) -> None: ...
+    def flush_buffer(self) -> None: ...

amsdal/migration/file_migration_executor.pyi

@@ -0,0 +1,35 @@
+from _typeshed import Incomplete
+from amsdal.configs.constants import CORE_MIGRATIONS_PATH as CORE_MIGRATIONS_PATH
+from amsdal.configs.main import settings as settings
+from amsdal.migration.data_classes import MigrationDirection as MigrationDirection, MigrationFile as MigrationFile, MigrationResult as MigrationResult, ModuleTypes as ModuleTypes
+from amsdal.migration.executors.base import BaseMigrationExecutor as BaseMigrationExecutor
+from amsdal.migration.executors.state_executor import StateMigrationExecutor as StateMigrationExecutor
+from amsdal.migration.file_migration_store import BaseMigrationStore as BaseMigrationStore, FileMigrationStore as FileMigrationStore
+from amsdal.migration.migrations import MigrateData as MigrateData, Migration as Migration
+from amsdal.migration.migrations_loader import MigrationsLoader as MigrationsLoader
+from amsdal.migration.utils import contrib_to_module_root_path as contrib_to_module_root_path
+from amsdal_utils.models.data_models.address import Address
+from amsdal_utils.models.enums import SchemaTypes
+
+logger: Incomplete
+
+class FileMigrationExecutorManager:
+    migration_address: Address
+    core_loader: Incomplete
+    contrib_loaders: Incomplete
+    app_loader: Incomplete
+    executor: Incomplete
+    _applied_migration_files: Incomplete
+    store: Incomplete
+    def __init__(self, app_migrations_loader: MigrationsLoader, executor: BaseMigrationExecutor, store: BaseMigrationStore | None = None) -> None: ...
+    def execute(self, migration_number: int | None = None, module_type: ModuleTypes | None = None, *, fake: bool = False, skip_data_migrations: bool = False) -> list[MigrationResult]: ...
+    @staticmethod
+    def _get_contrib_loaders() -> list[MigrationsLoader]: ...
+    def _apply(self, migration_number: int | None = None, module_type: ModuleTypes | None = None, *, fake: bool = False, skip_data_migrations: bool = False) -> list[MigrationResult]: ...
+    def _apply_migrations(self, loader: MigrationsLoader, module_type: ModuleTypes, migration_number: int | None = None, *, fake: bool = False, skip_data_migrations: bool = False) -> list[MigrationResult]: ...
+    def _init_state_from_applied_migrations(self, migrations: list[MigrationFile], module_type: ModuleTypes) -> None: ...
+    @staticmethod
+    def get_migration_class(migration: MigrationFile) -> type['Migration']: ...
+    def _is_migration_applied(self, migration: MigrationFile, module_type: ModuleTypes) -> bool: ...
+    @staticmethod
+    def _map_module_type_to_schema_type(module_type: ModuleTypes) -> SchemaTypes: ...

amsdal/migration/file_migration_generator.pyi

@@ -0,0 +1,49 @@
+from _typeshed import Incomplete
+from amsdal.configs.main import settings as settings
+from amsdal.migration.base_migration_schemas import BaseMigrationSchemas as BaseMigrationSchemas
+from amsdal.migration.data_classes import MigrateOperation as MigrateOperation, MigrationFile as MigrationFile, ModuleTypes as ModuleTypes, OperationTypes as OperationTypes
+from amsdal.migration.executors.state_executor import StateMigrationExecutor as StateMigrationExecutor
+from amsdal.migration.file_migration_executor import FileMigrationExecutorManager as FileMigrationExecutorManager
+from amsdal.migration.file_migration_store import BaseMigrationStore as BaseMigrationStore
+from amsdal.migration.file_migration_writer import FileMigrationWriter as FileMigrationWriter
+from amsdal.migration.migrations_loader import MigrationsLoader as MigrationsLoader
+from amsdal.migration.schemas_loaders import BaseClassSchemaLoader as BaseClassSchemaLoader
+from amsdal_models.schemas.data_models.schema import ObjectSchema
+from amsdal_utils.models.enums import SchemaTypes
+from collections.abc import Callable as Callable
+from pathlib import Path
+from typing import ClassVar
+
+class StateMigrationStore(BaseMigrationStore):
+    def save_migration(self, migration: MigrationFile) -> None: ...
+    def delete_migration(self, migration: MigrationFile) -> None: ...
+    def fetch_migrations(self) -> list[MigrationFile]: ...
+
+class StateMigrationSchemas(BaseMigrationSchemas):
+    state: Incomplete
+    def __init__(self) -> None: ...
+    def register_model(self, class_name: str, object_schema: ObjectSchema, schema_type: SchemaTypes) -> None: ...
+    def unregister_model(self, class_name: str) -> None: ...
+    def compile_buffered_classes(self) -> None: ...
+
+class FileMigrationGenerator:
+    _operations: ClassVar[dict[OperationTypes, Callable[..., MigrateOperation]]]
+    _app_migrations_path: Incomplete
+    _app_migrations_loader: Incomplete
+    _schema_loader: Incomplete
+    _state: Incomplete
+    def __init__(self, schema_loader: BaseClassSchemaLoader, app_migrations_path: Path | None = None) -> None: ...
+    def init_migrations(self, schema_type: SchemaTypes = ...) -> None:
+        """Reads class schemas and creates initial migration file"""
+    def make_migrations(self, name: str | None = None, *, is_data: bool = False, schema_type: SchemaTypes = ...) -> MigrationFile: ...
+    def generate_operations(self, schema_type: SchemaTypes) -> list[MigrateOperation]: ...
+    def _make_data_migrations(self, name: str | None = None) -> MigrationFile: ...
+    def _init_state(self) -> None: ...
+    @classmethod
+    def build_operations(cls, schema_type: SchemaTypes, class_schema: ObjectSchema, old_class_schema: ObjectSchema | None) -> list[MigrateOperation]: ...
+    def write_migration_file(self, operations: list[MigrateOperation], name: str | None = None) -> MigrationFile: ...
+    def write_data_migration_file(self, name: str | None = None) -> MigrationFile: ...
+    @staticmethod
+    def generate_name_from_operations(operations: list[MigrateOperation]) -> str: ...
+    @staticmethod
+    def _get_migration_file_name(number: int, name: str) -> str: ...

amsdal/migration/file_migration_store.pyi

@@ -0,0 +1,33 @@
+import abc
+from _typeshed import Incomplete
+from abc import ABC, abstractmethod
+from amsdal.configs.main import settings as settings
+from amsdal.migration.data_classes import MigrationFile as MigrationFile, ModuleTypes as ModuleTypes
+from amsdal.migration.utils import contrib_to_module_root_path as contrib_to_module_root_path, map_module_type_to_schema_type as map_module_type_to_schema_type
+from amsdal_utils.config.data_models.amsdal_config import AmsdalConfig as AmsdalConfig
+from amsdal_utils.models.data_models.address import Address
+from typing import Any
+
+class BaseMigrationStore(ABC, metaclass=abc.ABCMeta):
+    @abstractmethod
+    def fetch_migrations(self) -> list[MigrationFile]: ...
+    @abstractmethod
+    def save_migration(self, migration: MigrationFile) -> None: ...
+    @abstractmethod
+    def delete_migration(self, migration: MigrationFile) -> None: ...
+
+class FileMigrationStore(BaseMigrationStore):
+    migration_address: Address
+    _config_manager: Incomplete
+    _amsdal_config: Incomplete
+    _connections_manager: Incomplete
+    def __init__(self) -> None: ...
+    def fetch_migrations(self) -> list[MigrationFile]: ...
+    def save_migration(self, migration: MigrationFile) -> None: ...
+    def delete_migration(self, migration: MigrationFile) -> None: ...
+    def _save_historical_data(self, address: Address, data: dict[str, Any], metadata: dict[str, Any]) -> None: ...
+    @staticmethod
+    def _build_migration_data(migration: MigrationFile) -> dict[str, Any]: ...
+    @classmethod
+    def _build_migration_metadata(cls, migration: MigrationFile, object_id: str | None = None) -> tuple[Address, dict[str, Any]]: ...
+    def _init_migration_table(self) -> None: ...