sound check

aircox/management/commands/sounds_quality_check.py

@@ -1,118 +1,16 @@
 """
 Analyse and check files using Sox, prints good and bad files.
 """
-import sys
 import logging
-import re
-import subprocess
 from argparse import RawTextHelpFormatter
 
 from django.core.management.base import BaseCommand, CommandError
 
+from aircox.management.sound_stats import SoxStats, SoundStats
+
 logger = logging.getLogger('aircox.commands')
 
 
-class Stats:
-    attributes = [
-        'DC offset', 'Min level', 'Max level',
-        'Pk lev dB', 'RMS lev dB', 'RMS Pk dB',
-        'RMS Tr dB', 'Flat factor', 'Length s',
-    ]
-
-    def __init__(self, path, **kwargs):
-        """
-        If path is given, call analyse with path and kwargs
-        """
-        self.values = {}
-        if path:
-            self.analyse(path, **kwargs)
-
-    def get(self, attr):
-        return self.values.get(attr)
-
-    def parse(self, output):
-        for attr in Stats.attributes:
-            value = re.search(attr + r'\s+(?P<value>\S+)', output)
-            value = value and value.groupdict()
-            if value:
-                try:
-                    value = float(value.get('value'))
-                except ValueError:
-                    value = None
-                self.values[attr] = value
-        self.values['length'] = self.values['Length s']
-
-    def analyse(self, path, at=None, length=None):
-        """
-        If at and length are given use them as excerpt to analyse.
-        """
-        args = ['sox', path, '-n']
-
-        if at is not None and length is not None:
-            args += ['trim', str(at), str(length)]
-
-        args.append('stats')
-
-        p = subprocess.Popen(args, stdout=subprocess.PIPE,
-                             stderr=subprocess.PIPE)
-        # sox outputs to stderr (my god WHYYYY)
-        out_, out = p.communicate()
-        self.parse(str(out, encoding='utf-8'))
-
-
-class SoundStats:
-    path = None             # file path
-    sample_length = 120     # default sample length in seconds
-    stats = None            # list of samples statistics
-    bad = None              # list of bad samples
-    good = None             # list of good samples
-
-    def __init__(self, path, sample_length=None):
-        self.path = path
-        self.sample_length = sample_length if sample_length is not None \
-            else self.sample_length
-
-    def get_file_stats(self):
-        return self.stats and self.stats[0]
-
-    def analyse(self):
-        logger.info('complete file analysis')
-        self.stats = [Stats(self.path)]
-        position = 0
-        length = self.stats[0].get('length')
-
-        if not self.sample_length:
-            return
-
-        logger.info('start samples analysis...')
-        while position < length:
-            stats = Stats(self.path, at=position, length=self.sample_length)
-            self.stats.append(stats)
-            position += self.sample_length
-
-    def check(self, name, min_val, max_val):
-        self.good = [index for index, stats in enumerate(self.stats)
-                     if min_val <= stats.get(name) <= max_val]
-        self.bad = [index for index, stats in enumerate(self.stats)
-                    if index not in self.good]
-        self.resume()
-
-    def resume(self):
-        def view(array): return [
-            'file' if index is 0 else
-            'sample {} (at {} seconds)'.format(
-                index, (index-1) * self.sample_length)
-            for index in array
-        ]
-
-        if self.good:
-            logger.info(self.path + ' -> good: \033[92m%s\033[0m',
-                        ', '.join(view(self.good)))
-        if self.bad:
-            logger.info(self.path + ' -> bad: \033[91m%s\033[0m',
-                        ', '.join(view(self.bad)))
-
-
 class Command (BaseCommand):
     help = __doc__
     sounds = None
@@ -132,7 +30,7 @@ class Command (BaseCommand):
         parser.add_argument(
             '-a', '--attribute', type=str,
             help='attribute name to use to check, that can be:\n' +
-                 ', '.join(['"{}"'.format(attr) for attr in Stats.attributes])
+                 ', '.join(['"{}"'.format(attr) for attr in SoxStats.attributes])
         )
         parser.add_argument(
             '-r', '--range', type=float, nargs=2,
@@ -160,7 +58,7 @@ class Command (BaseCommand):
         self.bad = []
         self.good = []
         for sound in self.sounds:
-            logger.info('analyse ' + sound.file.name)
+            logger.info('analyse ' + sound.path)
             sound.analyse()
             sound.check(attr, minmax[0], minmax[1])
             if sound.bad:
@@ -171,6 +69,6 @@ class Command (BaseCommand):
         # resume
         if options.get('resume'):
             for sound in self.good:
-                logger.info('\033[92m+ %s\033[0m', sound.file.name)
+                logger.info('\033[92m+ %s\033[0m', sound.path)
             for sound in self.bad:
-                logger.info('\033[91m+ %s\033[0m', sound.file.name)
+                logger.info('\033[91m+ %s\033[0m', sound.path)