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