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

188 lines
6.8 KiB
Python

# 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'),
),
]