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

Merge pull request #1982 from MattPark/last-resort-movie-id

Browse source 
Last resort movie identification
This commit is contained in:
Clinton Hall 2023-12-16 09:32:35 +13:00 • committed by GitHub
commit b802aca7e1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

1
core/__init__.py
View file

@ -1047,6 +1047,7 @@ def initialize(section=None):
configure_utility_locations()
configure_sections(section)
configure_torrent_class()
configure_groups()
__INITIALIZED__ = True

2
core/auto_process/movies.py
View file

@ -66,7 +66,7 @@ def process(section, dir_name, input_name=None, status=0, client_agent='manual',
else:
extract = int(cfg.get('extract', 0))
imdbid = find_imdbid(dir_name, input_name, omdbapikey)
imdbid, dir_name = find_imdbid(dir_name, input_name, omdbapikey)
if section == 'CouchPotato':
base_url = '{0}{1}:{2}{3}/api/{4}/'.format(protocol, host, port, web_root, apikey)
if section == 'Radarr':

12
core/utils/identification.py
View file

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