jumpserver/apps/tickets/migrations/0007_auto_20201224_1821.py

# Generated by Django 3.1 on 2020-12-24 10:21

from django.conf import settings
from django.db import migrations, models
import django.db.models.deletion
import tickets.models.ticket

TICKET_TYPE_APPLY_ASSET = 'apply_asset'


def migrate_field_type(tp):
    if tp == 'request_asset':
        return TICKET_TYPE_APPLY_ASSET
    return tp


def migrate_field_meta(tp, old_meta):
    if tp != TICKET_TYPE_APPLY_ASSET or not old_meta:
        return old_meta
    old_meta_hostname = old_meta.get('hostname')
    old_meta_system_user = old_meta.get('system_user')
    new_meta = {
        'apply_ip_group': old_meta.get('ips', []),
        'apply_hostname_group': [old_meta_hostname] if old_meta_hostname else [],
        'apply_system_user_group': [old_meta_system_user] if old_meta_system_user else [],
        'apply_actions': old_meta.get('actions'),
        'apply_actions_display': [],
        'apply_date_start': old_meta.get('date_start'),
        'apply_date_expired': old_meta.get('date_expired'),

        'approve_assets': old_meta.get('confirmed_assets', []),
        'approve_assets_display': [],
        'approve_system_users': old_meta.get('confirmed_system_users', []),
        'approve_system_users_display': [],
        'approve_actions': old_meta.get('actions'),
        'approve_actions_display': [],
        'approve_date_start': old_meta.get('date_start'),
        'approve_date_expired': old_meta.get('date_expired'),
    }
    return new_meta


ACTION_OPEN = 'open'
ACTION_CLOSE = 'close'
STATUS_OPEN = 'open'
STATUS_CLOSED = 'closed'


def migrate_field_action(old_action, old_status):
    if old_action:
        return old_action
    if old_status == STATUS_OPEN:
        return ACTION_OPEN
    if old_status == STATUS_CLOSED:
        return ACTION_CLOSE


def migrate_field_assignees_display(assignees_display):
    if not assignees_display:
        return []
    assignees_display = assignees_display.split(', ')
    return assignees_display


def migrate_tickets_fields_name(apps, schema_editor):
    ticket_model = apps.get_model("tickets", "Ticket")
    tickets = ticket_model.origin_objects.all()

    for ticket in tickets:
        ticket.applicant = ticket.user
        ticket.applicant_display = ticket.user_display
        ticket.processor = ticket.assignee
        ticket.processor_display = ticket.assignee_display
        ticket.assignees_display_new = migrate_field_assignees_display(ticket.assignees_display)
        ticket.action = migrate_field_action(ticket.action, ticket.status)
        ticket.type = migrate_field_type(ticket.type)
        ticket.meta = migrate_field_meta(ticket.type, ticket.meta)
        ticket.meta['body'] = ticket.body

    fields = [
        'applicant', 'applicant_display', 'processor', 'processor_display',
        'assignees_display_new', 'action', 'type', 'meta'
    ]
    ticket_model.origin_objects.bulk_update(tickets, fields)


class Migration(migrations.Migration):

    dependencies = [
        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
        ('tickets', '0006_auto_20201023_1628'),
    ]

    operations = [
        # model ticket
        migrations.AddField(
            model_name='ticket',
            name='applicant',
            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='applied_tickets', to=settings.AUTH_USER_MODEL, verbose_name='Applicant'),
        ),
        migrations.AddField(
            model_name='ticket',
            name='applicant_display',
            field=models.CharField(default='', max_length=256, verbose_name='Applicant display'),
        ),
        migrations.AddField(
            model_name='ticket',
            name='processor',
            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='processed_tickets', to=settings.AUTH_USER_MODEL, verbose_name='Processor'),
        ),
        migrations.AddField(
            model_name='ticket',
            name='processor_display',
            field=models.CharField(blank=True, default='', max_length=256, null=True, verbose_name='Processor display'),
        ),
        migrations.AddField(
            model_name='ticket',
            name='assignees_display_new',
            field=models.JSONField(default=list, encoder=tickets.models.ticket.ModelJSONFieldEncoder, verbose_name='Assignees display'),
        ),
        migrations.AlterField(
            model_name='ticket',
            name='assignees',
            field=models.ManyToManyField(related_name='assigned_tickets', to=settings.AUTH_USER_MODEL, verbose_name='Assignees'),
        ),
        migrations.AlterField(
            model_name='ticket',
            name='meta',
            field=models.JSONField(default=dict, encoder=tickets.models.ticket.ModelJSONFieldEncoder, verbose_name='Meta'),
        ),
        migrations.AlterField(
            model_name='ticket',
            name='type',
            field=models.CharField(choices=[('general', 'General'), ('login_confirm', 'Login confirm'), ('apply_asset', 'Apply for asset'), ('apply_application', 'Apply for application')], default='general', max_length=64, verbose_name='Type'),
        ),
        migrations.AlterField(
            model_name='ticket',
            name='action',
            field=models.CharField(choices=[('open', 'Open'), ('approve', 'Approve'), ('reject', 'Reject'), ('close', 'Close')], default='open', max_length=16, verbose_name='Action'),
        ),
        migrations.AlterField(
            model_name='ticket',
            name='status',
            field=models.CharField(choices=[('open', 'Open'), ('closed', 'Closed')], default='open', max_length=16, verbose_name='Status'),
        ),
        migrations.RunPython(migrate_tickets_fields_name),
        migrations.RemoveField(
            model_name='ticket',
            name='user',
        ),
        migrations.RemoveField(
            model_name='ticket',
            name='user_display',
        ),
        migrations.RemoveField(
            model_name='ticket',
            name='assignee',
        ),
        migrations.RemoveField(
            model_name='ticket',
            name='assignee_display',
        ),
        migrations.RemoveField(
            model_name='ticket',
            name='body',
        ),
        migrations.RemoveField(
            model_name='ticket',
            name='assignees_display',
        ),
        migrations.RenameField(
            model_name='ticket',
            old_name='assignees_display_new',
            new_name='assignees_display',
        ),
        migrations.AlterModelManagers(
            name='ticket',
            managers=[
            ],
        ),
        # model comment
        migrations.AlterField(
            model_name='comment',
            name='user_display',
            field=models.CharField(max_length=256, verbose_name='User display name'),
        ),
    ]