mirror of
https://github.com/SickGear/SickGear.git
synced 2025-01-05 17:43:37 +00:00
Add file_name to possible names in history lookup post processing.
Add is_basedir check, default to True. Assume multiple shows during pp as default.
This commit is contained in:
parent
76f766fb5d
commit
7518837d4f
5 changed files with 24 additions and 9 deletions
|
@ -109,6 +109,8 @@ def processEpisode(dir_to_process, org_NZB_name=None, status=None):
|
|||
|
||||
params = {}
|
||||
|
||||
params['is_basedir'] = 0
|
||||
|
||||
params['quiet'] = 1
|
||||
|
||||
params['dir'] = dir_to_process
|
||||
|
|
|
@ -43,6 +43,6 @@ class PostProcesser():
|
|||
'(and probably not what you really want to process)' % sickbeard.TV_DOWNLOAD_DIR, logger.ERROR)
|
||||
return
|
||||
|
||||
processTV.processDir(sickbeard.TV_DOWNLOAD_DIR)
|
||||
processTV.processDir(sickbeard.TV_DOWNLOAD_DIR, is_basedir=True)
|
||||
|
||||
self.amActive = False
|
||||
|
|
|
@ -417,7 +417,7 @@ class PostProcessor(object):
|
|||
self.in_history = False
|
||||
|
||||
# if we don't have either of these then there's nothing to use to search the history for anyway
|
||||
if not self.nzb_name and not self.folder_name:
|
||||
if not self.nzb_name and not self.file_name and not self.folder_name:
|
||||
return to_return
|
||||
|
||||
# make a list of possible names to use in the search
|
||||
|
@ -426,6 +426,10 @@ class PostProcessor(object):
|
|||
names.append(self.nzb_name)
|
||||
if '.' in self.nzb_name:
|
||||
names.append(self.nzb_name.rpartition('.')[0])
|
||||
if self.file_name:
|
||||
names.append(self.file_name)
|
||||
if '.' in self.file_name:
|
||||
names.append(self.file_name.rpartition('.')[0])
|
||||
if self.folder_name:
|
||||
names.append(self.folder_name)
|
||||
|
||||
|
|
|
@ -58,12 +58,13 @@ except ImportError:
|
|||
class ProcessTVShow(object):
|
||||
""" Process a TV Show """
|
||||
|
||||
def __init__(self, webhandler=None):
|
||||
def __init__(self, webhandler=None, is_basedir=True):
|
||||
self.files_passed = 0
|
||||
self.files_failed = 0
|
||||
self.fail_detected = False
|
||||
self._output = []
|
||||
self.webhandler = webhandler
|
||||
self.is_basedir = is_basedir
|
||||
|
||||
@property
|
||||
def any_vid_processed(self):
|
||||
|
@ -163,8 +164,10 @@ class ProcessTVShow(object):
|
|||
|
||||
return result
|
||||
|
||||
@staticmethod
|
||||
def check_name(name):
|
||||
def check_name(self, name):
|
||||
if self.is_basedir:
|
||||
return None
|
||||
|
||||
so = None
|
||||
my_db = db.DBConnection()
|
||||
sql_results = my_db.select(
|
||||
|
@ -189,6 +192,9 @@ class ProcessTVShow(object):
|
|||
return (showObj, alt_showObj)[None is showObj and None is not alt_showObj]
|
||||
|
||||
def check_video_filenames(self, path, videofiles):
|
||||
if self.is_basedir:
|
||||
return None
|
||||
|
||||
video_pick = None
|
||||
video_size = 0
|
||||
for cur_video_file in videofiles:
|
||||
|
@ -247,6 +253,9 @@ class ProcessTVShow(object):
|
|||
u'you fill out your completed TV download folder in the PP config.')
|
||||
return self.result
|
||||
|
||||
if dir_name == sickbeard.TV_DOWNLOAD_DIR:
|
||||
self.is_basedir = True
|
||||
|
||||
if None is showObj:
|
||||
if isinstance(nzb_name, basestring):
|
||||
showObj = self.check_name(re.sub(r'\.(nzb|torrent)$', '', nzb_name, flags=re.I))
|
||||
|
@ -939,8 +948,8 @@ class ProcessTVShow(object):
|
|||
|
||||
# backward compatibility prevents the case of this function name from being updated to PEP8
|
||||
def processDir(dir_name, nzb_name=None, process_method=None, force=False, force_replace=None,
|
||||
failed=False, type='auto', cleanup=False, webhandler=None, showObj=None):
|
||||
failed=False, type='auto', cleanup=False, webhandler=None, showObj=None, is_basedir=True):
|
||||
|
||||
# backward compatibility prevents the case of this function name from being updated to PEP8
|
||||
return ProcessTVShow(webhandler).process_dir(
|
||||
return ProcessTVShow(webhandler, is_basedir).process_dir(
|
||||
dir_name, nzb_name, process_method, force, force_replace, failed, type, cleanup, showObj)
|
||||
|
|
|
@ -2558,7 +2558,7 @@ class HomePostProcess(Home):
|
|||
return t.respond()
|
||||
|
||||
def processEpisode(self, dir=None, nzbName=None, jobName=None, quiet=None, process_method=None, force=None,
|
||||
force_replace=None, failed='0', type='auto', stream='0', dupekey=None, **kwargs):
|
||||
force_replace=None, failed='0', type='auto', stream='0', dupekey=None, is_basedir='1', **kwargs):
|
||||
|
||||
if not dir and ('0' == failed or not nzbName):
|
||||
self.redirect('/home/postprocess/')
|
||||
|
@ -2582,7 +2582,7 @@ class HomePostProcess(Home):
|
|||
force_replace=force_replace in ['on', '1'],
|
||||
failed='0' != failed,
|
||||
webhandler=self.send_message if stream != '0' else None,
|
||||
showObj=showObj)
|
||||
showObj=showObj, is_basedir=is_basedir in ['on', '1'])
|
||||
|
||||
if '0' != stream:
|
||||
return
|
||||
|
|
Loading…
Reference in a new issue