github/nzbToMedia
1
0
Fork 0
mirror of https://github.com/clinton-hall/nzbToMedia.git synced 2025-07-06 05:01:10 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update identification.py

Browse source 
Return updated dir_name if needed
This commit is contained in:
Matt Park 2023-10-02 15:13:36 -04:00 • committed by Clinton Hall
parent 558970c212
commit 0c564243c2
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
core/utils/identification.py
View file

@ -27,14 +27,14 @@ def find_imdbid(dir_name, input_name, omdb_api_key):
if m: if m:
imdbid = m.group(1) imdbid = m.group(1)
logger.info('Found imdbID [{0}]'.format(imdbid)) logger.info('Found imdbID [{0}]'.format(imdbid))
return imdbid return imdbid, dir_name
if os.path.isdir(dir_name): if os.path.isdir(dir_name):
for file in os.listdir(text_type(dir_name)): for file in os.listdir(text_type(dir_name)):
m = re.search(r'\b(tt\d{7,8})\b', file) m = re.search(r'\b(tt\d{7,8})\b', file)
if m: if m:
imdbid = m.group(1) imdbid = m.group(1)
logger.info('Found imdbID [{0}] via file name'.format(imdbid)) logger.info('Found imdbID [{0}] via file name'.format(imdbid))
return imdbid return imdbid, dir_name
if 'NZBPR__DNZB_MOREINFO' in os.environ: if 'NZBPR__DNZB_MOREINFO' in os.environ:
dnzb_more_info = os.environ.get('NZBPR__DNZB_MOREINFO', '') dnzb_more_info = os.environ.get('NZBPR__DNZB_MOREINFO', '')
if dnzb_more_info != '': if dnzb_more_info != '':
@ -43,7 +43,7 @@ def find_imdbid(dir_name, input_name, omdb_api_key):
if m: if m:
imdbid = m.group(1) imdbid = m.group(1)
logger.info('Found imdbID [{0}] from DNZB-MoreInfo'.format(imdbid)) logger.info('Found imdbID [{0}] from DNZB-MoreInfo'.format(imdbid))
return imdbid return imdbid, dir_name
logger.info('Searching IMDB for imdbID ...') logger.info('Searching IMDB for imdbID ...')
try: try:
guess = guessit.guessit(input_name) guess = guessit.guessit(input_name)
@ -87,10 +87,12 @@ def find_imdbid(dir_name, input_name, omdb_api_key):
if imdbid: if imdbid:
logger.info('Found imdbID [{0}]'.format(imdbid)) logger.info('Found imdbID [{0}]'.format(imdbid))
return imdbid new_dir_name = '{}.cp({})'.format(dir_name, imdbid)
os.rename(dir_name, new_dir_name)
return imdbid, new_dir_name
logger.warning('Unable to find a imdbID for {0}'.format(input_name)) logger.warning('Unable to find a imdbID for {0}'.format(input_name))
return imdbid return imdbid, dir_name
def category_search(input_directory, input_name, input_category, root, categories): def category_search(input_directory, input_name, input_category, root, categories):