# Generated by Django 4.2.16 on 2024-10-30 17:57

from django.db import migrations, models
from django.db import connections
from django.conf import settings

from core.models import AsyncMigrationStatus
from core.redis import start_job_async_or_sync

import logging
logger = logging.getLogger(__name__)
migration_name = '0052_auto_20241030_1757'

def forward_migration(migration_name, db_alias):
    migration = AsyncMigrationStatus.objects.using(db_alias).create(
        name=migration_name,
        status=AsyncMigrationStatus.STATUS_STARTED,
    )
    logger.info(f'Start async migration {migration_name}')

    conn = connections[db_alias]
    if conn.vendor == 'sqlite':
        sql_update_created_at = """
        UPDATE tasks_tasklock
        SET created_at = datetime(expire_at, %s);
        """
        sql_params = (f'-{settings.TASK_LOCK_TTL} seconds',)
    else:
        sql_update_created_at = """
        UPDATE tasks_tasklock
        SET created_at = expire_at - INTERVAL %s;
        """
        sql_params = ('%s seconds' % settings.TASK_LOCK_TTL,)

    with conn.cursor() as cursor:
        cursor.execute(sql_update_created_at, sql_params)

    migration.status = AsyncMigrationStatus.STATUS_FINISHED
    migration.save(using=db_alias)
    logger.info(f'Async migration {migration_name} complete')

def forwards(apps, schema_editor):
    # Dispatch migrations to rqworkers
    db_alias = schema_editor.connection.alias
    start_job_async_or_sync(forward_migration, migration_name=migration_name, db_alias=db_alias)

def backwards(apps, schema_editor):
    pass

class Migration(migrations.Migration):
    atomic = False

    dependencies = [
        ('tasks', '0051_tasklock_created_at'),
    ]

    operations = [
        migrations.AlterField(
            model_name='tasklock',
            name='created_at',
            field=models.DateTimeField(auto_now_add=True, help_text='Creation time', null=True, verbose_name='created at'),
        ),
        migrations.RunPython(forwards, backwards),
    ]