Merge pull request #2 from FarisR99/release-1.0.1

Release 1.0.1

Author: FarisR99

Anime4K.py

@@ -89,7 +89,9 @@
 
 # Validate "output" argument
 if mode == "audio" or mode == "subs":
-    output = args['output'] or ""
+    output = args['output']
+    if output is None:
+        output = ""
     if output != "":
         if not os.path.isdir(output):
             print("Output directory {0} does not exist".format(output))
@@ -99,6 +101,8 @@
                 output = output + "/"
 elif mode == "mux" or mode == "shader":
     output = args['output'] or "out.mkv"
+elif mode == "split":
+    output = args['output']
 
 if mode == "audio":
     extract_audio(fn, output)
@@ -113,6 +117,6 @@
 elif mode == "split":
     length = get_video_length(fn)
     split_by_seconds(filename=fn, split_length=args['split_length'],
-                     video_length=length, split_dir=args['output'])
+                     video_length=length, split_dir=output)
 else:
     print("Unknown option: {0}".format(mode))

extract_subs.py

@@ -12,6 +12,10 @@ def genExt(codec):
         return "ass"
     elif "SRT" in codec or "SubRip" in codec:
         return "srt"
+    elif "VobSub" in codec:
+        return "idx"
+    else:
+        return None
 
 
 def extract_subs(fn: str, out_dir: str):
@@ -34,9 +38,13 @@ def extract_subs(fn: str, out_dir: str):
     for track in tracks:
         if track.track_type == 'subtitles':
             ext = genExt(track._track_codec)
+            if ext is None:
+                print("WARNING: Skipping unknown subtitle with id={0}, codec={1}".format(str(track._track_id), str(track._track_codec or "Unknown")))
+                continue
             lang = track._language
             id = str(track._track_id)
 
             subprocess.call(['mkvextract', 'tracks', fn,
-                             id + ':' + out_dir + lang + '_' + id + '.' + ext])
+                             id + ':' + str(out_dir) + str(
+                                 lang) + '_' + id + '.' + ext])
     print("Subtitle extraction end time: " + current_date())

mux.py

@@ -81,6 +81,7 @@ def mux(fn: str, out: str):
     addSubs(mkv, "sup")
     addSubs(mkv, "srt")
     addSubs(mkv, "ass")
+    addSubs(mkv, "idx")
 
     print("Mux start time: " + current_date())
     mkv.mux(out)
@@ -101,4 +102,6 @@ def mux(fn: str, out: str):
     delete_by_extension("sup")
     delete_by_extension("srt")
     delete_by_extension("ass")
+    delete_by_extension("idx")
+    delete_by_extension("sub")
     print("Cleaned!")

utils.py

@@ -2,7 +2,7 @@
 from datetime import datetime
 from shutil import which
 
-__current_version__ = '1.0.0'
+__current_version__ = '1.0.1'
 
 
 def credz():