cdc416eda638a1ece3c89c2d5c1f3af257c57ee3
[audio.git] / apps / archive / forms.py
1 from datetime import datetime
2 import os
3 import os.path
4
5 from django import forms
6 from django.utils.translation import ugettext_lazy as _
7 import mutagen
8
9 from archive.models import Audiobook
10 from archive.settings import FILES_PATH, NEW_PATH
11 from archive.utils import ExistingFile, sha1_file
12
13 class AudiobookForm(forms.ModelForm):
14     class Meta:
15         model = Audiobook
16
17     def save(self, commit=True, path=None):
18         """ Performs normal save, with given file as an source audiobook.
19
20             `path' is relative to NEW_PATH.
21         """
22         m = super(AudiobookForm, self).save(commit=False)
23         m.modified = datetime.now()
24
25         if path:
26             # adding a new audiobook
27             if not os.path.isdir(FILES_PATH):
28                 os.makedirs(FILES_PATH)
29             # save the file in model
30
31             abs_path = os.path.join(NEW_PATH, path)
32             m.source_file.save(
33                 path,
34                 ExistingFile(abs_path))
35
36             f = open(m.source_file.path)
37             m.source_sha1 = sha1_file(f)
38             f.close()
39
40         if commit:
41             m.save()