|
@@ -1,10 +1,18 @@
|
|
|
+import os
|
|
|
+
|
|
|
from django.contrib.auth.models import Group
|
|
|
from django.dispatch import receiver
|
|
|
from django.db.models.signals import post_migrate
|
|
|
+from django.contrib.auth import get_user_model
|
|
|
+from django.contrib.auth.hashers import make_password
|
|
|
+from django.conf import settings
|
|
|
|
|
|
from tickets.models import TicketList
|
|
|
|
|
|
|
|
|
+User = get_user_model()
|
|
|
+
|
|
|
+
|
|
|
@receiver(post_migrate)
|
|
|
def create_initial_groups(sender, **kwargs):
|
|
|
Group.objects.get_or_create(pk=21, name='METASERVICE-ADMIN')
|
|
@@ -21,6 +29,60 @@ def create_initial_groups(sender, **kwargs):
|
|
|
Group.objects.get_or_create(pk=51, name='CLIENT')
|
|
|
Group.objects.get_or_create(pk=61, name='GUEST')
|
|
|
|
|
|
+ Group.objects.get_or_create(pk=99, name='TEST')
|
|
|
+
|
|
|
+
|
|
|
+@receiver(post_migrate)
|
|
|
+def create_test_users(sender, **kwargs):
|
|
|
+ """
|
|
|
+ Создает тестовых пользователей, но только если установочный скрипт был запущен с параметром --test-users
|
|
|
+ или была задана переменная окружения TEST_USERS=true.
|
|
|
+ """
|
|
|
+ if os.getenv('TEST_USERS') == 'true':
|
|
|
+ password = make_password("sharix-open-test")
|
|
|
+ test_group = Group.objects.get(pk=99)
|
|
|
+
|
|
|
+ # Группа, в которой надо создать пользователя и их количество
|
|
|
+ test_users = [
|
|
|
+ ('METASERVICE-ADMIN', 1),
|
|
|
+ ('METASERVICE-SUPERVISOR', 3),
|
|
|
+ ('METASERVICE-SUPPORT', 3),
|
|
|
+ ('METASERVICE-TECHSUPPORT', 3),
|
|
|
+
|
|
|
+ ('PARTNER-ADMIN', 1),
|
|
|
+ ('PARTNER-SUPERVISOR', 3),
|
|
|
+ ('PARTNER-SUPPORT', 3),
|
|
|
+ ('PARTNER-TECHSUPPORT', 3),
|
|
|
+
|
|
|
+ ('PROVIDER', 3),
|
|
|
+ ('CLIENT', 3),
|
|
|
+ ('GUEST', 3)
|
|
|
+ ]
|
|
|
+
|
|
|
+ for test_user in test_users:
|
|
|
+ group_name = test_user[0]
|
|
|
+ group = Group.objects.get(name=group_name)
|
|
|
+
|
|
|
+ for i in range(1, test_user[1] + 1):
|
|
|
+ user, created = User.objects.get_or_create(
|
|
|
+ phone_number=f"{group.pk}0{i}",
|
|
|
+ defaults={
|
|
|
+ 'last_name': i,
|
|
|
+ 'first_name': group_name,
|
|
|
+ 'email': f"test-{group_name.lower()}-{i}@domain.org",
|
|
|
+ 'username': f"test-{group_name.lower()}-{i}",
|
|
|
+ 'middle_name': "Test",
|
|
|
+ 'password': password
|
|
|
+ }
|
|
|
+ )
|
|
|
+
|
|
|
+ if created:
|
|
|
+ user.groups.add(
|
|
|
+ test_group,
|
|
|
+ group
|
|
|
+ )
|
|
|
+ print(f"Test user created: {user}")
|
|
|
+
|
|
|
|
|
|
@receiver(post_migrate)
|
|
|
def create_initial_ticket_lists(sender, **kwargs):
|
|
@@ -61,4 +123,12 @@ def create_initial_ticket_lists(sender, **kwargs):
|
|
|
TicketList.objects.get_or_create(pk=3207, name='Ручное подтверждение заявок (ACCESS-REQUEST)', group=Group.objects.get(name='PARTNER-SUPERVISOR'))
|
|
|
|
|
|
## TECHSUPPORT
|
|
|
- TicketList.objects.get_or_create(pk=3401, name='Входящие технические заявки (ST_REQUEST)', group=Group.objects.get(name='PARTNER-TECHSUPPORT'))
|
|
|
+ TicketList.objects.get_or_create(pk=3401, name='Входящие технические заявки (ST_REQUEST)', group=Group.objects.get(name='PARTNER-TECHSUPPORT'))
|
|
|
+
|
|
|
+
|
|
|
+# Откючаем тестовых пользователей, если DEBUG=False и наоброт
|
|
|
+test_group = Group.objects.get(name="TEST")
|
|
|
+users_in_test_group = User.objects.filter(groups=test_group)
|
|
|
+
|
|
|
+if users_in_test_group:
|
|
|
+ users_in_test_group.update(is_active=settings.DEBUG)
|