Exemplo n.º 1
0
    def test_file_is_checked(self, mock_path):
        with pytest.raises(parse.FLACReadError) as excinfo:
            parse.get_info_from_flacs('/some/path')

        assert mock_path.call_args == call('/some/path/FLAC')
        assert '/some/path/FLAC is not a directory containg FLAC files'\
            == str(excinfo.value)
Exemplo n.º 2
0
    def test_method_works(self, mock_flac, mock_list_dir, mock_path):
        mock_flac.return_value = {
            'artist': ['Test Artist'],
            'album': ['Test Release']
        }

        artist, release = parse.get_info_from_flacs('/some/path')
        assert mock_path.call_args_list[0] == call('/some/path/FLAC')
        assert mock_path.called is True

        assert mock_list_dir.called is True
        assert mock_list_dir.call_args == call('/some/path/FLAC')

        assert mock_flac.called is True
        assert mock_flac.call_args == call('/some/path/FLAC/01 Song.flac')

        assert artist == 'Test Artist'
        assert release == 'Test Release'
Exemplo n.º 3
0
def main():
    if len(sys.argv) < 2:
        print('Usage:')
        print('femur <directory to load audio files from>')
        sys.exit(1)

    if not os.path.isfile('./config.yaml'):
        print('Please set up a config.yaml file in this directory.')
        sys.exit(1)

    config = yaml.load(open('config.yaml', 'r'))

    directory = sys.argv[1]

    if os.path.isdir(directory) is False:
        print('{} is not a directory.'.format(directory))
        sys.exit(1)

    files = os.listdir(directory)

    if any('FLAC' in item for item in files) is True:
        print('Couldn\'t find FLAC directory in {}'.format(directory))
        print('We need this to work some things out.')
        sys.exit(1)

    artist, album = parse.get_info_from_flacs(directory)

    print('Reading %s - %s for release.'.format(artist, album))

    import zipfile
    import tinys3
    from progressbar import ProgressBar
    from PIL import Image

    s3_config = config['s3']
    conn = tinys3.Connection(
        s3_config['access_key'],
        s3_config['secret_key'],
        endpoint='s3-eu-west-1.amazonaws.com'
    )

    artwork_location = os.path.join(directory, 'FLAC', 'Artwork.jpg')
    thumbnail_location = directory

    if not os.path.exists(artwork_location):
        print('Artwork not found, please place the artwork as JPEG at {}'
              .format(artwork_location))
        sys.exit(1)

    im = Image.open(artwork_location)

    print('Uploading artwork.')

    print('Creating thumbnails.')
    for size in [130, 308]:
        i = im.resize((size, size), Image.ANTIALIAS)
        image_file = utils.img_file_name(album, size)
        img_file_location = os.path.join(thumbnail_location, image_file)

        i.save(img_file_location, 'JPEG', quality=100)

        print('Uploading image {} to S3.'.format(image_file))
        f = open(img_file_location, 'rb')

        if size == 130:
            size_name = 'tiny'
        else:
            size_name = 'large'

        # This is for legacy reasons we name all our files on the server
        # using directories to work out what they are.
        conn.upload('/'.join(
            [size_name, album.lower() + '.jpg']),
            f,
            s3_config['artwork_bucket']
        )
        f.close()

    print('Uploading full size artwork.')
    f = open(artwork_location, 'rb')
    conn.upload(
        '/'.join(['huge', album.lower(), '.jpg']),
        f,
        s3_config['artwork_bucket']
    )

    f.close()

    # Our Amazon buckets are in different region so :(
    conn = tinys3.Connection(s3_config['access_key'], s3_config['secret_key'])

    # Permitted directories output by Max
    extensions = ('FLAC', 'MP3', 'Vorbis')

    for d in files:
        full_path = os.path.join(directory, d)

        if not d.startswith('.') and os.path.isdir(full_path) \
           and d.endswith(extensions):
            new_name = utils.directory_name(artist, album, d)
            zip_file = utils.zip_file_name(artist, album, d)

            print('Renaming directory from %s to %s'.format(d, new_name))
            os.rename(full_path, os.path.join(directory, new_name))
            full_path = os.path.join(directory, new_name)

            print('Zipping %s into file named %s'.format(new_name, zip_file))
            zip = zipfile.ZipFile(
                os.path.join(directory, zip_file),
                'w',
                zipfile.ZIP_DEFLATED
            )
            audio_files = os.listdir(full_path)

            pbar = ProgressBar(maxval=len(audio_files)).start()

            i = 0
            for f in audio_files:
                if f.startswith('.'):
                    continue
                zip.write(
                    os.path.join(full_path, f),
                    arcname=os.path.join(new_name, f)
                )
                pbar.update(i + 1)
                i += 1
            zip.close()
            pbar.finish()

            print('Uploading to S3, hold on this could take some time.')
            f = open(os.path.join(directory, zip_file), 'rb')
            conn.upload(zip_file, f, s3_config['releases_bucket'])
            f.close()

    print('All done!')