Commit 37621d4d authored by Gladys Forte's avatar Gladys Forte

Merge pull request #531 in RMS/api-main-service from gladys-dev2 to RMSv2

* commit '0e4f8c53':
  select_related
parents ca7d9461 0e4f8c53
...@@ -7,7 +7,7 @@ from django.conf import settings ...@@ -7,7 +7,7 @@ from django.conf import settings
class AppTokenAuthentication(TokenAuthentication): class AppTokenAuthentication(TokenAuthentication):
keyword = 'Bearer' keyword = 'Bearer'
def authenticate_credentials(self, key): def authenticate_credentials(self, key):
user, token = super(AppTokenAuthentication, user, token = super(AppTokenAuthentication,
......
...@@ -27,7 +27,7 @@ class Login(ObtainAuthToken): ...@@ -27,7 +27,7 @@ class Login(ObtainAuthToken):
def post(self, request, *args, **kwargs): def post(self, request, *args, **kwargs):
try: try:
serializer = self.serializer_class(data=request.data, serializer = self.serializer_class(data=request.data,
context={'request': request}) context={'request': request})
serializer.is_valid(raise_exception=True) serializer.is_valid(raise_exception=True)
user = serializer.validated_data['user'] user = serializer.validated_data['user']
token, created = Token.objects.get_or_create(user=user) token, created = Token.objects.get_or_create(user=user)
...@@ -87,7 +87,7 @@ class CurrentUser(APIView): ...@@ -87,7 +87,7 @@ class CurrentUser(APIView):
# @decorators.error_safe # @decorators.error_safe
def get(self, request, token=None, *args, **kwargs): def get(self, request, token=None, *args, **kwargs):
serializer = UserManagementRetreiveSerializer serializer = UserManagementRetreiveSerializer
context = {"request": request} context = {"request": request}
serializer = serializer(request.user, context=context) serializer = serializer(request.user, context=context)
......
...@@ -9,7 +9,7 @@ class AllowedCompanySerializer(serializers.ModelSerializer): ...@@ -9,7 +9,7 @@ class AllowedCompanySerializer(serializers.ModelSerializer):
ret['department'] = model_to_dict(instance.group_pivots) ret['department'] = model_to_dict(instance.group_pivots)
ret['company'] = model_to_dict(instance.company_pivot) ret['company'] = model_to_dict(instance.company_pivot)
return ret return ret
class Meta: class Meta:
model = AllowedCompany model = AllowedCompany
fields = '__all__' fields = '__all__'
......
...@@ -11,7 +11,7 @@ class AllowedCompanyFilterSet(filters.FilterSet): ...@@ -11,7 +11,7 @@ class AllowedCompanyFilterSet(filters.FilterSet):
# Q(username__icontains=value) | # Q(username__icontains=value) |
# Q(first_name__icontains=value) | # Q(first_name__icontains=value) |
# Q(last_name__icontains=value)) # Q(last_name__icontains=value))
class Meta: class Meta:
model = AllowedCompany model = AllowedCompany
fields = '__all__' fields = '__all__'
...@@ -20,7 +20,10 @@ from django.db import IntegrityError ...@@ -20,7 +20,10 @@ from django.db import IntegrityError
class AllowedCompanyViewSet(viewsets.ModelViewSet): class AllowedCompanyViewSet(viewsets.ModelViewSet):
queryset = AllowedCompany.objects.all() # queryset = AllowedCompany.objects.all()
queryset = AllowedCompany.objects.select_related(
'id_number', 'company_pivot', 'group_pivots'
).all()
serializer_class = AllowedCompanySerializer serializer_class = AllowedCompanySerializer
pagination_class = CustomPagination pagination_class = CustomPagination
lookup_field = 'code' lookup_field = 'code'
......
...@@ -8,6 +8,7 @@ class ChangeRequestFormApproversSerializer( ...@@ -8,6 +8,7 @@ class ChangeRequestFormApproversSerializer(
): ):
def to_representation(self, instance): def to_representation(self, instance):
ret = super().to_representation(instance) ret = super().to_representation(instance)
print(instance)
try: try:
ret['company'] = { ret['company'] = {
"id": instance.user.department.company.id, "id": instance.user.department.company.id,
......
...@@ -10,8 +10,7 @@ class headers(serializers.ModelSerializer): ...@@ -10,8 +10,7 @@ class headers(serializers.ModelSerializer):
ret['modified'] = instance.modified.strftime('%Y-%m-%d-%H:%M:%S') ret['modified'] = instance.modified.strftime('%Y-%m-%d-%H:%M:%S')
return ret return ret
class Meta: class Meta:
model = models.Company model = models.Company
fields = ( fields = (
......
...@@ -11,11 +11,11 @@ class UserTypeViewSet(APIView): ...@@ -11,11 +11,11 @@ class UserTypeViewSet(APIView):
enums_company = enums.UserTypeEnum.COMPANY_USER_ADMIN.value enums_company = enums.UserTypeEnum.COMPANY_USER_ADMIN.value
enums_department = enums.UserTypeEnum.DEPARTMENT_USER_ADMIN.value enums_department = enums.UserTypeEnum.DEPARTMENT_USER_ADMIN.value
enums_user = enums.UserTypeEnum.USER.value enums_user = enums.UserTypeEnum.USER.value
data = { data = {
'Super user': enums_super, 'Super user': enums_super,
'Overall user admin': enums_OUA, 'Overall user admin': enums_OUA,
"Company user admin": enums_company, "Company user admin": enums_company,
"Department user admin": enums_department, "Department user admin": enums_department,
"user": enums_user "user": enums_user
} }
......
...@@ -22,8 +22,6 @@ from django.conf import settings ...@@ -22,8 +22,6 @@ from django.conf import settings
from django.conf.urls.static import static from django.conf.urls.static import static
urlpatterns = [ urlpatterns = [
# path('admin/', admin.site.urls), # path('admin/', admin.site.urls),
path('api-auth/', include('rest_framework.urls')), path('api-auth/', include('rest_framework.urls')),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment