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             'youtube_id', 'youtube_queued'
  20         ]
  21
  22     def save(self, commit=True, path=None):
  23         """ Performs normal save, with given file as an source audiobook.
  24
  25             `path' is relative to NEW_PATH.
  26         """
  27         m = super(AudiobookForm, self).save(commit=False)
  28         m.modified = datetime.now()
  29
  30         if path:
  31             # adding a new audiobook
  32             if not os.path.isdir(FILES_PATH):
  33                 os.makedirs(FILES_PATH)
  34             # save the file in model
  35
  36             abs_path = os.path.join(NEW_PATH, path)
  37             m.duration = get_duration(abs_path)
  38             m.source_file.save(
  39                 path,
  40                 ExistingFile(abs_path))
  41
  42 #            f = open(force_bytes(m.source_file.path))
  43 #            m.source_sha1 = sha1_file(f)
  44 #            f.close()
  45
  46         if commit:
  47             m.save()