amsdal 0.0.40__cp311-cp311-macosx_10_9_universal2.whl → 0.1.0__cp311-cp311-macosx_10_9_universal2.whl

amsdal/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2023-present
 #
 # SPDX-License-Identifier: AMSDAL End User License Agreement
-__version__ = '0.0.40'
+__version__ = '0.1.0'

amsdal/configs/constants.py

@@ -6,6 +6,8 @@ BASE_DIR = Path(__file__).resolve().parent.parent
 TYPE_SCHEMAS_PATH: Path = BASE_DIR / 'schemas' / 'types'
 CORE_SCHEMAS_PATH: Path = BASE_DIR / 'schemas' / 'core'
 
+CORE_MIGRATIONS_PATH: Path = BASE_DIR / '__migrations__'
+
 # Environment
 TESTING_ENVIRONMENT = 'testing'
 DEVELOPMENT_ENVIRONMENT = 'development'

amsdal/configs/constants.pyi

@@ -4,6 +4,7 @@ from pathlib import Path
 BASE_DIR: Incomplete
 TYPE_SCHEMAS_PATH: Path
 CORE_SCHEMAS_PATH: Path
+CORE_MIGRATIONS_PATH: Path
 TESTING_ENVIRONMENT: str
 DEVELOPMENT_ENVIRONMENT: str
 PRODUCTION_ENVIRONMENT: str

amsdal/configs/main.py

@@ -35,6 +35,8 @@ class Settings(BaseSettings):
     """The static module name. The static files will be placed in this module."""
     TRANSACTIONS_MODULE_NAME: str = 'transactions'
     """The transactions module name. The transactions will be placed in this module."""
+    MIGRATIONS_DIRECTORY_NAME: str = 'migrations'
+    """The migrations directory name. The migration files will be placed in this folder."""
 
     ACCESS_KEY_ID: str | None = None
     """The access key that you will get during registering process."""
@@ -85,6 +87,10 @@ class Settings(BaseSettings):
     def transactions_root_path(self) -> Path:
         return self.models_root_path / self.TRANSACTIONS_MODULE_NAME
 
+    @property
+    def migrations_root_path(self) -> Path:
+        return self.models_root_path / self.MIGRATIONS_DIRECTORY_NAME
+
     @model_validator(mode='after')
     def check_passwords_match(self) -> 'Settings':
         config_path = self.CONFIG_PATH

amsdal/configs/main.pyi

@@ -12,6 +12,7 @@ class Settings(BaseSettings):
     FIXTURES_MODULE_NAME: str
     STATIC_MODULE_NAME: str
     TRANSACTIONS_MODULE_NAME: str
+    MIGRATIONS_DIRECTORY_NAME: str
     ACCESS_KEY_ID: str | None
     SECRET_ACCESS_KEY: str | None
     ACCESS_TOKEN: str | None
@@ -28,6 +29,8 @@ class Settings(BaseSettings):
     def static_root_path(self) -> Path: ...
     @property
     def transactions_root_path(self) -> Path: ...
+    @property
+    def migrations_root_path(self) -> Path: ...
     def check_passwords_match(self) -> Settings: ...
 
 base: TypeAlias

amsdal/contrib/frontend_configs/conversion/convert.py

@@ -9,6 +9,7 @@ from typing import ForwardRef
 from typing import Union
 
 from amsdal_models.classes.model import LegacyModel
+from amsdal_models.classes.model import Model
 from amsdal_utils.models.data_models.reference import Reference
 from pydantic import BaseModel
 from pydantic_core import PydanticUndefined
@@ -24,21 +25,21 @@ default_types_map = {
 }
 
 
-def _process_union(value: UnionType) -> dict[str, Any]:
+def _process_union(value: UnionType, *, is_transaction: bool = False) -> dict[str, Any]:
     arg_type = {'required': True}
     for arg in value.__args__:
         if arg == type(None):
             arg_type['required'] = False
             continue
 
-        control = convert_to_frontend_config(arg)
+        control = convert_to_frontend_config(arg, is_transaction=is_transaction)
         if control:
             arg_type.update(control)
 
     return arg_type
 
 
-def convert_to_frontend_config(value: Any) -> dict[str, Any]:
+def convert_to_frontend_config(value: Any, *, is_transaction: bool = False) -> dict[str, Any]:
     if hasattr(value, '__origin__'):
         origin_class = value.__origin__
 
@@ -46,7 +47,7 @@ def convert_to_frontend_config(value: Any) -> dict[str, Any]:
             return {}
 
         if origin_class is Union:
-            return _process_union(value)
+            return _process_union(value, is_transaction=is_transaction)
 
         if origin_class is list:
             return {
@@ -56,7 +57,7 @@ def convert_to_frontend_config(value: Any) -> dict[str, Any]:
                 'control': {
                     'name': 'array_items_values',
                     'label': 'array_items_values',
-                    **convert_to_frontend_config(value.__args__[0]),
+                    **convert_to_frontend_config(value.__args__[0], is_transaction=is_transaction),
                 },
             }
 
@@ -68,7 +69,7 @@ def convert_to_frontend_config(value: Any) -> dict[str, Any]:
                 'control': {
                     'name': 'dict_items_values',
                     'label': 'dict_items_values',
-                    **convert_to_frontend_config(value.__args__[1]),
+                    **convert_to_frontend_config(value.__args__[1], is_transaction=is_transaction),
                 },
             }
 
@@ -78,7 +79,7 @@ def convert_to_frontend_config(value: Any) -> dict[str, Any]:
         }
 
     if isinstance(value, UnionType):
-        return _process_union(value)
+        return _process_union(value, is_transaction=is_transaction)
 
     if value in default_types_map:
         return {
@@ -102,7 +103,10 @@ def convert_to_frontend_config(value: Any) -> dict[str, Any]:
         for arg_name in _parameters:
             if arg_name in value.__annotations__:
                 arg_type = value.__annotations__[arg_name]
-                control = convert_to_frontend_config(arg_type)
+                control = convert_to_frontend_config(
+                    arg_type,
+                    is_transaction=True,
+                )
 
                 if not control:
                     continue
@@ -136,6 +140,12 @@ def convert_to_frontend_config(value: Any) -> dict[str, Any]:
             'type': 'object_latest',
         }
 
+    if is_transaction and issubclass(value, Model):
+        return {
+            'type': 'object_latest',
+            'entityType': value.__name__,
+        }
+
     if issubclass(value, LegacyModel):
         return {}
 
@@ -148,7 +158,7 @@ def convert_to_frontend_config(value: Any) -> dict[str, Any]:
             schema = None
 
         for field_name, field in value.__annotations__.items():
-            control = convert_to_frontend_config(field)
+            control = convert_to_frontend_config(field, is_transaction=is_transaction)
 
             if not control:
                 continue

amsdal/contrib/frontend_configs/conversion/convert.pyi

@@ -5,5 +5,5 @@ from typing import Any
 
 default_types_map: Incomplete
 
-def _process_union(value: UnionType) -> dict[str, Any]: ...
-def convert_to_frontend_config(value: Any) -> dict[str, Any]: ...
+def _process_union(value: UnionType, *, is_transaction: bool = False) -> dict[str, Any]: ...
+def convert_to_frontend_config(value: Any, *, is_transaction: bool = False) -> dict[str, Any]: ...

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: ...