Pylint improvements

main/extra_func.py

@@ -36,7 +36,7 @@ def update_role(user_profile: UserProfile, role: int) -> None:
     zendesk.admin.users.update(user)
 
 
-def make_engineer(user_profile: UserProfile, who_changes: User) -> None:
+def make_engineer(user_profile: UserProfile) -> None:
     """
     Функция устанавливает пользователю роль инженера.
 
@@ -47,7 +47,7 @@ def make_engineer(user_profile: UserProfile, who_changes: User) -> None:
     update_role(user_profile, ROLES['engineer'])
 
 
-def make_light_agent(user_profile: UserProfile, who_changes: User) -> None:
+def make_light_agent(user_profile: UserProfile) -> None:
     """
     Функция устанавливает пользователю роль легкого агента.
 
@@ -293,8 +293,7 @@ class StatisticData:
             stat = self._use_display(stat)
             stat = self._use_interval(stat)
             return stat
-        else:
-            return None
+        return None
 
     def is_valid_statistic(self) -> bool:
         """
@@ -336,8 +335,7 @@ class StatisticData:
         """
         if self.is_valid_data():
             return self.data
-        else:
-            return None
+        return None
 
     def is_valid_data(self) -> bool:
         """

main/models.py

@@ -1,10 +1,8 @@
 """
 Модели, использующиеся в приложении.
 """
-
-
+from django.contrib.auth import get_user_model
 from django.db import models
-from django.contrib.auth.models import User
 from django.db.models.signals import post_save
 from django.dispatch import receiver
 from django.utils import timezone
@@ -24,7 +22,7 @@ class UserProfile(models.Model):
             ('has_control_access', 'Can view admin page'),
         )
 
-    user = models.OneToOneField(to=User, on_delete=models.CASCADE, help_text='Пользователь')
+    user = models.OneToOneField(to=get_user_model(), on_delete=models.CASCADE, help_text='Пользователь')
     role = models.CharField(default='None', max_length=100, help_text='Глобальное имя роли пользователя')
     custom_role_id = models.IntegerField(default=0, help_text='Код роли пользователя')
     image = models.URLField(null=True, blank=True, help_text='Аватарка')
@@ -44,7 +42,7 @@ class UserProfile(models.Model):
         return 'UNDEFINED'
 
 
-@receiver(post_save, sender=User)
+@receiver(post_save, sender=get_user_model())
 def create_user_profile(instance, created, **kwargs) -> None:
     """
     Функция создания профиля пользователя (Userprofile) при регистрации пользователя.
@@ -58,7 +56,7 @@ def create_user_profile(instance, created, **kwargs) -> None:
         UserProfile.objects.create(user=instance)
 
 
-@receiver(post_save, sender=User)
+@receiver(post_save, sender=get_user_model())
 def save_user_profile(instance, **kwargs) -> None:
     """
     Функция записи БД профиля пользователя.
@@ -74,12 +72,12 @@ class RoleChangeLogs(models.Model):
     """
     Модель для логирования изменений ролей пользователя.
     """
-    user = models.ForeignKey(to=User, on_delete=models.CASCADE,
+    user = models.ForeignKey(to=get_user_model(), on_delete=models.CASCADE,
                              help_text='Пользователь, которому присвоили другую роль')
     old_role = models.IntegerField(default=0, help_text='Старая роль')
     new_role = models.IntegerField(default=0, help_text='Присвоенная роль')
     change_time = models.DateTimeField(default=timezone.now, help_text='Дата и время изменения роли')
-    changed_by = models.ForeignKey(to=User, on_delete=models.CASCADE, related_name='changed_by',
+    changed_by = models.ForeignKey(to=get_user_model(), on_delete=models.CASCADE, related_name='changed_by',
                                    help_text='Кем была изменена роль')
 
 
@@ -106,7 +104,7 @@ class UnassignedTicket(models.Model):
     """
     Модель не распределенного тикета.
     """
-    assignee = models.ForeignKey(to=User, on_delete=models.CASCADE, related_name='tickets',
+    assignee = models.ForeignKey(to=get_user_model(), on_delete=models.CASCADE, related_name='tickets',
                                  help_text='Пользователь, с которого снят тикет')
     ticket_id = models.IntegerField(help_text='Номер тикера, для которого сняли ответственного')
     status = models.IntegerField(choices=UnassignedTicketStatus.choices, default=UnassignedTicketStatus.UNASSIGNED,

main/serializers.py

@@ -1,7 +1,7 @@
 """
 Сериализаторы.
 """
-from django.contrib.auth.models import User
+from django.contrib.auth import get_user_model
 from rest_framework import serializers
 from main.models import UserProfile
 from access_controller.settings import ZENDESK_ROLES
@@ -17,7 +17,7 @@ class UserSerializer(serializers.HyperlinkedModelSerializer):
     :type email: :class:`list`
     """
     class Meta:
-        model = User
+        model = get_user_model()
         fields = ['email']
 
 

main/views.py

@@ -230,7 +230,7 @@ def work_hand_over(request: WSGIRequest) -> HttpResponseRedirect:
     :param request: данные текущего пользователя (login_required)
     :return: перезагрузка текущей страницы после выполнения смены роли
     """
-    make_light_agent(request.user.userprofile, request.user)
+    make_light_agent(request.user.userprofile)
     return set_session_params_for_work_page(request)
 
 
@@ -243,7 +243,7 @@ def work_become_engineer(request: WSGIRequest) -> HttpResponseRedirect:
     :param request: данные текущего пользователя (login_required)
     :return: перезагрузка текущей страницы после выполнения смены роли
     """
-    make_engineer(request.user.userprofile, request.user)
+    make_engineer(request.user.userprofile)
     return set_session_params_for_work_page(request)
 
 
@@ -317,7 +317,7 @@ class AdminPageView(LoginRequiredMixin, PermissionRequiredMixin, SuccessMessageM
         :return: Обновленный список пользователей
         """
         for user in users:
-            make_engineer(user, self.request.user)
+            make_engineer(user)
             log(user, self.request.user.userprofile)
 
     def make_light_agents(self, users: list) -> None:
@@ -328,7 +328,7 @@ class AdminPageView(LoginRequiredMixin, PermissionRequiredMixin, SuccessMessageM
         :return: Обновленный список пользователей
         """
         for user in users:
-            make_light_agent(user, self.request.user)
+            make_light_agent(user)
             log(user, self.request.user.userprofile)