Function main

utils/marker.py:168–236

()

def main():
    parse = argparse.ArgumentParser(formatter_class=argparse.RawTextHelpFormatter)
    parse.add_argument("-f", "--file", type=str, default="./input",
                       help="image file path or directory")
    parse.add_argument("-m", "--mark", type=str, default="测试一句话行不行", help="watermark content")
    parse.add_argument("-o", "--out", default="./2",
                       help="image output directory, default is ./output")
    parse.add_argument("-c", "--color", default="#8B8B1B", type=str,
                       help="text color like '#000000', default is #8B8B1B")
    parse.add_argument("-s", "--space", default=75, type=int,
                       help="space between watermarks, default is 75")
    parse.add_argument("-a", "--angle", default=90, type=int,
                       help="rotate angle of watermarks, default is 30")
    parse.add_argument("--font-family", default="./font/青鸟华光简琥珀.ttf", type=str,
                       help=textwrap.dedent('''\
                       font family of text, default is './font/青鸟华光简琥珀.ttf'
                       using font in system just by font file name
                       for example 'PingFang.ttc', which is default installed on macOS
                       '''))
    parse.add_argument("--font-height-crop", default="1.2", type=str,
                       help=textwrap.dedent('''\
                       change watermark font height crop
                       float will be parsed to factor; int will be parsed to value
                       default is '1.2', meaning 1.2 times font size
                       this useful with CJK font, because line height may be higher than size
                       '''))
    parse.add_argument("--size", default=50, type=int,
                       help="font size of text, default is 50")
    parse.add_argument("--opacity", default=0.15, type=float,
                       help="opacity of watermarks, default is 0.15")
    parse.add_argument("--quality", default=100, type=int,
                       help="quality of output images, default is 90")

    args = parse.parse_args()

    # 随机参数，从[A,B]中随机选取一个整数
    space_ = [30, 40]
    angel_ = [0, 180]
    size_ = [20, 60]
    opacity_ = [50, 95]
    # 字体随机（需要别的字体的话，可以去字体网站下载）
    fonts = [r'./font/青鸟华光简琥珀.ttf', r'./font/楷体_GB2312.ttf', './font/方正仿宋_GBK.TTF', './font/times.ttf',
             './font/simhei.ttf']
    if isinstance(args.mark, str) and sys.version_info[0] < 3:
        args.mark = args.mark.decode("utf-8")

    if os.path.isdir(args.file):
        names = os.listdir(args.file)
        for name in names:
            image_file = os.path.join(args.file, name)
            space = random.randint(space_[0], space_[1])
            args.space = space
            angel = random.randint(angel_[0], angel_[1])
            args.angle = angel
            size = random.randint(size_[0], size_[1])
            args.size = size
            opacity = random.randint(opacity_[0], opacity_[1]) / 100
            args.opacity = opacity