SubsTool.py

# coding: utf-8

import os
import sys
import getopt
import codecs
import io

## Command Line: SubsTool -s <sourceTXT> [ -f <font file> ]

python_ver = sys.version_info.major
filename_length = 4
source = ''
fontname = "MFShangYa_Noncommercial-Regular.otf"
_usage = """
   Usage: python SubsTool.py -s <sourceTXT> [ -f <fontfile> ]

        sourceTXT (necessary) : SubTitles in txt file, use Enter to split
        fontfile  (optional)  : Font Family file 
                    default   : MFShangYa_Noncommercial-Regular.otf
 
   Library PIL (pillow) is necessary

"""

def usage():
    print(_usage)


def fix_input():
    if python_ver == 2:
        return raw_input()
    else:
        return input()


def del_file(path):
    for i in os.listdir(path):
        path_file = os.path.join(path,i)
        if os.path.isfile(path_file):
            os.remove(path_file)
        else:
             del_file(path_file)


def filename_fix(now):
    if len(str(now)) < filename_length :
        return filename_fix('0' + str(now))
    else:
        return now


# Check PIL Library
try:
    from PIL import Image, ImageFont, ImageDraw, ImageColor
except :
    print(
'''
Please install PIL

Install Command:

      pip install pillow
Or:   pip3 install pillow (MacOS)

''')
    exit()
    

## parse Args
try:
    options,args = getopt.getopt(sys.argv[1:],"hs:f:",["help","source=","font="])
except getopt.GetoptError:
    usage()
    sys.exit()

for name,value in options:
    if name in ("-h","--help"):
        usage()
    if name in ("-s","--source"):
        source=value
        if(os.path.exists(source) is False):
            print("source file doesn't exist")
            exit(0)
    if name in ("-f","--font"):
        fontname=value
        print("Custom Font Family: "+fontname)

if(os.path.exists(fontname) is False):
    print("Font Family file "+fontname+" doesn't exist")
    exit(0)
   
if source == "":
    usage()
    exit()


# fontsize为字体大小
# subcoord 为字幕左上角坐标
# save_directory 字幕图片保存目录
fontsize = 50
subcoord1 = (90,850)
subcoord2 = (90,920)
save_directory=os.path.splitext(os.path.split(source)[1])[0]
if(os.path.exists(save_directory) is False):
    os.mkdir(save_directory)
print("\nThe directory and file under '" + save_directory + "/' will be deleted.\npress y to continue, other input will stop this script")
if(fix_input() != "y"):
    exit(0)
del_file(save_directory)


## 打开文件并读取字幕数据
try:
    subText = open(source,"r",encoding='UTF-8')
    subList = subText.readlines()
except:
    print("Warning: Please use UTF-8 Encoding instead of GBK")
    subText = open(source,"r",encoding='gbk')
    subList = subText.readlines()
print("Read File Successfully, " + str(len(subList)) + " Lines Found.")

## 读取并设置字幕字体
font = ImageFont.truetype(fontname, fontsize, 0, "gbk")

# create an image to implement `Clear Layout` in `Wirecast Play List` 
im = Image.new("RGBA",(1920,1080))
im.save(save_directory + "/" + save_directory.replace(" ","") + "-" + filename_fix("0") + "-Clear Layout" + ".png")

if(len(subList)%2==1):
    subList.append("")
    
# for i in range(0,len(subList)):
counter = 0
file_counter=1
while counter < len(subList) -1:
    ## 创建新图片
    subString1 = subList[counter]
    subString2 = subList[counter+1]
    
    im = Image.new("RGBA", (1920,1080))
    ## 创建绘图句柄
    draw = ImageDraw.Draw(im)
    draw.text(subcoord1, subString1, font=font)
    draw.text(subcoord2, subString2, font=font)
    # draw.text(subcoord2, subList[i], font=font)
    dic = save_directory + "/" + save_directory.replace(" ","") + "-" + filename_fix(str(file_counter)) + ".png"
    print(dic)
    im.save(dic,"PNG")
    file_counter += 1
    counter += 2

im = Image.new("RGBA",(1920,1080))
im.save(save_directory + "/" + save_directory.replace(" ","") + "-" + pow(10,filename_length)-1 + "-End" + ".png")