createThumb.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Created on 2019年6月17日
@author: Irony
@site: 
@email: 892768447@qq.com
@file: 
@description: 生成缩略图
"""
import os
import sys
import traceback

import requests
from PIL import Image

try:
    os.mkdir('tmp')
except Exception:
    pass


def download(url, src, dst):
    try:
        resp = requests.get(url)
        open(src, 'wb').write(resp.content)
        thumbnail(src, dst)
    except Exception:
        traceback.print_exc()


def thumbnail(src, dst):
    try:
        img = Image.open(src)
        img = img.convert('RGB')
        img.thumbnail((80, 80), Image.ANTIALIAS)
        img.save(dst)
    except Exception:
        traceback.print_exc()


if __name__ == '__main__':
    length = len(sys.argv)
    if length == 3:
        # 本地文件生成缩略图
        _, src, dst = sys.argv
        thumbnail(src, dst)
    elif length == 4:
        # 远程文件生成缩略图
        _, url, src, dst = sys.argv
        if url.find('jsdelivr') > -1 or url.find('jsd.') > -1:
            print('ignore cdn file')
        else:
            download(url, src, dst)