Add support for querying datetime fields

.codespellrc

@@ -0,0 +1,3 @@
+[codespell]
+skip = .git,poetry.lock,*.pyc,__pycache__
+ignore-words-list = redis,migrator,datetime,timestamp,asyncio,redisearch,pydantic,ulid,hnsw

.github/wordlist.txt

@@ -70,4 +70,8 @@ unix
 utf
 validator
 validators
-virtualenv
+virtualenv
+datetime
+Datetime
+reindex
+schemas

.pre-commit-config.yaml

@@ -0,0 +1,7 @@
+repos:
+  - repo: https://github.com/codespell-project/codespell
+    rev: v2.2.6
+    hooks:
+      - id: codespell
+        args: [--write-changes]
+        exclude: ^(poetry\.lock|\.git/|docs/.*\.md)$

aredis_om/cli/__init__.py

@@ -0,0 +1 @@
+# CLI package

aredis_om/cli/main.py

@@ -0,0 +1,24 @@
+"""
+Redis OMCLI - Main entry point for the async 'om' command.
+"""
+
+import click
+
+from ..model.cli.migrate import migrate
+from ..model.cli.migrate_data import migrate_data
+
+
+@click.group()
+@click.version_option()
+def om():
+    """Redis OM Python CLI - Object mapping and migrations for Redis."""
+    pass
+
+
+# Add subcommands
+om.add_command(migrate)
+om.add_command(migrate_data, name="migrate-data")
+
+
+if __name__ == "__main__":
+    om()

aredis_om/model/cli/legacy_migrate.py

@@ -0,0 +1,123 @@
+import asyncio
+import os
+import warnings
+from typing import Optional
+
+import click
+
+from ...settings import get_root_migrations_dir
+from ..migrations.migrator import Migrator
+
+
+def run_async(coro):
+    """Run an async coroutine in an isolated event loop to avoid interfering with pytest loops."""
+    import concurrent.futures
+
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        future = executor.submit(asyncio.run, coro)
+        return future.result()
+
+
+def show_deprecation_warning():
+    """Show deprecation warning for the legacy migrate command."""
+    warnings.warn(
+        "The 'migrate' command is deprecated. Please use 'om migrate' for the new file-based migration system with rollback support.",
+        DeprecationWarning,
+        stacklevel=3,
+    )
+    click.echo(
+        click.style(
+            "⚠️  DEPRECATED: The 'migrate' command uses automatic migrations. "
+            "Use 'om migrate' for the new file-based system with rollback support.",
+            fg="yellow",
+        ),
+        err=True,
+    )
+
+
+@click.group()
+def migrate():
+    """[DEPRECATED] Automatic schema migrations for Redis OM models. Use 'om migrate' instead."""
+    show_deprecation_warning()
+
+
+@migrate.command()
+@click.option("--module", help="Python module to scan for models")
+def status(module: Optional[str]):
+    """Show pending automatic migrations (no file-based tracking)."""
+    migrator = Migrator(module=module)
+
+    async def _status():
+        await migrator.detect_migrations()
+        return migrator.migrations
+
+    migrations = run_async(_status())
+
+    if not migrations:
+        click.echo("No pending automatic migrations detected.")
+        return
+
+    click.echo("Pending Automatic Migrations:")
+    for migration in migrations:
+        action = "CREATE" if migration.action.name == "CREATE" else "DROP"
+        click.echo(
+            f"  {action}: {migration.index_name} (model: {migration.model_name})"
+        )
+
+
+@migrate.command()
+@click.option("--module", help="Python module to scan for models")
+@click.option(
+    "--dry-run", is_flag=True, help="Show what would be done without applying changes"
+)
+@click.option("--verbose", "-v", is_flag=True, help="Enable verbose output")
+@click.option(
+    "--yes",
+    "-y",
+    is_flag=True,
+    help="Skip confirmation prompt to run automatic migrations",
+)
+def run(
+    module: Optional[str],
+    dry_run: bool,
+    verbose: bool,
+    yes: bool,
+):
+    """Run automatic schema migrations (immediate DROP+CREATE)."""
+    migrator = Migrator(module=module)
+
+    async def _run():
+        await migrator.detect_migrations()
+        if not migrator.migrations:
+            if verbose:
+                click.echo("No pending automatic migrations found.")
+            return 0
+
+        if dry_run:
+            click.echo(f"Would run {len(migrator.migrations)} automatic migration(s):")
+            for migration in migrator.migrations:
+                action = "CREATE" if migration.action.name == "CREATE" else "DROP"
+                click.echo(f"  {action}: {migration.index_name}")
+            return len(migrator.migrations)
+
+        if not yes:
+            operations = []
+            for migration in migrator.migrations:
+                action = "CREATE" if migration.action.name == "CREATE" else "DROP"
+                operations.append(f"  {action}: {migration.index_name}")
+
+            if not click.confirm(
+                f"Run {len(migrator.migrations)} automatic migration(s)?\n"
+                + "\n".join(operations)
+            ):
+                click.echo("Aborted.")
+                return 0
+
+        await migrator.run()
+        if verbose:
+            click.echo(
+                f"Successfully applied {len(migrator.migrations)} automatic migration(s)."
+            )
+        return len(migrator.migrations)
+
+    run_async(_run())

