Users | Refactors/about yourself #397
Conversation
| @staticmethod | ||
| def get_user_model(): | ||
| return Admin | ||
|
|
||
| @staticmethod | ||
| def get_reverse_url(): | ||
| return "user-account" | ||
|
|
| } | ||
|
|
||
| def retrieve(self, request, user_uuid=None): | ||
| user = BaseAbstractUser.objects.filter(uuid=user_uuid).first() |
There was a problem hiding this comment.
user = BaseAbstractUser.objects.filter(uuid=user_uuid).first()
так не сработает
надо переопределить метод get_queryset
def get_queryset(self):
qs = {
'administrator': Admin.objects.all(),
'developer': CompanyUser.objects.all(),
'customer': CustomerUser.objects.all(),
}
role = self.request.user._meta.app_label
return qs.get(role, [])
и в методе retrive получать пользователя
user = self.get_object()
| permission_map = { | ||
| "default": [IsAuthenticated], | ||
| "retrieve": [IsAuthenticated], | ||
| "partial_update": [IsAuthenticated], | ||
| "destroy": [IsAuthenticated], | ||
| } |
There was a problem hiding this comment.
можно один permission_classes сделать
| elif role == 'developer': | ||
| return CompanyUser.objects.all() | ||
|
|
||
| return BaseAbstractUser.objects.none() |
There was a problem hiding this comment.
лучше пустой список возвращать
| def get_queryset(self): | ||
| role = self.request.user._meta.app_label | ||
|
|
||
| if role == 'administrator': | ||
| return Admin.objects.all() | ||
| elif role == 'customer': | ||
| return CustomerUser.objects.all() | ||
| elif role == 'developer': | ||
| return CompanyUser.objects.all() | ||
|
|
||
| return [] |
There was a problem hiding this comment.
давай тут тоже вот так сделаем
def get_queryset(self):
qs = {
'administrator': Admin.objects.all(),
'developer': CompanyUser.objects.all(),
'customer': CustomerUser.objects.all(),
}
role = self.request.user._meta.app_label
return qs.get(role, [])
| ) | ||
| class AccountMultipleUsersViewSet(viewsets.ViewSet): | ||
| http_method_names = ["post"] | ||
| permission_map = UserPermissionClass.get_permission_map() |
There was a problem hiding this comment.
это лишнее
я имел ввиду сделать так
permission_class = [IsAuthenticated]
так как все методы для авторизованного пользователя
|
|
||
| def create(self, request): | ||
| uuid_list = request.data.get('uuid_list', []) | ||
| users = BaseAbstractUser.objects.filter(uuid__in=uuid_list) |
There was a problem hiding this comment.
нельзя фильтровать по абстрактному классу
можно сделать отдельный метод который будет возвращать всех пользователей
def get_users_queryset(users_uuid: list):
admins = Admin.objects.filter(id__in=users_uuid)
customers = CustomerUser.objects.filter(id__in=users_uuid)
developers = CompanyUser.objects.filter(id__in=users_uuid)
return admin | customers | developers
There was a problem hiding this comment.
я описал как это можно сделать по другому, так что можешь удалить этот файл
| administrators = instance.filter(role='administrator') | ||
| developers = instance.filter(role='developer') | ||
| customers = instance.filter(role='customer') |
There was a problem hiding this comment.
по ролям вряд-ли у нас отфильтруется
| return administrators, customers, developers | ||
|
|
||
| def create(self, request): | ||
| uuid_list = request.data.get('uuid_list', []) | ||
| administrators, customers, developers = self.get_users_queryset(users_uuid=uuid_list) |
There was a problem hiding this comment.
лучше возвращать dict вместо tuple кверисетов
return {'administrator': administrator...}
так как случайно можем поменять порядок и данные перемешаются
и в методе create просто доставатб из словаря
| def get_permissions(self): | ||
| return [ | ||
| permission() | ||
| for permission in self.permission_map.get(self.action, self.permission_map["default"]) | ||
| ] | ||
| return [permission() for permission in self.permission_classes] |
There was a problem hiding this comment.
можно полностью удалить этот метод
плюс надо перенести все методы из вьюшки PersonalAccount в класс AccountViewSet
так как эта вьюшка наследуется от PersonalAccount
| path( | ||
| "account/me/<uuid:user_id>/", | ||
| AccountSingleUserViewSet.as_view( | ||
| {"get": "retrieve", "put": "partial_update", "delete": "destroy"}, |
There was a problem hiding this comment.
давай сделаем так во вьюшки AccountViewSet добавишь action bu_uuid для GET и для POST методов
для GET перенесешь функционал вьюшки AccountSingleUserViewSet
а для POST из вьюшки AccountMultipleUsersViewSet
There was a problem hiding this comment.
так будет удобнее, все будет в одной вьюшке, а не в трех
No description provided.