signals.py 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. import os
  2. from django.conf import settings
  3. from django.contrib.auth import get_user_model
  4. from django.contrib.auth.hashers import make_password
  5. from django.contrib.auth.models import Group
  6. from django.db import models
  7. from django.db.models.signals import post_migrate
  8. from django.dispatch import receiver
  9. from tickets.models import TicketList
  10. from core.settings_vars import GENERATE_TEST_USERS
  11. User = get_user_model()
  12. @receiver(post_migrate)
  13. def create_initial_groups(sender, **kwargs):
  14. groups = [
  15. ('METASERVICE-ADMIN', 21),
  16. ('METASERVICE-SUPERVISOR', 22),
  17. ('METASERVICE-SUPPORT', 23),
  18. ('METASERVICE-TECHSUPPORT', 24),
  19. ('PARTNER-ADMIN', 31),
  20. ('PARTNER-SUPERVISOR', 32),
  21. ('PARTNER-SUPPORT', 33),
  22. ('PARTNER-TECHSUPPORT', 34),
  23. ('PROVIDER', 41),
  24. ('CLIENT', 51),
  25. ('GUEST', 61),
  26. ('TEST', 99)
  27. ]
  28. for name, pk in groups:
  29. group, created = Group.objects.get_or_create(name=name, defaults={'pk': pk})
  30. if not created:
  31. print(f"Group {name} already exists.")
  32. @receiver(post_migrate)
  33. def create_test_users(sender, **kwargs):
  34. """
  35. Создает тестовых пользователей, если:
  36. - В файле settings_vers.py переменная GENERATE_TEST_USERS = True
  37. - Или была задана переменная окружения TEST_USERS=true
  38. """
  39. if os.getenv('TEST_USERS') == 'true' or GENERATE_TEST_USERS:
  40. password = make_password("sharix-open-test")
  41. test_group = Group.objects.get(pk=99)
  42. # Группа, в которой надо создать пользователя и их количество
  43. test_users = [
  44. ('METASERVICE-ADMIN', 1),
  45. ('METASERVICE-SUPERVISOR', 3),
  46. ('METASERVICE-SUPPORT', 3),
  47. ('METASERVICE-TECHSUPPORT', 3),
  48. ('PARTNER-ADMIN', 1),
  49. ('PARTNER-SUPERVISOR', 3),
  50. ('PARTNER-SUPPORT', 3),
  51. ('PARTNER-TECHSUPPORT', 3),
  52. ('PROVIDER', 3),
  53. ('CLIENT', 3),
  54. ('GUEST', 3)
  55. ]
  56. for test_user in test_users:
  57. group_name = test_user[0]
  58. group = Group.objects.get(name=group_name)
  59. for i in range(1, test_user[1] + 1):
  60. user, created = User.objects.get_or_create(
  61. phone_number=f"{group.pk}0{i}",
  62. defaults={
  63. 'last_name': i,
  64. 'first_name': group_name,
  65. 'email': f"test-{group_name.lower()}-{i}@domain.org",
  66. 'username': f"test-{group_name.lower()}-{i}",
  67. 'middle_name': "Test",
  68. 'password': password
  69. }
  70. )
  71. if created:
  72. user.groups.add(
  73. test_group,
  74. group
  75. )
  76. print(f"Test user created: {user}")
  77. @receiver(post_migrate)
  78. def create_initial_ticket_lists(sender, **kwargs):
  79. if sender.name == 'tickets':
  80. ticket_data = [
  81. # METASERVICE
  82. (2101, 'Активация партнеров (NEG_REQUEST)', 'METASERVICE-ADMIN'),
  83. (2102, 'Права в сервисе (ACCESS_REQUEST)', 'METASERVICE-ADMIN'),
  84. (2103, 'Проверка документов (ST_REQUEST)', 'METASERVICE-ADMIN'),
  85. (2201, 'Активность пользователей (ST_REQUEST)', 'METASERVICE-SUPERVISOR'),
  86. (2202, 'Оперативный доступ (ST_REQUEST)', 'METASERVICE-SUPERVISOR'),
  87. (2203, 'Оперативный доступ экстра (ACCESS_REQUEST)', 'METASERVICE-SUPERVISOR'),
  88. (2204, 'Права сервиса (ST_REQUEST)', 'METASERVICE-SUPERVISOR'),
  89. (2205, 'Проверка документов (ST_REQUEST)', 'METASERVICE-SUPERVISOR'),
  90. (2301, 'Входящие обычные заявки (ST_REQUEST)', 'METASERVICE-SUPPORT'),
  91. (2302, 'Заявки на услуги сервиса (SERVICE_REQUEST)', 'METASERVICE-SUPPORT'),
  92. (2401, 'Входящие технические заявки (ST_REQUEST)', 'METASERVICE-TECHSUPPORT'),
  93. # PARTNER
  94. (3101, 'Документы исполнителей (ST_REQUEST)', 'PARTNER-ADMIN'),
  95. (3102, 'Документы ресурсов (ST_REQUEST)', 'PARTNER-ADMIN'),
  96. (3103, 'Права в партнерке (ACCESS_REQUEST)', 'PARTNER-ADMIN'),
  97. (3201, 'Активность внутри партнера (ST_REQUEST)', 'PARTNER-SUPERVISOR'),
  98. (3202, 'Документы исполнителей (ST_REQUEST)', 'PARTNER-SUPERVISOR'),
  99. (3203, 'Документы ресурсов (ST_REQUEST)', 'PARTNER-SUPERVISOR'),
  100. (3204, 'Доступ внутри партнера (ST_REQUEST)', 'PARTNER-SUPERVISOR'),
  101. (3205, 'Оперативный доступ (ST_REQUEST)', 'PARTNER-SUPERVISOR'),
  102. (3206, 'Оперативный доступ экстра (ST_REQUEST)', 'PARTNER-SUPERVISOR'),
  103. (3207, 'Ручное подтверждение заявок (ACCESS-REQUEST)', 'PARTNER-SUPERVISOR'),
  104. (3401, 'Входящие технические заявки (ST_REQUEST)', 'PARTNER-TECHSUPPORT'),
  105. ]
  106. for pk, name, group_name in ticket_data:
  107. group = Group.objects.get(name=group_name)
  108. # Проверяем существование тикета с таким же name и group
  109. ticket_exists = TicketList.objects.filter(group=group, name=name).exists()
  110. if not ticket_exists:
  111. TicketList.objects.create(pk=pk, name=name, group=group)
  112. # Откючаем тестовых пользователей (если такие есть), если DEBUG=False и наоброт
  113. try:
  114. if Group.objects.filter(name="TEST").exists():
  115. test_group = Group.objects.get(name="TEST")
  116. users_in_test_group = User.objects.filter(groups=test_group)
  117. if users_in_test_group:
  118. users_in_test_group.update(is_active=settings.DEBUG)
  119. except:
  120. print("Test user validation is not available. Models have not been created yet")