dss/spa/management/commands/waveforms.py

import os
from django.core.files.storage import FileSystemStorage
from django.core.management.base import NoArgsCommand
from dss import settings

from spa.models.mix import Mix
from core.tasks import create_waveform_task


class Command(NoArgsCommand):
    help = "Generate all outstanding waveforms"

    def _generateWaveform(self, mix):
        #Check for file in mix directory
        in_file = mix.get_absolute_path()
        try:
            if os.path.isfile(in_file):
                create_waveform_task.delay(in_file=in_file, uid=mix.uid)
            else:
                fileName, extension = os.path.splitext(mix.local_file.name)
                in_file=os.path.join(os.path.join(settings.CACHE_ROOT, "mixes"), "%s.%s" % (fileName, extension))
                if os.path.isfile(in_file):
                    create_waveform_task.delay(in_file=in_file, uid=mix.uid)
                else:
                    print "File %s not found" % in_file

        except Exception, ex:
            print "Error generating waveform: %s" % ex.message

    def handle(self, *args, **options):
        print "Scanning for missing waveforms"
        unprocessed = Mix.objects.filter(waveform_generated=False)
        for mix in unprocessed:
            print "Generating waveform for mix %d" % mix.id
            self._generateWaveform(mix)

    def handle_noargs(self, **options):
        print "Scanning for missing waveforms"
        unprocessed = Mix.objects.filter(waveform_generated=False)
        for mix in unprocessed:
            print "Generating waveform for mix %d" % mix.id
            self._generateWaveform(mix.uid)