From 82977f902359d376fedc93fc9b8cabaf802ee5be Mon Sep 17 00:00:00 2001
From: fit2bot <68588906+fit2bot@users.noreply.github.com>
Date: Tue, 21 May 2024 19:24:26 +0800
Subject: [PATCH] perf: tickets notifications add default data migrate file
 (#13268)

Co-authored-by: feng <1304903146@qq.com>
---
 .../migrations/0003_initial_admin_msg.py      | 22 +++++++++++++++
 .../0003_initial_ticket_flow_data.py          | 27 +++++++++++++++++++
 2 files changed, 49 insertions(+)
 create mode 100644 apps/notifications/migrations/0003_initial_admin_msg.py
 create mode 100644 apps/tickets/migrations/0003_initial_ticket_flow_data.py

diff --git a/apps/notifications/migrations/0003_initial_admin_msg.py b/apps/notifications/migrations/0003_initial_admin_msg.py
new file mode 100644
index 000000000..cab077769
--- /dev/null
+++ b/apps/notifications/migrations/0003_initial_admin_msg.py
@@ -0,0 +1,22 @@
+# Generated by Django 4.1.13 on 2024-05-21 11:04
+
+from django.db import migrations
+
+
+def init_user_msg_subscription(apps, schema_editor):
+    User = apps.get_model('users', 'User')
+    UserMsgSubscription = apps.get_model('notifications', 'UserMsgSubscription')
+
+    receive_backends = ['site_msg', 'email']
+    user = User.objects.get(username='admin')
+    UserMsgSubscription.objects.create(user=user, receive_backends=receive_backends)
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ('notifications', '0002_auto_20210909_1946'),
+    ]
+
+    operations = [
+        migrations.RunPython(init_user_msg_subscription),
+    ]
diff --git a/apps/tickets/migrations/0003_initial_ticket_flow_data.py b/apps/tickets/migrations/0003_initial_ticket_flow_data.py
new file mode 100644
index 000000000..9e2b76e41
--- /dev/null
+++ b/apps/tickets/migrations/0003_initial_ticket_flow_data.py
@@ -0,0 +1,27 @@
+# Generated by Django 4.1.13 on 2024-05-21 09:14
+
+from django.db import migrations
+
+
+def create_ticket_flow_and_approval_rule(apps, schema_editor):
+    org_id = '00000000-0000-0000-0000-000000000000'
+
+    User = apps.get_model("users", "User")
+    TicketFlow = apps.get_model("tickets", "TicketFlow")
+    ApprovalRule = apps.get_model("tickets", "ApprovalRule")
+
+    super_user = User.objects.get(username='admin')
+    flow = TicketFlow.objects.create(created_by='System', type='apply_asset', org_id=org_id)
+    rule_instance = ApprovalRule.objects.create(strategy='super_admin')
+    rule_instance.assignees.set([super_user])
+    flow.rules.set([rule_instance, ])
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ('tickets', '0002_auto_20200728_1146'),
+    ]
+
+    operations = [
+        migrations.RunPython(create_ticket_flow_and_approval_rule),
+    ]