69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from collections import OrderedDict
|
|
|
|
from django.contrib.auth import user_logged_in
|
|
|
|
from rest_framework import status
|
|
from rest_framework.authtoken.models import Token
|
|
from rest_framework.response import Response
|
|
from rest_framework.reverse import reverse
|
|
from rest_framework.viewsets import ViewSet
|
|
|
|
from djoser.views import (LoginView, PasswordResetView,
|
|
PasswordResetConfirmView)
|
|
|
|
|
|
class Login(LoginView):
|
|
def action(self, serializer):
|
|
user = serializer.user
|
|
token, _ = Token.objects.get_or_create(user=user)
|
|
user_logged_in.send(sender=user.__class__,
|
|
request=self.request,
|
|
user=user)
|
|
return Response(
|
|
data={'token': token.key, 'id': token.user_id},
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
|
|
class PasswordReset(PasswordResetView):
|
|
"""Overriding to return empty object, for ember-ajax"""
|
|
def action(self, serializer):
|
|
response = super(PasswordReset, self).action(serializer)
|
|
response.data = {}
|
|
return response
|
|
|
|
|
|
class PasswordResetConfirm(PasswordResetConfirmView):
|
|
"""Overriding to return empty object, for ember-ajax"""
|
|
def action(self, serializer):
|
|
response = super(PasswordResetConfirm, self).action(serializer)
|
|
response.data = {}
|
|
return response
|
|
|
|
|
|
class AdminURLs(ViewSet):
|
|
def get_view_name(self):
|
|
return 'Admin URLs List'
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
urls = OrderedDict([
|
|
('projects', [
|
|
['project', 'projects', 'project'],
|
|
['grant', 'projects', 'grant'],
|
|
['grant-report', 'projects', 'grantreport'],
|
|
]),
|
|
|
|
('collections', [
|
|
['collection-type', 'collections_ccdb', 'collectiontype'],
|
|
]),
|
|
])
|
|
|
|
data = OrderedDict()
|
|
for category, group in urls.items():
|
|
paths = []
|
|
for url in group:
|
|
lookup = 'admin:{}_{}_changelist'.format(url[1], url[2])
|
|
path = reverse(lookup, request=request)
|
|
paths.append({'id': url[0], 'url': path})
|
|
data[category] = paths
|
|
return Response(data)
|