Anime4K-Encoder-4.0.1-/mux.py at master · Secksdendfordff/Anime4K-Encoder-4.0.1- · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import glob, os, sys, subprocess
from pymkv import MKVFile, MKVTrack
from utils import langToShort, shortToLong


def addAudio(source, ext):
    for file in glob.glob("*." + ext):
        t = MKVTrack(file)
        lang = file.split('.')[0]
        t.track_name = lang
        t.language = langToShort(lang)
        print("Adding AudioTrack")
        source.add_track(t)

def addSubs(source, ext):
    for file in glob.glob("*." + ext):
        t = MKVTrack(file)
        lang = file.split('.')[0]
        if "_" in lang:
            lang = lang.split('_')[0]
        else:
            lang = lang
        t.track_name = shortToLong(lang)
        t.language = lang
        print("Adding new SUB")
        source.add_track(t)

def delete_by_extension(ext):
    for file in glob.glob("*." + ext):
        os.remove(file)

def mux(fn, out):
    mkv = MKVFile(fn)

    addAudio(mkv, "AAC")
    addAudio(mkv, "MP3")
    addAudio(mkv, "DTS")
    addAudio(mkv, "Opus")
    addAudio(mkv, "FLAC")
    addAudio(mkv, "TrueHD Atmos")
    addAudio(mkv, "AC-3")
    addAudio(mkv, "DTS-HD Master Audio")

    addSubs(mkv, "sup")
    addSubs(mkv, "srt")
    addSubs(mkv, "ass")

    mkv.mux(out)

    #clean up
    print("Cleaning...")

    delete_by_extension("AAC")
    delete_by_extension("MP3")
    delete_by_extension("DTS")
    delete_by_extension("Opus")
    delete_by_extension("FLAC")
    delete_by_extension("TrueHD Atmos")
    delete_by_extension("AC-3")
    delete_by_extension("DTS-HD Master Audio")

    delete_by_extension("sup")
    delete_by_extension("srt")
    delete_by_extension("ass")
    print("KTHXBYE")