First commit

converter.py

@@ -0,0 +1,166 @@
+#!/usr/bin/env python3
+'''Its a script!
+    ./converter.py list -dir "/"
+    ./converter.py convert -del -dir "/mnt/media/"
+
+'''
+
+import sys
+import os
+from pathlib import Path
+from pprint import pprint
+from hurry.filesize import size
+
+from subprocess import check_output, STDOUT, CalledProcessError
+
+
+
+def main():
+    if len(sys.argv) >= 2:
+        if sys.argv[1] == "list":
+            if "-file" in sys.argv:
+                pass
+            elif "-dir" in sys.argv:
+                directory = sys.argv[sys.argv.index("-dir") + 1]
+                videolist = get_videolist(directory)
+                for video in videolist:
+                    print(f"{get_codec(video)} - {video}")
+            else:
+                print("Missing directory: ")
+        elif sys.argv[1] == "test":
+            if "-file" in sys.argv:
+                pass
+            elif "-dir" in sys.argv:
+                directory = sys.argv[sys.argv.index("-dir") + 1]
+                videolist = get_videolist(directory)
+                video = videolist[0]
+
+                folder = str(video)[:str(video).rindex("/") + 1]
+                oldfilename = str(video)[str(video).rindex("/") + 1:]
+                newfilename = oldfilename[:-4] + ".mkv"
+
+                if convert(folder + oldfilename, folder + newfilename):
+                    if "-del" in sys.argv:
+                        delete(folder + oldfilename)
+            else:
+                print("Missing directory: ")
+
+            
+        elif sys.argv[1] == "convert":
+            if "-file" in sys.argv:
+                video = sys.argv[sys.argv.index("-file") + 1]
+                folder = str(video)[:str(video).rindex("/") + 1]
+                oldfilename = str(video)[str(video).rindex("/") + 1:]
+                newfilename = oldfilename[:-4] + ".mkv"
+
+                print(f"***********   converting {oldfilename} to {newfilename}   ***********")
+                try:
+                    if convert(folder + oldfilename, folder + newfilename):
+                        if "-del" in sys.argv:
+                            delete(folder + oldfilename)
+                except:
+                    delete(folder + newfilename)
+                    return False
+            elif "-dir" in sys.argv:
+                directory = sys.argv[sys.argv.index("-dir") + 1]
+                videolist = get_videolist(directory)
+                counter = 0
+                for video in videolist:
+                    folder = str(video)[:str(video).rindex("/") + 1]
+                    oldfilename = str(video)[str(video).rindex("/") + 1:]
+                    newfilename = oldfilename[:-4] + ".mkv"
+
+                    counter += 1
+                    print(f"***********   convert {counter} of {len(videolist)}   ***********")
+                    try:
+                        if convert(folder + oldfilename, folder + newfilename):
+                            if "-del" in sys.argv:
+                                delete(folder + oldfilename)
+                    except:
+                        delete(folder + newfilename)
+                        return False
+
+
+def get_wmv_list(path):
+    return list(path.rglob("*.[wW][mM][vV]"))
+
+def get_avi_list(path):
+    return list(path.rglob("*.[aA][vV][iI]"))
+
+def get_mkv_list(path):
+    return list(path.rglob("*.[mM][kK][vV]"))
+
+def get_mp4_list(path):
+    return list(path.rglob("*.[mM][pP]4"))
+
+def get_videolist(parentdir):
+    videolist = []
+    path = Path(parentdir)
+    if "-all_wmv" in sys.argv or "-any" in sys.argv:
+        videolist += get_wmv_list(path)
+    if "-all_avi" in sys.argv or "-any" in sys.argv:
+        videolist += get_avi_list(path)
+    if "-all_mkv" in sys.argv or "-any" in sys.argv:
+        videolist += get_mkv_list(path)
+    if "-all_mp4" in sys.argv or "-any" in sys.argv:
+        videolist += get_mp4_list(path)
+    
+    videolist_tmp = videolist
+    videolist = [video for video in videolist_tmp if video.is_file()]
+
+    if "-old" in sys.argv:
+        videolist = get_old(videolist)
+        
+    return videolist
+
+def get_old(videolist_temp):
+    return [video for video in videolist_temp if get_codec(video) not in ["hevc", "av1"]]
+
+def get_codec(filename):
+    try:
+        args = ["/usr/bin/ffprobe", "-v", "error", "-select_streams", "v:0", "-show_entries", "stream=codec_name", "-of", "default=noprint_wrappers=1:nokey=1", str(filename)]
+        output = check_output(args, stderr=STDOUT)
+    except CalledProcessError:
+        print(f"There seams to be an error with {filename}")
+        return False
+    return output.decode().strip()
+
+def convert(oldfilename, newfilename):
+    oldsize = size(Path(oldfilename).stat().st_size)
+    print(f"Starting conversion of {oldfilename}({oldsize}) ...")
+
+    # Preparing ffmpeg command and input file
+    args = ['/usr/bin/ffmpeg', '-i', oldfilename]
+
+    # conversion options
+    args += ['-c:v', 'libx265']
+    args += ['-max_muxing_queue_size', '1000']
+
+    # conversion output
+    args += [newfilename]
+
+    #args = ['/usr/bin/ffmpeg', '-i', oldfilename, newfilename]
+    try:
+        txt = check_output(args, stderr=STDOUT)
+    except CalledProcessError as e:
+        print(f"Conversion failed {e}")
+        return False
+    else:
+        newsize = size(Path(newfilename).stat().st_size)
+        oldfilename = str(oldfilename)[str(oldfilename).rindex("/") + 1:]
+        newfilename = str(newfilename)[str(newfilename).rindex("/") + 1:]
+        print(f"Converted {oldfilename}({oldsize}) to {newfilename}({newsize}) successfully")
+        return True
+
+def delete(filename):
+    try:
+        os.remove(filename)
+    except OSError:
+        print(f"Error deleting {filename}")
+        return False
+
+    print(f"Deleted {filename}")
+    return True
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -0,0 +1,2 @@
+hurry.filesize
+pathlib