-from datetime import datetime
-import errno
-import mimetypes
-import os
-import os.path
-import pipes
-import subprocess
-from tempfile import NamedTemporaryFile
-from time import sleep
-
-#from celery.decorators import task
-from celery.task import Task
-from fabric import api
-from mutagen import File
-from mutagen import id3
-
-import mutagen
-
-from archive.constants import status
-from archive.models import Audiobook
-from archive.settings import (BUILD_PATH, COVER_IMAGE,
- UPLOAD_HOST, UPLOAD_USER, UPLOAD_PATH, UPLOAD_CMD, UPLOAD_SUDO)
-from archive.utils import ExistingFile
-
-api.env.host_string = UPLOAD_HOST
-api.env.user = UPLOAD_USER
-
-class AudioFormatTask(Task):
- abstract = True
-
- @classmethod
- def set_status(cls, audiobook, status):
- setattr(audiobook, '%s_status' % cls.ext, status)
- audiobook.save()
-
- @staticmethod
- def encode(in_path, out_path):
- pass
-
- @classmethod
- def set_tags(cls, audiobook, file_name):
- audio = File(file_name)
- for k, v in getattr(audiobook, "%s_tags" % cls.ext)['tags'].items():
- audio[k] = v
- audio.save()
-
- @classmethod
- def save(cls, audiobook, file_name):
- getattr(audiobook, "%s_file" % cls.ext).save(
- "%d.%s" % (audiobook.pk, cls.ext),
- ExistingFile(file_name)
- )
-
- @classmethod
- def published(cls, audiobook):
- setattr(audiobook, "%s_published_tags" % cls.ext,
- getattr(audiobook, "%s_tags" % cls.ext))
- setattr(audiobook, "%s_tags" % cls.ext, None)
- setattr(audiobook, "%s_published" % cls.ext, datetime.now())
- cls.set_status(audiobook, None)
-
- @classmethod
- def put(cls, audiobook):
- tags = getattr(audiobook, "%s_tags" % cls.ext)
- prefix, slug = tags['url'].rstrip('/').rsplit('/', 1)
- name = tags['name']
- path = getattr(audiobook, "%s_file" % cls.ext).path
- api.put(path, UPLOAD_PATH)
- command = UPLOAD_CMD + (u' %s %s %s > output.txt' % (
- pipes.quote(os.path.join(UPLOAD_PATH, os.path.basename(path))),
- pipes.quote(slug),
- pipes.quote(name)
- )).encode('utf-8')
- print command
- if UPLOAD_SUDO:
- api.sudo(command, user=UPLOAD_SUDO, shell=False)
- else:
- api.run(command)
-
- def run(self, aid):
- audiobook = Audiobook.objects.get(id=aid)
- self.set_status(audiobook, status.ENCODING)
-
- try:
- os.makedirs(BUILD_PATH)
- except OSError as e:
- if e.errno == errno.EEXIST:
- pass
- else:
- raise
-
- out_file = NamedTemporaryFile(delete=False, prefix='audiobook-', suffix='.%s' % self.ext, dir=BUILD_PATH)
- out_file.close()
- self.encode(audiobook.source_file.path, out_file.name)
- self.set_status(audiobook, status.TAGGING)
- self.set_tags(audiobook, out_file.name)
- self.save(audiobook, out_file.name)
- self.set_status(audiobook, status.SENDING)
-
- #self.put(audiobook)
-
- self.published(audiobook)
- audiobook.save()
-
-
-class Mp3Task(AudioFormatTask):
- ext = 'mp3'
-
- # these shouldn't be staticmethods
- def id3_text(tag, text):
- return tag(encoding=1, text=text)
- def id3_url(tag, text):
- return tag(url=text)
- def id3_comment(tag, text, lang=u'pol'):
- return tag(encoding=1, lang=lang, desc=u'', text=text)
- def id3_sha1(tag, text, what=u''):
- return tag(owner='http://wolnelektury.pl?%s' % what, data=text)
-
- TAG_MAP = {
- 'album': (id3_text, id3.TALB),
- 'albumartist': (id3_text, id3.TPE2),
- 'artist': (id3_text, id3.TPE1),
- 'conductor': (id3_text, id3.TPE3),
- 'copyright': (id3_text, id3.TCOP),
- 'date': (id3_text, id3.TDRC),
- 'genre': (id3_text, id3.TCON),
- 'language': (id3_text, id3.TLAN),
- 'organization': (id3_text, id3.TPUB),
- 'title': (id3_text, id3.TIT2),
- 'comment': (id3_comment, id3.COMM, 'pol'),
- 'contact': (id3_url, id3.WOAF),
- 'license': (id3_url, id3.WCOP),
- 'flac_sha1': (id3_sha1, id3.PRIV, 'flac_sha1'),
- }
-
- @staticmethod
- def encode(in_path, out_path):
- # 44.1kHz 64kbps mono MP3
- subprocess.check_call(['ffmpeg',
- '-i', in_path,
- '-ar', '44100',
- '-ab', '64k',
- '-ac', '1',
- '-y',
- out_path
- ])
-
- @classmethod
- def set_tags(cls, audiobook, file_name):
- audio = id3.ID3(file_name)
- for k, v in audiobook.mp3_tags['tags'].items():
- factory_tuple = cls.TAG_MAP[k]
- factory, tagtype = factory_tuple[:2]
- audio.add(factory(tagtype, v, *factory_tuple[2:]))
-
- if COVER_IMAGE:
- mime = mimetypes.guess_type(COVER_IMAGE)
- f = open(COVER_IMAGE)
- audio.add(id3.APIC(encoding=0, mime=mime, type=3, desc=u'', data=f.read()))
- f.close()
-
- audio.save()
-
-
-class OggTask(AudioFormatTask):
- ext = 'ogg'
-
- @staticmethod
- def encode(in_path, out_path):
- # 44.1kHz 64kbps mono Ogg Vorbis
- subprocess.check_call(['oggenc',
- in_path,
- '--discard-comments',
- '--resample', '44100',
- '--downmix',
- '-b', '64',
- '-o', out_path
- ])
fnp / audio.git / blobdiff