jumpserver/apps/tickets/migrations/0013_ticket_serial_num.py

# Generated by Django 3.1.13 on 2022-01-10 07:37

from django.db import migrations, models

from common.utils.timezone import as_current_tz


def fill_ticket_serial_number(apps, schema_editor):
    Ticket = apps.get_model('tickets', 'Ticket')
    tickets = Ticket.objects.all().order_by('date_created')

    curr_day = '00000000'
    curr_num = 1

    print(f'\n\tFill ticket serial number ... ')
    for ticket in tickets:
        # 跑这个脚本的时候，所有 ticket.serial_num == null
        date_created = as_current_tz(ticket.date_created)
        date_str = date_created.strftime('%Y%m%d')
        if date_str != curr_day:
            curr_day = date_str
            curr_num = 1

        ticket.serial_num = curr_day + '%04d' % curr_num
        curr_num += 1

    Ticket.objects.bulk_update(tickets, fields=('serial_num',))


class Migration(migrations.Migration):

    dependencies = [
        ('tickets', '0012_ticketsession'),
    ]

    operations = [
        migrations.AddField(
            model_name='ticket',
            name='serial_num',
            field=models.CharField(max_length=128, null=True, unique=True, verbose_name='Serial number'),
        ),
        migrations.RunPython(fill_ticket_serial_number),
    ]