Home
Diffusion Blender My Data (master) Browse

blender-my-data/mydata_benchmarks/views.py
71c09be2a6abmaster

Last Change
Raw File

views.py
View Options

import logging
import requests
import json
import uuid
import urllib.parse
from django.conf import settings
from django.contrib.auth.decorators import login_required
from django.core.paginator import Paginator
from django.http import JsonResponse, HttpResponseBadRequest
from django.shortcuts import render, redirect
from django.views.decorators.http import require_http_methods
from django.views.generic import View
from django.utils.decorators import method_decorator
from mydata_benchmarks.models import Benchmark, ClientToken
from mydata_benchmarks.decorators import client_token_required
log = logging.getLogger(__name__)
@login_required
def index(request):
samples_list = request.user.benchmarks.all().order_by('-date_created')
paginator = Paginator(samples_list, 25)
page = request.GET.get('page')
samples = paginator.get_page(page)
return render(request, 'index.html', {
'user': request.user,
'samples': samples,
'show_url': urllib.parse.urljoin(settings.BLENDER_OPENDATA['BASE_URL'], 'api/benchmarks'),
})
# @login_required
# def benchmark_delete(request, id):
# benchmark = Benchmark.objects.get(pk=id)
#
# if benchmark is None:
# return redirect('index')
#
# # FIXME use queryset from user
# if not benchmark.belongs_to_blender_id(request.user.blender_id):
# raise HttpResponseForbidden()
#
# try:
# opendata_delete_benchmark(benchmark.manage_id)
# benchmark.delete()
#
# except ServiceUnavailable:
# return HttpResponseServerError('Service Unavailable, please try again later')
#
# return redirect('index')
class GenerateClientTokenView(View):
"""Create a Client auth token for the logged-in user."""
log = logging.getLogger(f'{__name__}.GenerateClientTokenView')
@method_decorator(login_required)
def post(self, request):
self.log.info('Generating client token for user %s', request.user)
# TODO(fsiddi): check for return url (redirect to localhost started by Benchmark Client)
client_token = ClientToken.objects.create(
user=request.user,
token=uuid.uuid4().hex, # TODO sign the token, and maybe embed user id into it?
)
return JsonResponse({'message': 'All was well.', 'token': client_token.token})
@method_decorator(login_required)
def get(self, request):
"""Show a page that performs a POST request to our own URL."""
auth_callback = request.GET.get('auth_callback')
if not auth_callback:
return HttpResponseBadRequest('missing auth_callback')
parsed_url = urllib.parse.urlparse(auth_callback)
if (
parsed_url.scheme not in {'http', 'https'} or
parsed_url.password or
parsed_url.username or
parsed_url.fragment or
parsed_url.query or
parsed_url.hostname != 'localhost'):
return HttpResponseBadRequest('malformed auth_callback')
return render(request, 'token_generate.html', {
'post_to': request.path,
'success_url': auth_callback,
})
@require_http_methods(["POST"])
@client_token_required
def benchmark_submit_view(request):
# TODO change the order of submission. First we should store the data on mydata, then
# we index it (asyncronously?).
headers = {"Authorization": "Token " + settings.BLENDER_OPENDATA['TOKEN']}
try:
benchmark_data = json.loads(request.body)
except ValueError:
return JsonResponse({'message': 'The submitted data is not valid.'}, 422)
try:
url = urllib.parse.urljoin(settings.BLENDER_OPENDATA['BASE_URL'], 'api/benchmarks/')
r = requests.post(url, json=benchmark_data, verify=True, headers=headers)
except requests.exceptions.ConnectionError:
raise JsonResponse({'message': 'Service temporarily unavailable, try again later.'}, 503)
if r.status_code != 201:
# TODO provide meaningful error message
raise JsonResponse({'message': 'The submitted data is not valid'}, r.status_code)
response = r.json()
log.info('Indexed benchmark %s' % response['benchmark_id'])
# Create local Benchmark sample
Benchmark.objects.create(
user=request.user,
benchmark_id=uuid.UUID(response['benchmark_id']),
manage_id=uuid.UUID(response['manage_id']),
data_raw=benchmark_data,
)
log.info('Stored benchmark %s' % response['benchmark_id'])
return JsonResponse({'message': 'All was well.'})
#
# @require_http_methods(["POST"])
# @login_required
# def benchmarks_delete_view(request):
# """Given a list of Benchmark ids, delete them."""
# try:
# url_benchmark = urllib.parse.urljoin('api/benchmarks/', id)
# url = urllib.parse.urljoin(settings.BLENDER_OPENDATA['BASE_URL'], url_benchmark)
# r = requests.delete(url)
#
# except requests.exceptions.ConnectionError:
# raise exceptions.ServiceUnavailable
#
# return JsonResponse({'message': 'All was well.'})
@require_http_methods(["POST"])
@login_required
def benchmark_delete_bulk_view(request):
"""Get a list of sample ids and remove them.
First the samples will be removed from the Open Data platform, and on
success they will also be removed from My Data.
"""
# Lists from jQuery are provided with a [] suffix)
samples = request.POST.getlist('samples[]')
url_base = urllib.parse.urljoin(settings.BLENDER_OPENDATA['BASE_URL'], 'api/benchmarks/')
if not samples:
return JsonResponse({'message': 'No samples provided for deletion'}, status=400)
try:
samples = [int(s) for s in samples]
except ValueError:
return JsonResponse({'message': 'Some samples ids are not valid'}, status=400)
requests_session = requests.Session()
for sample_id in samples:
try:
benchmark = request.user.benchmarks.get(pk=sample_id)
except Benchmark.DoesNotExist:
return JsonResponse({'message': 'Some samples ids are not valid'}, status=400)
url_benchmark = urllib.parse.urljoin(url_base, str(benchmark.manage_id))
try:
resp = requests_session.delete(url_benchmark)
except requests.exceptions.ConnectionError:
return JsonResponse({'message': 'Service unavailable'}, status=400)
if resp.status_code != 204:
return JsonResponse({'message': 'Error deleting the benchmark'}, status=400)
benchmark.delete()
log.debug('Deleted sample %i' % sample_id)
return JsonResponse({'message': 'Samples deleted'}, status=200)