commit
66398056f1
8 changed files with 127 additions and 27 deletions
@ -0,0 +1,56 @@ |
||||
# -*- coding: utf-8 -*- |
||||
from __future__ import unicode_literals |
||||
|
||||
|
||||
import os |
||||
import subprocess |
||||
|
||||
from .common import PostProcessor |
||||
|
||||
from ..utils import ( |
||||
check_executable, |
||||
compat_urlretrieve, |
||||
encodeFilename, |
||||
PostProcessingError, |
||||
prepend_extension, |
||||
shell_quote |
||||
) |
||||
|
||||
|
||||
class AtomicParsleyPPError(PostProcessingError): |
||||
pass |
||||
|
||||
|
||||
class AtomicParsleyPP(PostProcessor): |
||||
def run(self, info): |
||||
if not check_executable('AtomicParsley', ['-v']): |
||||
raise AtomicParsleyPPError('AtomicParsley was not found. Please install.') |
||||
|
||||
filename = info['filepath'] |
||||
temp_filename = prepend_extension(filename, 'temp') |
||||
temp_thumbnail = prepend_extension(filename, 'thumb') |
||||
|
||||
if not info.get('thumbnail'): |
||||
raise AtomicParsleyPPError('Thumbnail was not found. Nothing to do.') |
||||
|
||||
compat_urlretrieve(info['thumbnail'], temp_thumbnail) |
||||
|
||||
cmd = ['AtomicParsley', filename, '--artwork', temp_thumbnail, '-o', temp_filename] |
||||
|
||||
self._downloader.to_screen('[atomicparsley] Adding thumbnail to "%s"' % filename) |
||||
|
||||
if self._downloader.params.get('verbose', False): |
||||
self._downloader.to_screen('[debug] AtomicParsley command line: %s' % shell_quote(cmd)) |
||||
|
||||
p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE) |
||||
stdout, stderr = p.communicate() |
||||
|
||||
if p.returncode != 0: |
||||
msg = stderr.decode('utf-8', 'replace').strip() |
||||
raise AtomicParsleyPPError(msg) |
||||
|
||||
os.remove(encodeFilename(filename)) |
||||
os.remove(encodeFilename(temp_thumbnail)) |
||||
os.rename(encodeFilename(temp_filename), encodeFilename(filename)) |
||||
|
||||
return True, info |
Loading…
Reference in new issue