mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.3 KiB
44 lines
1.3 KiB
# 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'\nFill ticket serial number ... ', end='') |
|
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',)) |
|
print(len(tickets), end='') |
|
|
|
|
|
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), |
|
]
|
|
|