ojuso-map/apps/files/views.py

57 lines
1.4 KiB
Python
Raw Normal View History

2018-05-01 02:23:36 +00:00
from django.core.exceptions import PermissionDenied
from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import JsonResponse
from django.shortcuts import render
2018-04-23 05:15:33 +00:00
from django.views.generic import FormView, DetailView
from .forms import ImageFileForm, FileForm
from .models import ImageFile, File
2018-05-01 02:23:36 +00:00
class FileUploadView(LoginRequiredMixin, FormView):
model = File
form_class = FileForm
def form_valid(self, form):
2018-05-01 02:23:36 +00:00
self.object = form.save(commit=False)
self.object.user = self.request.user
self.object.save()
2018-04-23 05:15:33 +00:00
return JsonResponse({
'is_valid': True,
'url': self.object.file.url,
2018-04-23 05:15:33 +00:00
'name': self.object.file.name,
'id': self.object.pk
})
def form_invalid(self, form):
return JsonResponse({'is_valid': False, 'errors': form.errors})
2018-04-23 05:15:33 +00:00
class ImageFileUploadView(FileUploadView):
model = ImageFile
form_class = ImageFileForm
2018-05-01 02:23:36 +00:00
class FileDeleteView(LoginRequiredMixin, DetailView):
2018-04-23 05:15:33 +00:00
model = File
def get(self, request, *args, **kwargs):
return self.post(request, *args, **kwargs)
def post(self, request, *args, **kwargs):
self.object = self.get_object()
2018-05-01 02:23:36 +00:00
if request.user != self.object.user:
raise PermissionDenied
2018-04-23 05:15:33 +00:00
self.object.delete()
2018-04-23 05:15:33 +00:00
return JsonResponse({
'success': True
})
2018-09-23 02:25:04 +00:00
class ImageFileDeleteView(FileDeleteView):
model = ImageFile