new_tiku_structure_2021/photo_upload.py

#!/usr/bin/env/python
# -*- coding:utf-8 -*-
import os
import re
import time
import datetime
import random
import hashlib
from ufile import filemanager
from ufile import config as ufile_config
from ufile import logger as img_log
public_key = 'ucloudyunkaopei@outlook.com13615403931104805307'
private_key = 'bcfd5bb66ca527c9be9fd7f3e784fbfc90c4bba5'

addr = '.cn-bj.ufileos.com'  # 后缀拼接
ufile_config.set_default(uploadsuffix=addr)

# 公共空间名称
public_bucket = 'zxhx'  # 线上正式环境
# public_bucket = 'zxhx-test'  # 线上测试环境

image_upload_log = './logs/image_log.txt'  # 图片上传日志
img_log.set_log_file(image_upload_log)


# todo 上传图片进程
def upload_img_to_ucloud(param_ucloud):
    put_key, localfile = param_ucloud
    putufile_handler = filemanager.FileManager(public_key, private_key)
    # 普通上传文件至公共空间
    ret, resp = putufile_handler.putfile(public_bucket, put_key, localfile, header=None)
    assert resp.status_code == 200


def upload_replace_image(filename_root, sid, html):
    return_error = {"errcode": 1,
                    "errmsg": "word图片上传失败。"}

    daytime = datetime.datetime.now().strftime('/%Y/%m/%d/')
    image_path = filename_root + "/files"

    # todo 判断试卷是否含有图片，如果有就替换上传，没有就不处理
    local2online_dict = {}
    judge_file = os.path.isdir(image_path)
    if judge_file:  # 可以不用判断
        image_number = re.findall(r'<img\s*src\s*=\s*"files/image', str(html))

        local_images_path_list = os.listdir(image_path)  # 本地图片文件名
        local_images_path_list = list(filter(
            lambda x: str(x).endswith(".png") or str(x).endswith(".gif") or str(x).endswith(".jpeg") or str(x).endswith(
                ".jpg"), local_images_path_list))  # 只允许特定后缀名的图片

        if len(image_number) != len(local_images_path_list):
            return return_error
        else:
            # 從大到小把圖片进行排序
            try:
                local_images_path_list.sort(key=lambda x: int(re.search(r"image(\d+)\.[pngifje]+", str(x)).group(1)))
            except:
                return return_error

            # logger.info("local_images_path_list==>{}".format(str(local_images_path_list)))

            put_key_list = []  # 线上图片地址
            localfile_list = []  # 本地图片地址
            rawpic_list = []
            online_image_url_list = []

            if local_images_path_list:
                try:
                    for i, img in enumerate(local_images_path_list, start=1):  # 所有的图片【image1.png，image2.png，。。。】

                        src_pat2 = re.compile(r'<img\s*src\s*=\s*"files/image{}\.png"'.format(i))
                        src_pat3 = re.compile(r'<img\s*src\s*=\s*"files/image{}\.gif"'.format(i))
                        src_pat4 = re.compile(r'<img\s*src\s*=\s*"files/image{}\.jpeg"'.format(i))
                        src_pat5 = re.compile(r'<img\s*src\s*=\s*"files/image{}\.jpg"'.format(i))
                        if str(img).endswith("png"):
                            pat = src_pat2
                            # re_list.append(src_pat2)
                        elif str(img).endswith("gif"):
                            pat = src_pat3
                            # re_list.append(src_pat3)
                        elif str(img).endswith("jpeg"):
                            pat = src_pat4
                            # re_list.append(src_pat4)
                        elif str(img).endswith("jpg"):
                            pat = src_pat5
                            # re_list.append(src_pat5)

                        # todo 上传线上，并替换线上图片
                        localfile = image_path + "/{}".format(img)
                        rawfile = '<img src="files'+ "/{}".format(img)
                        localfile_list.append(localfile)
                        rawpic_list.append(rawfile)

                        # todo 图片压缩
                        # resize_img(localfile, localfile)
                        hash_img = get_md5(img)

                        # 上传文件在空间中的名称
                        put_key = "teacher/uploadfiles/wording/" + str(sid) + str(daytime) + str(hash_img)
                        put_key_list.append(put_key)

                        # html替换为线上的地址
                        online_image_url = "http://" + str(public_bucket) + str(addr) + "/" + str(put_key)
                        # html = pat.sub(r'<img src={}'.format(online_image_url), str(html))
                        online_image_url='<img src="{}'.format(online_image_url)
                        online_image_url_list.append(online_image_url)
                    local2online_dict = dict(zip(rawpic_list,online_image_url_list))

                    return local2online_dict, put_key_list, localfile_list

                except:
                    return return_error
            else:
                return return_error
    else:
        return local2online_dict, [], []


def get_md5(image):
    """
    由于hash不处理unicode编码的字符串（python3默认字符串是unicode）
        所以这里判断是否字符串，如果是则进行转码
        初始化md5、将image_name进行加密、然后返回加密字串
    """
    image_name, image_type = str(image).split(".")
    image_name = str(image_name) + str(time.time()) + str(random.random())

    if isinstance(image_name, str):
        image_name = image_name.encode("utf-8")
    md = hashlib.md5()
    md.update(image_name)
    # a = time.time()
    # b = random.random()
    return str(md.hexdigest()) + "." + str(image_type)