views.py
4.23 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
from django.shortcuts import render, get_object_or_404, redirect
from django.conf import settings
from django.views import generic
from django.contrib import messages
from django.core.urlresolvers import reverse_lazy
from django.contrib.auth.mixins import LoginRequiredMixin
from rolepermissions.mixins import HasRoleMixin
from rolepermissions.verifications import has_role
from .forms import FileForm, UpdateFileForm
from .models import TopicFile
from .utils import mime_type_to_material_icons
from courses.models import Topic
from core.models import MimeType
# Create your views here.
class CreateFile(LoginRequiredMixin, HasRoleMixin, generic.edit.CreateView):
allowed_roles = ['professor', 'system_admin']
login_url = reverse_lazy("core:home")
redirect_field_name = 'next'
model = TopicFile
template_name = 'files/create_file.html'
form_class = FileForm
success_url = reverse_lazy('course:file:render_file')
def form_invalid(self, form, **kwargs):
context = super(CreateFile, self).form_invalid(form)
context.status_code = 400
return context
def form_valid(self, form):
self.object = form.save(commit = False)
topic = get_object_or_404(Topic, slug = self.kwargs.get('slug'))
self.object.topic = topic
# Set MimeType
file = self.request.FILES['file_url']
try:
if file:
file_type = file.content_type
# Check if exist a mimetype in database
try:
self.object.file_type = MimeType.objects.get(typ = file_type)
# Create if not
except:
mtype = MimeType.objects.create(
typ = file_type,
icon = mime_type_to_material_icons[file_type]
)
mtype.save()
self.object.file_type = mtype
except:
print('File not uploaded')
# self.object.file_type = MimeType.objects.get(id = 1)
self.object.save()
return self.render_to_response(self.get_context_data(form = form), status = 200)
def get_context_data(self, **kwargs):
context = super(CreateFile, self).get_context_data(**kwargs)
topic = get_object_or_404(Topic, slug = self.kwargs.get('slug'))
context["topic"] = topic
context['subject'] = topic.subject
context['subjects'] = topic.subject.course.subjects.all()
try:
context['latest_file'] = TopicFile.objects.latest('id')
except:
pass
return context
def get_success_url(self):
self.success_url = reverse('course:file:render_file', args = (self.object.id, ))
return self.success_url
def render_file(request, id):
template_name = 'files/render_file.html'
context = {
'file': get_object_or_404(TopicFile, id = id)
}
return render(request, template_name, context)
class UpdateFile(LoginRequiredMixin, HasRoleMixin, generic.UpdateView):
allowed_roles = ['professor', 'system_admin']
login_url = reverse_lazy("core:home")
redirect_field_name = 'next'
model = TopicFile
template_name = 'files/update_file.html'
form_class = UpdateFileForm
context_object_name = 'file'
success_url = reverse_lazy('course:file:render_file')
def form_invalid(self, form, **kwargs):
context = super(UpdateFile, self).form_invalid(form)
context.status_code = 400
return context
def get_object(self, queryset=None):
return get_object_or_404(TopicFile, slug = self.kwargs.get('slug'))
def get_success_url(self):
self.success_url = reverse_lazy('course:file:render_file', args = (self.object.id, ))
return self.success_url
class DeleteFile(LoginRequiredMixin, HasRoleMixin, generic.DeleteView):
allowed_roles = ['professor', 'system_admin']
login_url = reverse_lazy("core:home")
redirect_field_name = 'next'
model = TopicFile
template_name = 'files/delete_file.html'
def dispatch(self, *args, **kwargs):
file = get_object_or_404(TopicFile, slug = self.kwargs.get('slug'))
if(not (file.topic.owner == self.request.user) and not(has_role(self.request.user, 'system_admin')) ):
return self.handle_no_permission()
return super(DeleteFile, self).dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = super(DeleteFile, self).get_context_data(**kwargs)
context['course'] = self.object.topic.subject.course
context['subject'] = self.object.topic.subject
context['file'] = self.object
context["topic"] = self.object.topic
return context
def get_success_url(self):
return reverse_lazy('course:view_topic', kwargs={'slug' : self.object.topic.slug})