def createReleases(self, search_results, media, quality_type):
available_status = fireEvent('status.get', ['available'], single=True)
db = get_session()
found_releases = []
for rel in search_results:
nzb_identifier = md5(rel['url'])
found_releases.append(nzb_identifier)
rls = db.query(Release).filter_by(
identifier=nzb_identifier).first()
if not rls:
rls = Release(
identifier=nzb_identifier,
movie_id=media.get('id'),
#media_id = media.get('id'),
quality_id=quality_type.get('quality_id'),
status_id=available_status.get('id'))
db.add(rls)
else:
[db.delete(old_info) for old_info in rls.info]
rls.last_edit = int(time.time())
db.commit()
for info in rel:
try:
if not isinstance(rel[info],
(str, unicode, int, long, float)):
continue
rls_info = ReleaseInfo(identifier=info,
value=toUnicode(rel[info]))
rls.info.append(rls_info)
except InterfaceError:
log.debug('Couldn\'t add %s to ReleaseInfo: %s',
(info, traceback.format_exc()))
db.commit()
rel['status_id'] = rls.status_id
return found_releases
def single(self, movie):
done_status = fireEvent('status.get', 'done', single=True)
if not movie['profile'] or movie['status_id'] == done_status.get('id'):
log.debug(
'Movie doesn\'t have a profile or already done, assuming in manage tab.'
)
return
db = get_session()
pre_releases = fireEvent('quality.pre_releases', single=True)
release_dates = fireEvent('library.update_release_date',
identifier=movie['library']['identifier'],
merge=True)
available_status = fireEvent('status.get', 'available', single=True)
ignored_status = fireEvent('status.get', 'ignored', single=True)
default_title = getTitle(movie['library'])
if not default_title:
log.error(
'No proper info found for movie, removing it from library to cause it from having more issues.'
)
fireEvent('movie.delete', movie['id'], single=True)
return
fireEvent('notify.frontend',
type='searcher.started.%s' % movie['id'],
data=True,
message='Searching for "%s"' % default_title)
ret = False
for quality_type in movie['profile']['types']:
if not self.couldBeReleased(quality_type['quality']['identifier'],
release_dates, pre_releases):
log.info(
'Too early to search for %s, %s',
(quality_type['quality']['identifier'], default_title))
continue
has_better_quality = 0
# See if better quality is available
for release in movie['releases']:
if release['quality']['order'] <= quality_type['quality'][
'order'] and release['status_id'] not in [
available_status.get('id'),
ignored_status.get('id')
]:
has_better_quality += 1
# Don't search for quality lower then already available.
if has_better_quality is 0:
log.info('Search for %s in %s',
(default_title, quality_type['quality']['label']))
quality = fireEvent(
'quality.single',
identifier=quality_type['quality']['identifier'],
single=True)
results = fireEvent('yarr.search', movie, quality, merge=True)
sorted_results = sorted(results,
key=lambda k: k['score'],
reverse=True)
if len(sorted_results) == 0:
log.debug(
'Nothing found for %s in %s',
(default_title, quality_type['quality']['label']))
download_preference = self.conf('preferred_method')
if download_preference != 'both':
sorted_results = sorted(
sorted_results,
key=lambda k: k['type'],
reverse=(download_preference == 'torrent'))
# Check if movie isn't deleted while searching
if not db.query(Movie).filter_by(id=movie.get('id')).first():
break
# Add them to this movie releases list
for nzb in sorted_results:
rls = db.query(Release).filter_by(
identifier=md5(nzb['url'])).first()
if not rls:
rls = Release(
identifier=md5(nzb['url']),
movie_id=movie.get('id'),
quality_id=quality_type.get('quality_id'),
status_id=available_status.get('id'))
db.add(rls)
db.commit()
else:
[db.delete(info) for info in rls.info]
db.commit()
for info in nzb:
try:
if not isinstance(
nzb[info],
(str, unicode, int, long, float)):
continue
rls_info = ReleaseInfo(identifier=info,
value=toUnicode(nzb[info]))
rls.info.append(rls_info)
db.commit()
except InterfaceError:
log.debug('Couldn\'t add %s to ReleaseInfo: %s',
(info, traceback.format_exc()))
nzb['status_id'] = rls.status_id
for nzb in sorted_results:
if nzb['status_id'] == ignored_status.get('id'):
log.info('Ignored: %s', nzb['name'])
continue
if nzb['score'] <= 0:
log.info('Ignored, score to low: %s', nzb['name'])
continue
downloaded = self.download(data=nzb, movie=movie)
if downloaded is True:
ret = True
break
elif downloaded != 'try_next':
break
else:
log.info(
'Better quality (%s) already available or snatched for %s',
(quality_type['quality']['label'], default_title))
fireEvent('movie.restatus', movie['id'])
break
# Break if CP wants to shut down
if self.shuttingDown() or ret:
break
fireEvent('notify.frontend',
type='searcher.ended.%s' % movie['id'],
data=True)
#db.close()
return ret
def single(self, movie):
available_status = fireEvent('status.get', 'available', single=True)
for type in movie['profile']['types']:
has_better_quality = 0
default_title = movie['library']['titles'][0]['title']
# See if beter quality is available
for release in movie['releases']:
if release['quality']['order'] <= type['quality'][
'order'] and release[
'status_id'] is not available_status.get('id'):
has_better_quality += 1
# Don't search for quality lower then already available.
if has_better_quality is 0:
log.info('Search for %s in %s' %
(default_title, type['quality']['label']))
results = fireEvent('provider.yarr.search',
movie,
type['quality'],
merge=True)
sorted_results = sorted(results,
key=lambda k: k['score'],
reverse=True)
# Add them to this movie releases list
for nzb in sorted_results:
db = get_session()
rls = db.query(Release).filter_by(
identifier=md5(nzb['url'])).first()
if not rls:
rls = Release(identifier=md5(nzb['url']),
movie_id=movie.get('id'),
quality_id=type.get('quality_id'),
status_id=available_status.get('id'))
db.add(rls)
db.commit()
for info in nzb:
try:
if not isinstance(nzb[info],
(str, unicode, int, long)):
continue
rls_info = ReleaseInfo(identifier=info,
value=nzb[info])
rls.info.append(rls_info)
db.commit()
except InterfaceError:
log.debug(
'Couldn\'t add %s to ReleaseInfo: %s' %
(info, traceback.format_exc()))
for nzb in sorted_results:
return self.download(data=nzb, movie=movie)
else:
log.info(
'Better quality (%s) already available or snatched for %s'
% (type['quality']['label'], default_title))
break
# Break if CP wants to shut down
if self.shuttingDown():
break
return False
def single(self, movie, search_protocols=None, manual=False):
# Find out search type
try:
if not search_protocols:
search_protocols = fireEvent('searcher.protocols', single=True)
except SearchSetupError:
return
done_status = fireEvent('status.get', 'done', single=True)
if not movie['profile'] or (movie['status_id'] == done_status.get('id')
and not manual):
log.debug(
'Movie doesn\'t have a profile or already done, assuming in manage tab.'
)
return
db = get_session()
pre_releases = fireEvent('quality.pre_releases', single=True)
release_dates = fireEvent('library.update.movie.release_date',
identifier=movie['library']['identifier'],
merge=True)
available_status, ignored_status, failed_status = fireEvent(
'status.get', ['available', 'ignored', 'failed'], single=True)
found_releases = []
too_early_to_search = []
default_title = getTitle(movie['library'])
if not default_title:
log.error(
'No proper info found for movie, removing it from library to cause it from having more issues.'
)
fireEvent('movie.delete', movie['id'], single=True)
return
fireEvent('notify.frontend',
type='movie.searcher.started.%s' % movie['id'],
data=True,
message='Searching for "%s"' % default_title)
ret = False
for quality_type in movie['profile']['types']:
if not self.conf('always_search') and not self.couldBeReleased(
quality_type['quality']['identifier'] in pre_releases,
release_dates, movie['library']['year']):
too_early_to_search.append(
quality_type['quality']['identifier'])
continue
has_better_quality = 0
# See if better quality is available
for release in movie['releases']:
if release['quality']['order'] <= quality_type['quality'][
'order'] and release['status_id'] not in [
available_status.get('id'),
ignored_status.get('id'),
failed_status.get('id')
]:
has_better_quality += 1
# Don't search for quality lower then already available.
if has_better_quality is 0:
log.info('Search for %s in %s',
(default_title, quality_type['quality']['label']))
quality = fireEvent(
'quality.single',
identifier=quality_type['quality']['identifier'],
single=True)
results = []
for search_protocol in search_protocols:
protocol_results = fireEvent('provider.search.%s.movie' %
search_protocol,
movie,
quality,
merge=True)
if protocol_results:
results += protocol_results
sorted_results = sorted(results,
key=lambda k: k['score'],
reverse=True)
if len(sorted_results) == 0:
log.debug(
'Nothing found for %s in %s',
(default_title, quality_type['quality']['label']))
download_preference = self.conf('preferred_method',
section='searcher')
if download_preference != 'both':
sorted_results = sorted(
sorted_results,
key=lambda k: k['protocol'][:3],
reverse=(download_preference == 'torrent'))
# Check if movie isn't deleted while searching
if not db.query(Media).filter_by(id=movie.get('id')).first():
break
# Add them to this movie releases list
for nzb in sorted_results:
nzb_identifier = md5(nzb['url'])
found_releases.append(nzb_identifier)
rls = db.query(Release).filter_by(
identifier=nzb_identifier).first()
if not rls:
rls = Release(
identifier=nzb_identifier,
movie_id=movie.get('id'),
quality_id=quality_type.get('quality_id'),
status_id=available_status.get('id'))
db.add(rls)
else:
[db.delete(old_info) for old_info in rls.info]
rls.last_edit = int(time.time())
db.commit()
for info in nzb:
try:
if not isinstance(
nzb[info],
(str, unicode, int, long, float)):
continue
rls_info = ReleaseInfo(identifier=info,
value=toUnicode(nzb[info]))
rls.info.append(rls_info)
except InterfaceError:
log.debug('Couldn\'t add %s to ReleaseInfo: %s',
(info, traceback.format_exc()))
db.commit()
nzb['status_id'] = rls.status_id
for nzb in sorted_results:
if not quality_type.get(
'finish', False) and quality_type.get(
'wait_for',
0) > 0 and nzb.get('age') <= quality_type.get(
'wait_for', 0):
log.info('Ignored, waiting %s days: %s',
(quality_type.get('wait_for'), nzb['name']))
continue
if nzb['status_id'] in [
ignored_status.get('id'),
failed_status.get('id')
]:
log.info('Ignored: %s', nzb['name'])
continue
if nzb['score'] <= 0:
log.info('Ignored, score to low: %s', nzb['name'])
continue
downloaded = fireEvent('searcher.download',
data=nzb,
movie=movie,
manual=manual,
single=True)
if downloaded is True:
ret = True
break
elif downloaded != 'try_next':
break
# Remove releases that aren't found anymore
for release in movie.get('releases', []):
if release.get('status_id') == available_status.get(
'id') and release.get(
'identifier') not in found_releases:
fireEvent('release.delete',
release.get('id'),
single=True)
else:
log.info(
'Better quality (%s) already available or snatched for %s',
(quality_type['quality']['label'], default_title))
fireEvent('movie.restatus', movie['id'])
break
# Break if CP wants to shut down
if self.shuttingDown() or ret:
break
if len(too_early_to_search) > 0:
log.info2('Too early to search for %s, %s',
(too_early_to_search, default_title))
fireEvent('notify.frontend',
type='movie.searcher.ended.%s' % movie['id'],
data=True)
return ret
