From 1a312644ca6cc6dcedb389959a63d557276ef323 Mon Sep 17 00:00:00 2001
From: Fabrice Quenneville <fabquenneville@gmail.com>
Date: Sat, 14 Nov 2020 20:49:48 -0500
Subject: [PATCH] Added support for listing / filtering by resolution

---
 curator.py | 24 +++++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/curator.py b/curator.py
index a871a73..630f2e4 100755
--- a/curator.py
+++ b/curator.py
@@ -50,8 +50,10 @@ def main():
                 #directory = sys.argv[sys.argv.index("-dir") + 1]
                 for directory in directories:
                     videolist += get_videolist(directory, inputs, filters)
+                videolist.sort()
                 for video in videolist:
-                    print(f"{get_codec(video)} - {video}")
+                    resolution = get_resolution(video)
+                    print(f"{get_codec(video)} - {resolution[0]}p - {video}")
             else:
                 print(f"{bcolors.FAIL}Missing directory: {bcolors.ENDC}")
         elif sys.argv[1] == "test":
@@ -154,7 +156,7 @@ def get_videolist(parentdir, inputs = ["any"], filters = []):
     # Filter the list for specifi codecs
     videolist_tmp = videolist
     print(f"{bcolors.OKGREEN}Filtering {len(videolist)} videos for the requested parameters{bcolors.ENDC}")
-    if len(filters) > 0:
+    if len([filt for filt in filters if filt not in ["lowres"]]) > 0:
         videolist = []
 
         if "old" in filters:
@@ -172,10 +174,25 @@ def get_videolist(parentdir, inputs = ["any"], filters = []):
         if "x264" in filters:
             videolist += [video for video in videolist_tmp if get_codec(video) in ["x264"]]
         
+    if len(filters) > 0 and "lowres" in filters:
+        #print("here")
+        videolist_tmp = videolist
+        videolist = [video for video in videolist_tmp if get_resolution(video)[0] < 720]
+
     print(f"{bcolors.OKGREEN}Found {len(videolist)} videos for the requested parameters{bcolors.ENDC}")
     return videolist
 
 
+def get_resolution(filename):
+    try:
+        args = ["ffprobe","-v","error","-select_streams","v:0","-show_entries","stream=width,height","-of","csv=s=x:p=0",str(filename)]
+        output = subprocess.check_output(args, stderr=subprocess.STDOUT)
+    except subprocess.CalledProcessError:
+        print(f"{bcolors.FAIL}There seams to be an error with {filename}{bcolors.ENDC}")
+        return False
+    return [int(output.decode().strip().split("x")[1]), int(output.decode().strip().split("x")[0])]
+
+
 def get_codec(filename):
     try:
         args = ["ffprobe", "-v", "error", "-select_streams", "v:0", "-show_entries", "stream=codec_name", "-of", "default=noprint_wrappers=1:nokey=1", str(filename)]
@@ -187,7 +204,8 @@ def get_codec(filename):
 
 def convert(oldfilename, newfilename, codec = "x265"):
     oldsize = size(Path(oldfilename).stat().st_size)
-    print(f"{bcolors.OKGREEN}Starting conversion of {oldfilename}({oldsize}) from {get_codec(oldfilename)} to {codec}...{bcolors.ENDC}")
+    resolution = get_resolution(oldfilename)
+    print(f"{bcolors.OKGREEN}Starting conversion of {oldfilename}{bcolors.OKCYAN}({oldsize})({resolution[0]}p){bcolors.OKGREEN} from {bcolors.OKCYAN}{get_codec(oldfilename)}{bcolors.OKGREEN} to {bcolors.OKCYAN}{codec}{bcolors.OKGREEN}...{bcolors.ENDC}")
 
     # Preparing ffmpeg command and input file
     args = ['ffmpeg', '-i', oldfilename]