Rename class

Author: gmazoyer

backend/infrahub/cli/db.py

@@ -35,10 +35,7 @@
     GraphRelationshipIsPartOf,
     GraphRelationshipProperties,
 )
-from infrahub.core.initialization import (
-    get_root_node,
-    initialize_registry,
-)
+from infrahub.core.initialization import get_root_node, initialize_registry
 from infrahub.core.migrations.graph import get_graph_migrations, get_migration_by_number
 from infrahub.core.migrations.schema.models import SchemaApplyMigrationData
 from infrahub.core.migrations.schema.tasks import schema_apply_migrations
@@ -69,7 +66,7 @@ def get_timestamp_string() -> str:
         ArbitraryMigration,
         GraphMigration,
         InternalSchemaMigration,
-        MigrationWithRebase,
+        MigrationRequiringRebase,
     )
     from infrahub.database import InfrahubDatabase
     from infrahub.database.index import IndexManagerBase
@@ -289,10 +286,10 @@ async def index(
 
 async def detect_migration_to_run(
     current_graph_version: int, migration_number: int | str | None = None
-) -> Sequence[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationWithRebase]:
+) -> Sequence[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationRequiringRebase]:
     """Return a sequence of migrations to apply to upgrade the database."""
     rprint("Checking current state of the database")
-    migrations: list[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationWithRebase] = []
+    migrations: list[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationRequiringRebase] = []
 
     if migration_number:
         migration = get_migration_by_number(migration_number)
@@ -319,7 +316,7 @@ async def detect_migration_to_run(
 
 async def migrate_database(
     db: InfrahubDatabase,
-    migrations: Sequence[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationWithRebase],
+    migrations: Sequence[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationRequiringRebase],
     initialize: bool = False,
 ) -> bool:
     """Apply the latest migrations to the database, this function will print the status directly in the console.

backend/infrahub/core/migrations/graph/__init__.py

@@ -47,9 +47,9 @@
 from .m043_backfill_hfid_display_label_in_db import Migration043
 
 if TYPE_CHECKING:
-    from ..shared import ArbitraryMigration, GraphMigration, InternalSchemaMigration, MigrationWithRebase
+    from ..shared import ArbitraryMigration, GraphMigration, InternalSchemaMigration, MigrationRequiringRebase
 
-MIGRATIONS: list[type[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationWithRebase]] = [
+MIGRATIONS: list[type[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationRequiringRebase]] = [
     Migration001,
     Migration002,
     Migration003,
@@ -98,7 +98,7 @@
 
 async def get_graph_migrations(
     current_graph_version: int,
-) -> Sequence[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationWithRebase]:
+) -> Sequence[GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationRequiringRebase]:
     applicable_migrations = []
     for migration_class in MIGRATIONS:
         migration = migration_class.init()
@@ -111,7 +111,7 @@ async def get_graph_migrations(
 
 def get_migration_by_number(
     migration_number: int | str,
-) -> GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationWithRebase:
+) -> GraphMigration | InternalSchemaMigration | ArbitraryMigration | MigrationRequiringRebase:
     # Convert to string and pad with zeros if needed
     try:
         num = int(migration_number)

backend/infrahub/core/migrations/graph/m043_backfill_hfid_display_label_in_db.py

@@ -14,7 +14,7 @@
 from infrahub.core.schema.manager import SchemaManager
 from infrahub.types import is_large_attribute_type
 
-from ..shared import MigrationWithRebase
+from ..shared import MigrationRequiringRebase
 
 if TYPE_CHECKING:
     from infrahub.core.branch import Branch
@@ -353,7 +353,7 @@ async def query_init(self, db: InfrahubDatabase, **kwargs: dict[str, Any]) -> No
         self.add_to_query(set_value_query)
 
 
-class Migration043(MigrationWithRebase):
+class Migration043(MigrationRequiringRebase):
     """
     Backfill `human_friendly_id` and `display_label` attributes for nodes with schemas that define them.
     """

backend/infrahub/core/migrations/runner.py

@@ -5,7 +5,7 @@
 from infrahub.core.migrations.graph import MIGRATIONS
 
 from .exceptions import MigrationFailureError
-from .shared import MigrationWithRebase
+from .shared import MigrationRequiringRebase
 
 if TYPE_CHECKING:
     from infrahub.core.branch import Branch
@@ -17,9 +17,9 @@ def __init__(self, branch: Branch) -> None:
         self.branch = branch
         self.applicable_migrations = self._get_applicable_migrations()
 
-    def _get_applicable_migrations(self) -> Sequence[MigrationWithRebase]:
+    def _get_applicable_migrations(self) -> Sequence[MigrationRequiringRebase]:
         applicable_migrations = []
-        for migration_class in [m for m in MIGRATIONS if issubclass(m, MigrationWithRebase)]:
+        for migration_class in [m for m in MIGRATIONS if issubclass(m, MigrationRequiringRebase)]:
             migration = migration_class.init()
             if self.branch.graph_version and self.branch.graph_version > migration.minimum_version:
                 continue

backend/infrahub/core/migrations/shared.py

@@ -224,7 +224,7 @@ async def execute(self, db: InfrahubDatabase) -> MigrationResult:
         raise NotImplementedError()
 
 
-class MigrationWithRebase(BaseModel):
+class MigrationRequiringRebase(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)
     name: str = Field(..., description="Name of the migration")
     minimum_version: int = Field(..., description="Minimum version of the graph to execute this migration")