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