python鉴黄程序

# -*- coding: utf8 -*-
# Author:wxq
#python 2.7
import sys
import os
import os.path as p
import shutil
import _io
from collections import namedtuple
from PIL import Image

def check_porn(file):
    img = Image.open(file).convert(‘YCbCr‘)
    w, h = img.size
    data = img.getdata()
    cnt = 0
    for i, ycbcr in enumerate(data):
        y, cb, cr = ycbcr
        if 86 <= cb <= 117 and 140 <= cr <= 168:
            cnt += 1
    if cnt > w * h * 0.1:
        return True
    else:
        return False

if __name__ == ‘__main__‘:
    imgdir = sys.argv[1]
    dstdir = sys.argv[2]
    if imgdir == ‘‘ or dstdir == ‘‘:
        raise Exception(‘参数不能为空, 用法: porn_check.py <图片目录> <目标目录>‘)
    if p.isdir(imgdir):
        filelist = [p.join(imgdir, file) for file in os.listdir(imgdir)]
    else:
        raise Exception(‘参数错误, 图片目录不合法‘)
        sys.exit(1)
    if not p.isdir(dstdir):
        raise Exception(‘参数错误, 目标目录不合法‘)

    for file in filelist:
        if p.isfile(file):
            is_porn = check_porn(file)
            if is_porn:
                print(file, ‘is porn‘)
                shutil.move(file, dstdir)
            else:
                print(file, ‘is not porn‘)