You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
1.1 KiB
33 lines
1.1 KiB
13 years ago
|
from couchpotato.core.event import addEvent, fireEvent
|
||
|
|
from couchpotato.core.helpers.variable import getExt
|
||
|
|
from couchpotato.core.logger import CPLog
|
||
|
|
from couchpotato.core.plugins.base import Plugin
|
||
|
|
import os
|
||
|
|
|
||
|
|
log = CPLog(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class Trailer(Plugin):
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
addEvent('renamer.after', self.searchSingle)
|
||
|
|
|
||
|
|
def searchSingle(self, group):
|
||
|
|
|
||
|
|
if self.isDisabled() or len(group['files']['trailer']) > 0: return
|
||
|
|
|
||
|
|
trailers = fireEvent('trailer.search', group = group, merge = True)
|
||
|
|
|
||
|
|
for trailer in trailers.get(self.conf('quality'), []):
|
||
|
|
destination = '%s-trailer.%s' % (self.getRootName(group), getExt(trailer))
|
||
|
|
if not os.path.isfile(destination):
|
||
|
|
fireEvent('file.download', url = trailer, dest = destination, urlopen_kwargs = {'headers': {'User-Agent': 'Quicktime'}}, single = True)
|
||
|
|
else:
|
||
|
|
log.debug('Trailer already exists: %s' % destination)
|
||
|
|
|
||
|
|
# Download first and break
|
||
|
|
break
|
||
|
|
|
||
|
|
def getRootName(self, data = {}):
|
||
|
|
return os.path.join(data['destination_dir'], data['filename'])
|