Tmp commit

This commit is contained in:
littlewolf 2019-12-10 17:09:43 +03:00
parent ed18d1c44d
commit 47cb7f8e63
2 changed files with 22 additions and 0 deletions

View File

@ -10,4 +10,5 @@ urlpatterns = [
path('user-role/', views.UserRoleLstView.as_view(), name='user-role-list-create'), path('user-role/', views.UserRoleLstView.as_view(), name='user-role-list-create'),
path('user/', views.UserLstView.as_view(), name='user-create-list'), path('user/', views.UserLstView.as_view(), name='user-create-list'),
path('user/<int:id>/', views.UserRUDView.as_view(), name='user-rud'), path('user/<int:id>/', views.UserRUDView.as_view(), name='user-rud'),
path('user/<int:id>/csv', views.get_user_csv, name='user-csv'),
] ]

View File

@ -1,6 +1,8 @@
from django_filters.rest_framework import DjangoFilterBackend from django_filters.rest_framework import DjangoFilterBackend
from rest_framework import generics, permissions from rest_framework import generics, permissions
from rest_framework.filters import OrderingFilter from rest_framework.filters import OrderingFilter
import csv
from django.http import HttpResponse, HttpResponseNotFound
from account import models from account import models
from account.models import User from account.models import User
@ -46,3 +48,22 @@ class UserRUDView(generics.RetrieveUpdateDestroyAPIView):
serializer_class = serializers.BackDetailUserSerializer serializer_class = serializers.BackDetailUserSerializer
permission_classes = (permissions.IsAdminUser,) permission_classes = (permissions.IsAdminUser,)
lookup_field = 'id' lookup_field = 'id'
def get_user_csv(request, user_id):
try:
user = User.objects.get(id=user_id).values_list()
except User.DoesNotExist:
return HttpResponseNotFound("User not found")
#
#
# response = HttpResponse(content_type='text/csv')
# response['Content-Disposition'] = f'attachment; filename="{user}.csv"'
#
# writer = csv.writer(response)
# writer.writerow(['First row', 'Foo', 'Bar', 'Baz'])
# writer.writerow(['Second row', 'A', 'B', 'C', '"Testing"', "Here's a quote"])
#
# return response
return HttpResponseNotFound(user)