Complete registration, small update profile.html

main/extra_func.py

@@ -7,55 +7,66 @@ from main.models import UserProfile
 
 
 # Дополнительные функции
-def set_and_get_name(user_profile: UserProfile):
+class ZendeskAdmin:
+    # Класс существует, чтобы в каждой фунциии отдельно не проверять аккаунт администратора
+    credentials = {
+        'subdomain': 'ngenix1612197338'
+    }
+    email = os.getenv('ACCESS_CONTROLLER_API_EMAIL')
+    token = os.getenv('ACCESS_CONTROLLER_API_TOKEN')
+    password = os.getenv('ACCESS_CONTROLLER_API_PASSWORD')
+
+    def __init__(self):
+        self.create_admin()
+
+    def check_user(self, email: str) -> bool:
+        return True if self.admin.search(email, type='user') else False
+
+    def get_user_name(self, email: str) -> str:
+        user = self.admin.users.search(email).values[0]
+        return user.name
+
+    def get_user_role(self, email: str) -> str:
+        user = self.admin.users.search(email).values[0]
+        return user.role
+
+    def get_user_id(self, email: str) -> str:
+        user = self.admin.users.search(email).values[0]
+        return user.id
+
+    def get_user_image(self, email: str) -> str:
+        user = self.admin.users.search(email).values[0]
+        return user.photo['content_url'] if user.photo else None
+
+    def create_admin(self) -> None:
+        if self.email is None:
+            raise ValueError('access_controller email not in env')
+        self.credentials['email'] = os.getenv('ACCESS_CONTROLLER_API_EMAIL')
+
+        if self.token:
+            self.credentials['token'] = self.token
+        elif self.password:
+            self.credentials['password'] = self.password
+        else:
+            raise ValueError('access_controller token or password not in env')
+        self.admin = Zenpy(**self.credentials)
+        try:
+            self.admin.search(self.email, type='user')
+        except APIException:
+            raise ValueError('invalid access_controller`s login data')
+
+
+def update_profile(user_profile: UserProfile):
     """
-    Функция устанавливает поле :class:`username` текущим именем в Zendesk
+    Функция обновляет профиль пользователя в соотвтетствии с текущим в Zendesk
 
-    .. TODO::
+    :param user_profile: Объект профиля пользователя
-      Переделать с получением данных через API
+    :type user_profile: :class:`main.models.UserProfile`
-
-    :param UP: Объект профиля пользователя
-    :type UP: :class:`main.models.UserProfile`
-    :return: Имя пользователя
-    :rtype: :class:`str`
     """
-    return user_profile.user.username
+    user_profile.name = ZendeskAdmin().get_user_name(user_profile.user.email)
-
+    user_profile.role = ZendeskAdmin().get_user_role(user_profile.user.email)
-
+    user_profile.image = ZendeskAdmin().get_user_image(user_profile.user.email)
-def set_and_get_email(user_profile: UserProfile):  # TODO: Переделать с получением данных через API
+    user_profile.save()
-    """
-    Функция устанавливает поле :class:`user.email` текущей почтой в Zendesk
-
-    :param UP: Объект профиля пользователя
-    :type UP: :class:`main.models.UserProfile`
-    :return: Почта пользователя
-    :rtype: :class:`str`
-    """
-    return user_profile.user.email
-
-
-def set_and_get_role(user_profile: UserProfile):  # TODO: Переделать с получением данных через API
-    """
-    Функция устанавливает поле :class:`role` текущей ролью в Zendesk
-
-    :param UP: Объект профиля пользователя
-    :type UP: :class:`main.models.UserProfile`
-    :return: Роль пользователя
-    :rtype: :class:`str`
-    """
-    return user_profile.role
-
-
-def load_and_get_image(user_profile: UserProfile):  # TODO: Переделать с получением изображения через API
-    """
-    Функция загружает и устанавливает изображение в поле :class:`image`
-
-    :param UP: Объект профиля пользователя
-    :type UP: :class:`main.models.UserProfile`
-    :return: Название изображения
-    :rtype: :class:`str`
-    """
-    return user_profile.image.name
 
 
 def check_user_exist(email: str) -> bool:
@@ -67,16 +78,7 @@ def check_user_exist(email: str) -> bool:
     :return: True, если существует, иначе False
     :rtype: :class:`bool`
     """
-    admin_creds = {
+    return ZendeskAdmin().check_user(email)
-        'email': os.environ.get('Admin_email'),
-        'subdomain': 'ngenix1612197338',
-        'token': os.environ.get('Oauth_token'),
-    }
-    admin = Zenpy(**admin_creds)
-    zenpy_user = admin.search(email, type='user')
-    if zenpy_user:
-        return True
-    return False
 
 
 def check_user_auth(email: str, password: str) -> bool:
@@ -91,12 +93,12 @@ def check_user_auth(email: str, password: str) -> bool:
     :raise :class:`APIException`: исключение, вызываемое если пользователь не аутентифицирован
     :rtype: :class:`bool`
     """
-    try:
+    creds = {
-        creds = {
             'email': email,
-            'subdomain': 'ngenix1612197338',
             'password': password,
+            'subdomain': 'ngenix1612197338',
         }
+    try:
         user = Zenpy(**creds)
         user.search(email, type='user')
     except APIException:

main/models.py

@@ -1,11 +1,22 @@
-import os
-
 from django.contrib.auth.models import User
 from django.db import models
+from django.db.models.signals import post_save
+from django.dispatch import receiver
 
 
 class UserProfile(models.Model):
     user = models.OneToOneField(to=User, on_delete=models.CASCADE)
-    role = models.IntegerField()
+    role = models.CharField(default='None', max_length=100)
-    image = models.ImageField(upload_to='user_avatars')
+    image = models.URLField(null=True, blank=True)
     name = models.CharField(default='None', max_length=100)
+
+
+@receiver(post_save, sender=User)
+def create_user_profile(sender, instance, created, **kwargs):
+    if created:
+        UserProfile.objects.create(user=instance)
+
+
+@receiver(post_save, sender=User)
+def save_user_profile(sender, instance, **kwargs):
+    instance.userprofile.save()

main/templates/pages/profile.html

@@ -14,8 +14,8 @@
     .img{
       width:auto;
       height:auto;
-      max-width:300px!important;
+      max-width:100px!important;
-      max-height:500px!important;
+      max-height:100px!important;
     }
 
 
@@ -28,8 +28,8 @@
 <div class="row">
   <div class="col-auto">
     <div class="container">
-      {% if image_name %}
+      {% if image_url %}
-      <img src="{% static image_name %}" class="img img-thumbnail" alt="Аватар">
+      <img src={{image_url}} class="img img-thumbnail" alt="Аватар">
       {% else %}
       <img src="{% static 'no_avatar.png' %}" class="img img-thumbnail" alt="Нет изображения">
       {% endif %}

main/views.py

@@ -1,8 +1,7 @@
 from django.shortcuts import render
 from django.urls import reverse_lazy
 
-from main.extra_func import set_and_get_name, set_and_get_email, load_and_get_image, set_and_get_role, check_user_exist, \
+from main.extra_func import check_user_exist, check_user_auth, update_profile
-    check_user_auth
 from main.models import UserProfile
 
 from django.contrib.auth.models import User
@@ -24,21 +23,15 @@ class CustomRegistrationView(RegistrationView):
     is_allowed = True
 
     def register(self, form):
+        self.is_allowed = True
         if check_user_exist(form.data['email']) and check_user_auth(form.data['email'], form.data['password_zen']):
             user = User.objects.create_user(
                 username=form.data['username'],
                 email=form.data['email'],
                 password=form.data['password1']
             )
-            profile = UserProfile(
+            profile = user.userprofile
-                user=user,
+            update_profile(profile)
-                role=0,
-            )
-            set_and_get_name(profile)
-            set_and_get_email(profile)
-            set_and_get_role(profile)
-            load_and_get_image(profile)
-            profile.save()
         else:
             self.is_allowed = False
 
@@ -58,19 +51,18 @@ def profile_page(request):
     """
     Отображение страницы профиля
 
-
     :param request: объект с деталями запроса
     :type request: :class:`django.http.HttpResponse`
     :return: объект ответа сервера с HTML-кодом внутри
     """
-    if request.user.is_authenticated:
+    user_profile = request.user.userprofile
-        user_profile = request.user.userprofile
+    update_profile(user_profile)
 
     context = {
-        'name': set_and_get_name(user_profile),
+        'email': user_profile.user.email,
-        'email': set_and_get_email(user_profile),
+        'name': user_profile.name,
-        'role': set_and_get_role(user_profile),
+        'role': user_profile.role,
-        'image_name': load_and_get_image(user_profile),
+        'image_url': user_profile.image,
         'pagename': 'Страница профиля'
     }
     return render(request, 'pages/profile.html', context)