aredis_om/model/cli/migrate.py

@@ -1,18 +1,183 @@
+import asyncio
+import os
+from typing import Optional
+
 import click
 
-from aredis_om.model.migrations.migrator import Migrator
+from ...settings import get_root_migrations_dir
+from ..migrations.schema_migrator import SchemaMigrator
+
+
+def run_async(coro):
+    """Run an async coroutine in an isolated event loop to avoid interfering with pytest loops."""
+    import concurrent.futures
+
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        future = executor.submit(asyncio.run, coro)
+        return future.result()
+
+
+@click.group()
+def migrate():
+    """Manage schema migrations for Redis OM models."""
+    pass
+
+
+@migrate.command()
+@click.option("--migrations-dir", help="Directory containing schema migration files")
+def status(migrations_dir: Optional[str]):
+    """Show current schema migration status from files."""
+    dir_path = migrations_dir or os.path.join(
+        get_root_migrations_dir(), "schema-migrations"
+    )
+    migrator = SchemaMigrator(migrations_dir=dir_path)
+    status_info = run_async(migrator.status())
+
+    click.echo("Schema Migration Status:")
+    click.echo(f"  Total migrations: {status_info['total_migrations']}")
+    click.echo(f"  Applied: {status_info['applied_count']}")
+    click.echo(f"  Pending: {status_info['pending_count']}")
+
+    if status_info["pending_migrations"]:
+        click.echo("\nPending migrations:")
+        for migration_id in status_info["pending_migrations"]:
+            click.echo(f"- {migration_id}")
+
+    if status_info["applied_migrations"]:
+        click.echo("\nApplied migrations:")
+        for migration_id in status_info["applied_migrations"]:
+            click.echo(f"- {migration_id}")
+
+
+@migrate.command()
+@click.option("--migrations-dir", help="Directory containing schema migration files")
+@click.option(
+    "--dry-run", is_flag=True, help="Show what would be done without applying changes"
+)
+@click.option("--verbose", "-v", is_flag=True, help="Enable verbose output")
+@click.option("--limit", type=int, help="Limit number of migrations to run")
+@click.option(
+    "--yes",
+    "-y",
+    is_flag=True,
+    help="Skip confirmation prompt to create directory or run",
+)
+def run(
+    migrations_dir: Optional[str],
+    dry_run: bool,
+    verbose: bool,
+    limit: Optional[int],
+    yes: bool,
+):
+    """Run pending schema migrations from files."""
+    dir_path = migrations_dir or os.path.join(
+        get_root_migrations_dir(), "schema-migrations"
+    )
+
+    if not os.path.exists(dir_path):
+        if yes or click.confirm(f"Create schema migrations directory at '{dir_path}'?"):
+            os.makedirs(dir_path, exist_ok=True)
+        else:
+            click.echo("Aborted.")
+            return
+
+    migrator = SchemaMigrator(migrations_dir=dir_path)
+
+    # Show list for confirmation
+    if not dry_run and not yes:
+        status_info = run_async(migrator.status())
+        if status_info["pending_migrations"]:
+            listing = "\n".join(
+                f"- {m}"
+                for m in status_info["pending_migrations"][
+                    : (limit or len(status_info["pending_migrations"]))
+                ]
+            )
+            if not click.confirm(
+                f"Run {min(limit or len(status_info['pending_migrations']), len(status_info['pending_migrations']))} migration(s)?\n{listing}"
+            ):
+                click.echo("Aborted.")
+                return
+
+    count = run_async(migrator.run(dry_run=dry_run, limit=limit, verbose=verbose))
+    if verbose and not dry_run:
+        click.echo(f"Successfully applied {count} migration(s).")
+
+
+@migrate.command()
+@click.argument("name")
+@click.option("--migrations-dir", help="Directory to create migration in")
+@click.option(
+    "--yes", "-y", is_flag=True, help="Skip confirmation prompt to create directory"
+)
+def create(name: str, migrations_dir: Optional[str], yes: bool):
+    """Create a new schema migration snapshot file from current pending operations."""
+    dir_path = migrations_dir or os.path.join(
+        get_root_migrations_dir(), "schema-migrations"
+    )
+
+    if not os.path.exists(dir_path):
+        if yes or click.confirm(f"Create schema migrations directory at '{dir_path}'?"):
+            os.makedirs(dir_path, exist_ok=True)
+        else:
+            click.echo("Aborted.")
+            return
+
+    migrator = SchemaMigrator(migrations_dir=dir_path)
+    filepath = run_async(migrator.create_migration_file(name))
+    if filepath:
+        click.echo(f"Created migration: {filepath}")
+    else:
+        click.echo("No pending schema changes detected. Nothing to snapshot.")
+
+
+@migrate.command()
+@click.argument("migration_id")
+@click.option("--migrations-dir", help="Directory containing schema migration files")
+@click.option(
+    "--dry-run", is_flag=True, help="Show what would be done without applying changes"
+)
+@click.option("--verbose", "-v", is_flag=True, help="Enable verbose output")
+@click.option(
+    "--yes",
+    "-y",
+    is_flag=True,
+    help="Skip confirmation prompt to create directory or run",
+)
+def rollback(
+    migration_id: str,
+    migrations_dir: Optional[str],
+    dry_run: bool,
+    verbose: bool,
+    yes: bool,
+):
+    """Rollback a specific schema migration by ID."""
+    dir_path = migrations_dir or os.path.join(
+        get_root_migrations_dir(), "schema-migrations"
+    )
 
+    if not os.path.exists(dir_path):
+        if yes or click.confirm(f"Create schema migrations directory at '{dir_path}'?"):
+            os.makedirs(dir_path, exist_ok=True)
+        else:
+            click.echo("Aborted.")
+            return
 
-@click.command()
-@click.option("--module", default="aredis_om")
-def migrate(module: str):
-    migrator = Migrator(module)
-    migrator.detect_migrations()
+    migrator = SchemaMigrator(migrations_dir=dir_path)
 
-    if migrator.migrations:
-        print("Pending migrations:")
-        for migration in migrator.migrations:
-            print(migration)
+    if not yes and not dry_run:
+        if not click.confirm(f"Rollback migration '{migration_id}'?"):
+            click.echo("Aborted.")
+            return
 
-        if input("Run migrations? (y/n) ") == "y":
-            migrator.run()
+    success = run_async(
+        migrator.rollback(migration_id, dry_run=dry_run, verbose=verbose)
+    )
+    if success:
+        if verbose:
+            click.echo(f"Successfully rolled back migration: {migration_id}")
+    else:
+        click.echo(
+            f"Migration '{migration_id}' does not support rollback or is not applied.",
+            err=True,
+        )