def movie_convert(source_file):
"""Convert the MKV file to valid M4V file"""
import os
import disklibrary
print 'Checking file %s' % source_file
file_check = disklibrary.file_check(source_file, 'mkv')
if file_check is None:
print 'Not a valid MKV file %s' % source_file
return 1
os.nice(10)
source_link = disklibrary.file_split(source_file)
print 'Processing file %s' % source_link.file_name
final_link = disklibrary.file_path(source_link.file_path,
source_link.file_title + '.m4v')
if disklibrary.file_check(final_link, 'm4v') is not None:
disklibrary.file_delete(final_link)
final_link = disklibrary.file_split(final_link)
print 'Creating new file %s' % final_link.file_name
error = create_output_file(source_link.full_path, final_link.full_path)
if error is not None:
print error
return 2
print 'Removing source file'
disklibrary.file_delete(source_link.full_path)
return 0
def validate_file(source_file):
"""Validate MKV file"""
import os
import disklibrary
print 'Checking file %s' % source_file
file_check = disklibrary.file_check(source_file, 'mkv')
if file_check is None:
print 'Not a valid MKV file %s' % source_file
return 1
os.nice(10)
print 'Loading MKV file %s' % source_file
source_file = disklibrary.file_split(source_file)
print 'Processing file %s' % source_file.file_name
source_mkv_file = read_mkv_file(source_file.full_path)
if source_mkv_file.error is not None:
print source_mkv_file.error
return 2
source_mkv_file = source_mkv_file.mkv_file
print 'Video Tracks: ' + str(len(source_mkv_file.video_tracks))
print 'Audio Tracks: ' + str(len(source_mkv_file.audio_tracks))
print 'Subtitle Tracks: ' + str(len(source_mkv_file.subtitle_tracks))
print 'Checking video tracks'
use_video_track = find_valid_video_track(source_mkv_file)
print 'Checking audio tracks'
use_audio_track = find_valid_audio_track(source_mkv_file)
if use_video_track is None:
print 'Selected MKV does not contain a valid video track'
return 2
if use_audio_track is None:
if use_audio_track is None:
print 'Selected MKV does not contain a valid audio track'
return 2
convert_file = len(source_mkv_file.video_tracks) != 1
convert_file = convert_file or len(source_mkv_file.audio_tracks) != 1
convert_file = convert_file or len(source_mkv_file.subtitle_tracks) > 0
if not convert_file:
print 'File already in correct format'
return 0
final_link = source_file
source_link = disklibrary.file_path(source_file.file_path,
source_file.file_title + '.original')
source_link = disklibrary.file_split(source_link)
source_moved = disklibrary.file_rename(final_link.full_path,
source_link.full_path)
if not source_moved:
print 'Could not create backup of %s' % final_link.file_name
return 2
check_file = disklibrary.file_check(final_link.full_path, 'mkv')
if check_file is not None:
disklibrary.file_delete(final_link)
print 'Creating new file %s' % final_link.file_name
error = create_output_file(source_link.full_path, final_link.full_path,
final_link.file_title, use_video_track,
use_audio_track)
if error is not None:
print error
return 2
disklibrary.file_delete(source_link.full_path)
print 'Created new file %s' % final_link.file_name
return 0
def main():
"""Missing movie info finder"""
import os
import sys
import argparse
import disklibrary
print 'Starting Movie Info Missing Finder'
parser = argparse.ArgumentParser(
description='Checks for missing movie information')
parser.add_argument('folder', metavar='folder',
type=str, help='Folder to be checked')
args = parser.parse_args()
source_folder = args.folder
if source_folder is None or not os.path.isdir(source_folder):
print 'Folder %s not found or valid' % source_folder
sys.exit(2)
content_list = os.listdir(source_folder)
content_count = len(content_list)
if content_list is None or content_count == 0:
print 'No files or folders found'
sys.exit(2)
os.nice(10)
item_count = 0
for item in content_list:
item_folder = os.path.join(source_folder, item)
if not os.path.isdir(item_folder):
continue
try:
m4v_file = disklibrary.file_first(item_folder, 'm4v')
if m4v_file is None:
print '%s Failed' % item_folder
continue
m4v_file = disklibrary.file_split(m4v_file)
nfo_found = False
nfo_file = disklibrary.file_first(item_folder, 'nfo')
if nfo_file:
nfo_file = disklibrary.file_split(nfo_file)
nfo_found = m4v_file.file_title == nfo_file.file_title
cover_found = False
cover_file = '%s/%s-cover.jpg' % (m4v_file.file_path,
m4v_file.file_title)
cover_file = disklibrary.file_check(cover_file, 'jpg')
if cover_file:
cover_found = True
poster_found = False
poster_file = '%s/%s-fanart.jpg' % (
m4v_file.file_path, m4v_file.file_title)
poster_file = disklibrary.file_check(poster_file, 'jpg')
if poster_file:
poster_found = True
if not nfo_found or not cover_file or not poster_file:
if len(m4v_file.file_title) > 70:
print '{0:70}'.format(m4v_file.file_title[:70]),
else:
print '{0:70}'.format(m4v_file.file_title),
print('NFO:'),
if nfo_found:
print '{0:15}'.format('Found'),
else:
print '{0:15}'.format('Not found'),
print('\tCover:'),
if cover_found:
print '{0:15}'.format('Found'),
else:
print '{0:15}'.format('Not found'),
print('\tPoster:'),
if poster_found:
print '{0:15}'.format('Found')
else:
print '{0:15}'.format('Not found')
item_count += 1
except Exception as exception:
print '%s error: %s' % (item_folder, exception.message)
break
print 'Missing NFO files: %d' % item_count
sys.exit(0)
def movie_update(source_file):
"""Update movie with tag information"""
import disklibrary
import movielibrary
import movielistslibrary
file_path = disklibrary.file_check(source_file, 'm4v')
if file_path is None:
return 2
file_parts = disklibrary.file_split(file_path)
if file_parts is None:
print 'Unable to parse file path'
return 2
movie_title = file_parts.file_title
movie_year = movielibrary.movie_parse_year(file_parts.file_title)
movie_title = movie_title.replace('(' + str(movie_year) + ')', '').strip()
print 'Loading movie %s from %s' % (movie_title, movie_year)
movie_data = movielibrary.movie_load_data(
MOVIE_DB_API, movie_title, movie_year)
if movie_data.error is not None:
print movie_data.error
return 2
if movie_data.data is None:
print 'Movie %s from %s not found' % (movie_title, movie_year)
return 2
cover_link = None
if 'poster_path' in movie_data.data and movie_data.data['poster_path'] is not None:
poster_path = movielibrary.movie_poster(
file_parts.file_path, movie_data.data['poster_path'])
cover_link = disklibrary.file_path(
file_parts.file_path, file_parts.file_title + '-cover.jpg')
print 'Downloading movie cover'
if not disklibrary.file_rename(poster_path, cover_link):
print 'Error creating movie cover'
print 'Retrieving additional information for %s' % movie_title
sort_name = movielibrary.movie_sort_name_builder(
MOVIE_DB_API, movie_data.data)
genre = movielibrary.movie_imdb_genre(movie_data.data['imdb_id'])
if genre is None:
genre = movie_data.data['genres'][0]['name']
studio = movielibrary.movie_imdb_studio(movie_data.data['imdb_id'])
if studio is None:
company_count = len(movie_data.data['production_companies'])
if 'production_companies' in movie_data.data and company_count > 0:
studio = movie_data.data['production_companies'][0]['name']
if sort_name.error is not None:
print sort_name.error
rating = movielibrary.movie_rating(
movie_data.data['releases']['countries'], True)
print 'Generating cast and crew info for %s' % movie_title
credit_xml = movielistslibrary.build_plist_data(movie_data.data, studio)
print 'Updating movie tags for %s' % movie_title
result = update_file_tag(file_parts.full_path, movie_data.data, sort_name.sort_name,
genre, rating, credit_xml, cover_link)
if result is not None:
print result
return 1
return 0
def generate_nfo(source_file, title, year):
"""Generate NFO content and data"""
import disklibrary
import movielibrary
file_path = disklibrary.file_check(source_file, 'm4v')
if file_path is None:
file_path = disklibrary.file_check(source_file, 'mkv')
if file_path is None:
print 'File %s not found or incorrect type' % source_file
return 2
file_parts = disklibrary.file_split(file_path)
if file_parts is None:
print 'Unable to parse file path'
return 2
movie_title = file_parts.file_title
movie_year = movielibrary.movie_parse_year(file_parts.file_title)
if title is not None:
movie_title = title
if year is not None:
movie_year = year
movie_title = movie_title.replace('(' + str(movie_year) + ')', '').strip()
print 'Loading movie %s from %s' % (movie_title, movie_year)
movie_data = movielibrary.movie_load_data(
MOVIE_DB_API, movie_title, movie_year)
if movie_data.error is not None:
print movie_data.error
return 2
if movie_data.data is None:
print 'Movie %s from %s not found' % (movie_title, movie_year)
return 2
movie_sort_name = movielibrary.movie_sort_name_builder(
MOVIE_DB_API, movie_data.data)
if movie_sort_name.error is not None:
print movie_sort_name.error
return 2
print 'Loading additional data for %s' % movie_title
movie_sort_name = movie_sort_name.sort_name
movie_rating = movielibrary.movie_rating(
movie_data.data['releases']['countries'])
movie_genre = movielibrary.movie_imdb_genre(movie_data.data['imdb_id'])
if movie_genre is None:
movie_genre = movie_data.data['genres'][0]['name']
movie_studio = movielibrary.movie_imdb_studio(movie_data.data['imdb_id'])
if movie_studio is None:
company_count = len(movie_data.data['production_companies'])
if 'production_companies' in movie_data.data and company_count > 0:
movie_studio = movie_data.data['production_companies'][0]['name']
movie_info = collections.namedtuple(
'movie_info', 'data genre studio sort_name rating')
movie_data = movie_info(movie_data.data, movie_genre,
movie_studio, movie_sort_name, movie_rating)
nfo_xml = build_nfo_xml(movie_data)
if nfo_xml.error is not None:
print nfo_xml.error
return 2
nfo_file = disklibrary.file_path(
file_parts.file_path, file_parts.file_title + '.nfo')
print 'Writing NFO file'
disklibrary.file_delete(nfo_file)
disklibrary.file_write(nfo_file, nfo_xml.xml)
imdb_url = "http://www.imdb.com/title/" + nfo_xml.imdb + '/'
disklibrary.file_write(nfo_file, imdb_url)
if nfo_xml.poster is not None:
poster_path = movielibrary.movie_poster(
file_parts.file_path, nfo_xml.poster)
save_poster = disklibrary.file_path(
file_parts.file_path, file_parts.file_title + '-cover.jpg')
print 'Downloading movie poster'
if not disklibrary.file_rename(poster_path, save_poster):
print 'Error creating movie cover'
if nfo_xml.backdrop is not None:
backdrop_path = movielibrary.movie_backdrop(
file_parts.file_path, nfo_xml.backdrop)
save_backdrop = disklibrary.file_path(
file_parts.file_path, file_parts.file_title + '-fanart.jpg')
print 'Downloading movie background'
if not disklibrary.file_rename(backdrop_path, save_backdrop):
print 'Error creating movie background'
return 0