59239d25fb86a2cf517c1e3fe24fd79dd998c3c6
[audio.git] / apps / archive / views.py
1 # Create your views here.
2
3 from datetime import datetime
4 import os
5 import os.path
6
7 from archive import settings
8 from django.contrib.auth import logout
9 from django.contrib.auth.decorators import login_required, permission_required
10 from django.core.urlresolvers import reverse
11 from django.db.models import Q, Max
12 from django.http import Http404
13 from django.shortcuts import render, redirect, get_object_or_404
14 from django.views.decorators.http import require_POST
15
16 import mutagen
17
18 from archive.constants import status
19 from archive import models
20 from archive.forms import AudiobookForm
21 from archive import tasks
22 from archive.utils import all_files
23
24
25 @login_required
26 def list_new(request):
27     division = 'new'
28
29     path = settings.NEW_PATH
30     objects = sorted(all_files(path))
31     return render(request, "archive/list_new.html", locals())
32
33
34 @permission_required('archive.change_audiobook')
35 def file_new(request, filename):
36     division = 'new'
37
38     filepath = filename.encode('utf-8')
39     root_filepath = os.path.join(settings.NEW_PATH, filename.encode('utf-8'))
40     if request.POST:
41         form = AudiobookForm(request.POST)
42         if form.is_valid():
43             try:
44                 form.save(path=filepath)
45             except IOError:
46                 raise Http404
47             return redirect(list_new)
48
49     try:
50         tags = mutagen.File(root_filepath)
51     except IOError:
52         raise Http404
53     d = {}
54     if tags:
55         for tag in tags:
56             value = tags[tag]
57             if isinstance(value, list):
58                 d[tag] = value[0]
59             else:
60                 d[tag] = value
61             if tag == 'project':
62                 try:
63                     d[tag] = models.Project.objects.get(name=d[tag]).pk
64                 except models.Project.DoesNotExist:
65                     d[tag] = None
66
67     if not request.POST:
68         form = AudiobookForm(d)
69     return render(request, "archive/file_new.html", locals())
70
71
72 @require_POST
73 @permission_required('archive.change_audiobook')
74 def move_to_archive(request, filename):
75     """ move a new file to the unmanaged files dir """
76
77     filename_str = filename.encode('utf-8')
78     old_path = os.path.join(settings.NEW_PATH, filename_str)
79     new_path = os.path.join(settings.UNMANAGED_PATH, filename_str)
80     new_dir = os.path.split(new_path)[0]
81     if not os.path.isdir(new_dir):
82         os.makedirs(new_dir)
83
84     if not os.path.isfile(old_path):
85         raise Http404
86
87     try:
88         os.link(old_path, new_path)
89         os.unlink(old_path)
90     except OSError:
91         # destination file exists, don't overwrite it
92         # TODO: this should probably be more informative
93         return redirect(file_new, filename)
94
95     return redirect(list_new)
96
97
98 @require_POST
99 @permission_required('archive.change_audiobook')
100 def remove_to_archive(request, aid):
101     """ move a managed file to the unmanaged files dir """
102
103     audiobook = get_object_or_404(models.Audiobook, id=aid)
104     old_path = audiobook.source_file.path
105     new_path = os.path.join(settings.UNMANAGED_PATH,
106         str(audiobook.source_file)[len(settings.FILES_SAVE_PATH):].lstrip('/'))
107     new_dir = os.path.split(new_path)[0]
108     if not os.path.isdir(new_dir):
109         os.makedirs(new_dir)
110
111     if not os.path.isfile(old_path):
112         raise Http404
113
114     try:
115         os.link(old_path, new_path)
116     except OSError:
117         # destination file exists, don't overwrite it
118         # TODO: this should probably be more informative
119         return redirect(file_new, filename)
120     else:
121         os.unlink(old_path)
122         audiobook.delete()
123
124     return redirect(list_unmanaged)
125
126 @require_POST
127 @permission_required('archive.change_audiobook')
128 def move_to_new(request, filename):
129     """ move a unmanaged file to new files dir """
130
131     filename_str = filename.encode('utf-8')
132     old_path = os.path.join(settings.UNMANAGED_PATH, filename_str)
133     new_path = os.path.join(settings.NEW_PATH, filename_str)
134     new_dir = os.path.split(new_path)[0]
135     if not os.path.isdir(new_dir):
136         os.makedirs(new_dir)
137
138     if not os.path.isfile(old_path):
139         raise Http404
140
141     try:
142         os.link(old_path, new_path)
143         os.unlink(old_path)
144     except OSError:
145         # destination file exists, don't overwrite it
146         # TODO: this should probably be more informative
147         return redirect(reverse(file_unmanaged, args=[filename]) + "?exists=1")
148
149     return redirect(list_unmanaged)
150
151
152 @require_POST
153 @permission_required('archive.change_audiobook')
154 def publish(request, aid):
155     """ mark file for publishing """
156     audiobook = get_object_or_404(models.Audiobook, id=aid)
157     tags = {
158         'name': audiobook.title,
159         'url': audiobook.url,
160         'tags': audiobook.new_publish_tags(),
161         }
162     audiobook.mp3_tags = tags
163     audiobook.ogg_tags = tags
164     audiobook.mp3_status = audiobook.ogg_status = status.WAITING
165     audiobook.save()
166     # isn't there a race here?
167     audiobook.mp3_task = tasks.Mp3Task.delay(aid).task_id
168     audiobook.ogg_task = tasks.OggTask.delay(aid).task_id
169     audiobook.save()
170
171     return redirect(file_managed, aid)
172
173
174 @require_POST
175 @permission_required('archive.change_audiobook')
176 def cancel_publishing(request, aid):
177     """ cancel scheduled publishing """
178     audiobook = get_object_or_404(models.Audiobook, id=aid)
179     # TODO: cancel tasks
180     audiobook.mp3_status = None
181     audiobook.ogg_status = None
182     audiobook.save()
183     return redirect(file_managed, aid)
184
185
186 @login_required
187 def list_unpublished(request):
188     division = 'unpublished'
189
190     objects = models.Audiobook.objects.filter(Q(mp3_published=None) | Q(ogg_published=None))
191     return render(request, "archive/list_unpublished.html", locals())
192
193
194 @login_required
195 def list_publishing(request):
196     division = 'publishing'
197
198     objects = models.Audiobook.objects.exclude(mp3_status=None, ogg_status=None)
199     objects_by_status = {}
200     for o in objects:
201         if o.mp3_status:
202             k = o.mp3_status, o.get_mp3_status_display()
203             objects_by_status.setdefault(k, []).append(o)
204         if o.ogg_status and o.ogg_status != o.mp3_status:
205             k = o.ogg_status, o.get_ogg_status_display()
206             objects_by_status.setdefault(k, []).append(o)
207     status_objects = sorted(objects_by_status.items(), reverse=True)
208
209     return render(request, "archive/list_publishing.html", locals())
210
211
212 @login_required
213 def list_published(request):
214     division = 'published'
215
216     objects = models.Audiobook.objects.exclude(Q(mp3_published=None) | Q(ogg_published=None))
217     return render(request, "archive/list_published.html", locals())
218
219
220 @permission_required('archive.change_audiobook')
221 def file_managed(request, id):
222     audiobook = get_object_or_404(models.Audiobook, id=id)
223
224     if request.POST:
225         form = AudiobookForm(request.POST, instance=audiobook)
226         if form.is_valid():
227             try:
228                 form.save()
229             except IOError:
230                 raise Http404
231
232     division = 'published' if audiobook.published() else 'unpublished'
233     path = audiobook.source_file.path[len(settings.FILES_PATH):].lstrip('/')
234
235     # for tags update
236     tags = mutagen.File(audiobook.source_file.path)
237     if not tags:
238         tags = {}
239     form = AudiobookForm(instance=audiobook)
240
241     return render(request, "archive/file_managed.html", locals())
242
243
244 @login_required
245 def list_unmanaged(request):
246     division = 'unmanaged'
247
248     objects = sorted(all_files(settings.UNMANAGED_PATH))
249     return render(request, "archive/list_unmanaged.html", locals())
250
251
252 @login_required
253 def file_unmanaged(request, filename):
254     division = 'unmanaged'
255
256     tags = mutagen.File(os.path.join(settings.UNMANAGED_PATH, filename.encode('utf-8')))
257     if not tags:
258         tags = {}
259     
260     err_exists = request.GET.get('exists')
261     return render(request, "archive/file_unmanaged.html", locals())