python遍历目录加水印

python遍历目录加水印

from watermarker.marker import  add_mark
import os
import shutil

def mark(file, outdir, mark = "火苗999℃"):
    add_mark(file = file, out = outdir, mark = mark
            #, color = "#000000"
            , opacity=0.1, angle=30, space=300, size=20)
    return 

def mark_dir2(root, out):
    if not os.path.exists(out):
        # os.path.isdir(out):
        os.mkdir(out)
    files = os.listdir(root)
    for fname in files:
        fpath = root + "\\" + fname
        if os.path.isfile(fpath):   # not a dir
            file_name, file_extension = os.path.splitext(fpath)
            if ".gif" != file_extension:
                mark(fpath, out)
            else:
                shutil.copy(fpath, out + "\\" + fname)
        elif os.path.isdir(fpath):  # is a dir
            subdir = root + "\\" + fname
            subout =  out + "\\" + fname
            mark_dir2(subdir, subout)
        else :
            print("no defined {}!".format(fpath))
    return 

if __name__ == '__main__' :
    src = 'H:\\myblog\\blog\\source\\picture\\src'
    out = 'H:\\myblog\\blog\\source\\picture\\mark'
    mark_dir2(src, out)