A amt.py

@@ -0,0 +1,269 @@ 
+'''
+Its a ugly script and its also my first ever python script. I wrote this whitout knowing alot about the language (like Classes, functions and other OOP stuff, 
+which If I knew, would make this script more readable and pretty). I just know some basic stuff like core object types and some begginer functions. 
+Everything else, I either know by my past knowledge of coding and referencing StackOverflow and the modules' docs
+
+This script has three components:
+1. A mkv to mp4 convertor in which it also can add one subtitle file named exactly the same as the movie file (with srt extension) in the same 
+folder as the script and it aslo adds the 'title', 'year', 'genre and 'comment' metadata tags to the resultant mp4 file by using imdb's data by 'imdbpie'
+
+2. A movie poster fetcher. For which it uses the 'tmdbsimple' module that is a wrapper library for themoviedb.org APIs. Here it calls the 'poster' module
+which as a function to get the path of the poster and it is downoaded in the same path using urllib.request.urlretrieve
+
+3. A poster adder ( I know, its a silly name, couldn't come up with anything else) that adds the poster to the mp4 file using 'mutagen' module
+
+TO-DO
+1. Add a way to notify when the script is done running
+2. Seems a little too much for now but I could change/add stream specific metadata according to the file
+For example depending upon what kind of audio the mkv file has (like AAC 5.1, DTS 5.1), the script can also change the title/handler of the audio stream
+right now it just blanks it out.
+The same can be done for subtitles depending upon language
+Again this is too much for now. I don't see my self implementing this anytime soon. All I need is one english.srt for now and that does it and I don't donwload multi-audio movies.
+3. Add proper error handleling for ffmpeg
+
+'''
+
+
+import os
+import subprocess
+import urllib
+import tmdbsimple as tmdb
+from imdbpie import Imdb
+from mutagen.mp4 import MP4, MP4Cover
+
+#Setting the API key for usage of TMDB API
+tmdb.API_KEY = 'b888b64c9155c26ade5659ea4dd60e64'
+
+
+def collect_files(file_type):
+    '''
+    returns a list of files in the current directory that are of 
+	the extension passed as string\n
+	eg: collect_files('txt') would return a list of all txt files
+    '''
+    filenames = []
+    for filename in os.listdir(os.getcwd()):
+        if filename.endswith(file_type):
+            filenames.append(filename)
+    return filenames
+
+
+def get_common_files(mediafile_list, srtfile_list):
+	'''
+	returns a list of filenames that are common in mediafile_list and strfile_list. \n
+	While getting common filenames it ignores the extension.\n
+	Also the returned list will have the same file extension the mediafile_list files have 
+	'''
+	media_filenames = [i[:-4] for i in mediafile_list]
+	subtitle_filenames = [i[:-4] for i in srtfile_list]
+	media_type = mediafile_list[0][-4:]
+	media_set = set(media_filenames)
+	srt_set = set(subtitle_filenames)
+	common_files = list(media_set & srt_set)
+	common_files = [i + media_type for i in common_files]
+	common_files.sort()
+	return common_files
+
+def remove_common_files(list1, list2):
+	'''
+	returns a subset of list1 that has common elements removed that were found in both lists
+	or in other words - returns a subset of list1 that has elements unique to only list1
+	'''
+	new_list1 = list(set(list1) - set(list2))		# results in a list of values that are unique to list1
+	new_list1.sort()
+	return new_list1
+
+
+def start_process(filenames, mode ):
+	'''
+	This is the main funtion of the script
+	where it does its main processing\n
+	filenames is the list of files to be processed\n
+	mode = 1,2,3 or 4\n
+	1 means mp4 to tagged mp4\n
+	2 means mp4 with sub to subbed and tagged mp4\n
+	3 means mkv to tagged mp4\n
+	4 means mkv with sub to subbed and tagged mp4
+	'''
+	searchindex = 0
+
+	for filename in filenames:
+		try:
+			title = filename[:-4]
+
+			print('\nFetching movie data for "' + title +'"')
+			search = tmdb.Search()
+			response = search.movie(query=title)
+			# getting a Movies object from the id that we got from the search results
+			try:		# sometimes blank search results are returned
+				tmdb_movie = tmdb.Movies(search.results[searchindex]['id'])		# Don't mind if any error is linted
+			except IndexError:
+				while len(search.results) == 0:		# Don't mind if any error is linted
+					title = input('\nCould not find the movie, Enter alternate movie title >> ')
+					searchindex = int(input('Search result index >> '))
+					response = search.movie(query=title)
+					try:
+						tmdb_movie = tmdb.Movies(search.results[searchindex]['id'])		# Don't mind if any error is linted
+					except IndexError:
+						continue
+			# we get the info about the movie
+			response = tmdb_movie.info() 
+			# making an imdb object
+			imdb = Imdb()
+			# tmdb_movie.imdb_id is the imdb id of the moovie that we searched before usng tmdb
+			imdb_movie = imdb.get_title_by_id(tmdb_movie.imdb_id)		# Don't mind if any error is linted
+			# using imdb provided movie name and 
+			newfilename = imdb_movie.title + ' (' + str(imdb_movie.year) + ').mp4'
+			newfilename = newfilename.replace(':', ' -').replace('/', ' ').replace('?', '')
+
+			command = ""
+
+			if mode == 1:
+				os.rename(filename, newfilename)	# it is required to rename it as its already an mp4 file that wasn't proccessed by ffmpeg
+				# command = 'ffmpeg -i "' + filename + '" -codec copy -c:d bin_data -map_metadata -1 "' + newfilename + '" '
+			if mode == 2:
+				command = 'ffmpeg -i "' + filename + '" -sub_charenc UTF-8 -i "' + filename[:-4] +'.srt" ' + '-map 0 -map 1 -c copy -c:s mov_text -metadata:s:s:0 handler="English Subtitle" -metadata:s:s:0 language=eng -metadata:s:a:0 handler="" -metadata:s:v:0 handler="" "' + newfilename + '"'
+				subprocess.run(command)
+			if mode == 3:
+				command = 'ffmpeg -i "' + filename + '" -c copy -c:s mov_text -metadata:s:s:0 handler="English" -metadata:s:s:0 language=eng -metadata:s:a:0 handler="" -metadata:s:v:0 handler="" "' + newfilename + '"'
+				subprocess.run(command)
+			if mode == 4:
+				command = 'ffmpeg -i "' + filename + '" -sub_charenc UTF-8 -i "' + filename[:-4] +'.srt" ' + '-map 0 -map 1 -c copy -c:s mov_text -metadata:s:s:0 handler="English Subtitle" -metadata:s:s:0 language=eng -metadata:s:a:0 handler="" -metadata:s:v:0 handler="" "' + newfilename + '"'
+				subprocess.run(command)
+			
+
+
+			# the poster is fetched from tmdb only if there is no file named " filename + '.jpg' " in the working directory
+			# this way user can provide their own poster image to be used
+			poster_filename = filename[:-4] + '.jpg'
+			if not os.path.isfile(poster_filename):
+				print('\nFetching the movie poster...')
+				path = search.results[searchindex]['poster_path']	# Don't mind if any error is linted
+				poster_path = r'https://image.tmdb.org/t/p/w640' + path
+				
+				uo = urllib.request.urlopen(poster_path)
+				with open(poster_filename, "wb") as poster_file:
+					poster_file.write(uo.read())
+					poster_file.close()
+
+			# I have commented out the tmbd rating and overview. imdb is a better choice 
+			# foe ratings and it also has a smaller version of the movie plot (plot_outline)
+			# tmdb_rating_and_plot = str('[' + str(search.results[0]['vote_average']) + '/10] - ' + search.results[0]['overview'])
+			imdb_rating_and_plot = str('IMDb rating [' + str(float(imdb_movie.rating)) + '/10] - ' + imdb_movie.plot_outline)
+			# setting the genres of the movie. I use ';' as a delimeter to searate the multiple genre values
+			genre = ';'.join(imdb_movie.genres)
+			# Going overboard and adding directors name to artist taf of the mp4 file
+			directors = imdb_movie.directors_summary
+			director = directors[0].name
+
+
+			
+			video = MP4(newfilename)
+			with open(poster_filename, "rb") as f:
+				video["covr"] = [
+								MP4Cover(f.read(), imageformat=MP4Cover.FORMAT_JPEG)]
+				video['\xa9day'] = str(imdb_movie.year)
+				video['\xa9nam'] = imdb_movie.title
+				video['\xa9cmt'] = imdb_rating_and_plot
+				video['\xa9gen'] = genre
+				video['\xa9ART'] = director
+			print('\nAdding poster and tagging file...')
+
+		# an error occurs somethimes, I am not sure what is the reason for now so I just handle it by letting the loop continue
+		# and saving thr file names that cause the error, in a list
+			try:
+				video.save()
+			except OverflowError:
+				
+
+				remove_meta_command = 'ffmpeg -i "' + newfilename + '" -codec copy -map_metadata -1 "' + newfilename[:-4] + 'new.mp4"'
+				subprocess.run(remove_meta_command)
+				video_new = MP4(newfilename[:-4] + 'new.mp4')
+				with open(poster_filename, "rb") as f:
+					video_new["covr"] = [
+									MP4Cover(f.read(), imageformat=MP4Cover.FORMAT_JPEG)]
+					video_new['\xa9day'] = str(imdb_movie.year)
+					video_new['\xa9nam'] = imdb_movie.title
+					video_new['\xa9cmt'] = imdb_rating_and_plot
+					video_new['\xa9gen'] = genre
+					video_new['\xa9ART'] = director
+				print('\nAdding poster and tagging file...')
+
+				try:
+					video_new.save()
+					if not os.path.exists('auto fixed files'):
+						os.makedirs('auto fixed files')
+					os.rename(newfilename[:-4] + 'new.mp4', 'auto fixed files\\' + newfilename[:-4] + '.mp4')
+					os.remove(newfilename)
+
+				except OverflowError:
+					errored_files.append(filename + ' - Could not save even after striping metadata' )
+					continue
+				
+			os.remove(poster_filename)
+			print('\n' + filename + ' was proccesed successfuly!\n\n==========================================================')
+		except Exception as e:
+			print('\nSome error occured while processing ' + filename + '\n\n==========================================================')
+			errored_files.append(filename + ' - ' + str(e))
+
+
+mp4_filenames = []
+mkv_filenames = []
+srt_filenames = []
+mp4_with_srt_filenames = []
+mkv_with_srt_filenames = []
+errored_files = []
+
+mp4_filenames = collect_files('mp4')
+mkv_filenames = collect_files('mkv')
+srt_filenames = collect_files('srt')
+
+# We check whether there are mp4 files and if yes, then are there any srt files, 
+# if yes, then get the mp4 files that have srts associated with them, then if there
+# are mp4 files that have srt files associated with them then remove the others as they 
+# are to be proccessed separately
+if not len(mp4_filenames) == 0:
+	if not len(srt_filenames) == 0:
+		mp4_with_srt_filenames = get_common_files(mp4_filenames, srt_filenames)
+		if not len(mp4_with_srt_filenames) == 0:
+			mp4_filenames = remove_common_files(mp4_filenames, mp4_with_srt_filenames)
+
+if not len(mkv_filenames) == 0:
+	if not len(srt_filenames) == 0:
+		mkv_with_srt_filenames = get_common_files(mkv_filenames, srt_filenames)
+		if not len(mkv_with_srt_filenames) == 0:
+			mkv_filenames = remove_common_files(mkv_filenames, mkv_with_srt_filenames)
+
+# This is where the main orocess of conversion takes place.
+# We simply check the file lists are not empty and then execute the main task depending 
+# on what type it is according to mode in the funtion "start_process"
+if not len(mp4_filenames) == 0:
+	start_process(mp4_filenames, 1)
+
+if not len(mp4_with_srt_filenames) == 0:
+	start_process(mp4_with_srt_filenames, 2)
+
+if not len(mkv_filenames) == 0:
+	start_process(mkv_filenames, 3)
+
+if not len(mkv_with_srt_filenames) == 0:
+	start_process(mkv_with_srt_filenames, 4)
+
+
+
+if len(mp4_filenames) == 0 and len(mkv_filenames) == 0 and len(mp4_with_srt_filenames) == 0 and len(mkv_with_srt_filenames) == 0:
+	print('There were no MP4 or MKV files found in the directory')
+else:
+	# Checks if there were any files that caused the Overflow Error, if yes then prints them out.
+	if len(errored_files) == 0:
+		print('\n\n\nAll files proccessed successfuly!')
+	else:
+		print('\n\n\nThe files that were not proccessed: \n')
+		for er in errored_files:
+			print(er)
+# TODO add notification at the end
+
+
+
+
+