Add word boundary to imdb match. #1639

Prevents matching (and truncating) longer ids.
Thanks @currently-off-my-rocker
This commit is contained in:
Clinton Hall 2019-07-23 14:24:31 +12:00 committed by GitHub
commit d7eab5d2d3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -23,14 +23,14 @@ def find_imdbid(dir_name, input_name, omdb_api_key):
# find imdbid in dirName # find imdbid in dirName
logger.info('Searching folder and file names for imdbID ...') logger.info('Searching folder and file names for imdbID ...')
m = re.search(r'(tt\d{7,8})', dir_name + input_name) m = re.search(r'\b(tt\d{7,8})\b', dir_name + input_name)
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
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'(tt\d{7,8})', 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))