#encoding: utf-8

from flask import (

    Blueprint,

    request,

    jsonify,

    url_for,

    send_from_directory,

    current_app as app

)

import json

import re

import string

import time

import hashlib

import random

import base64

import sys

import os

from urllib import parse

# 更改工作目录。这么做的目的是七牛qiniu的sdk

# 在设置缓存路径的时候默认会设置到C:/Windows/System32下面

# 会造成没有权限创建。

os.chdir(os.path.abspath(sys.path[0]))

try:

    import qiniu

except:

    pass

from io import BytesIO

bp = Blueprint('ueditor',__name__,url_prefix='/ueditor')

UEDITOR_UPLOAD_PATH = ""

UEDITOR_UPLOAD_TO_QINIU = False

UEDITOR_QINIU_ACCESS_KEY = ""

UEDITOR_QINIU_SECRET_KEY = ""

UEDITOR_QINIU_BUCKET_NAME = ""

UEDITOR_QINIU_DOMAIN = ""

@bp.before_app_first_request

def before_first_request():

    global UEDITOR_UPLOAD_PATH

    global UEDITOR_UPLOAD_TO_QINIU

    global UEDITOR_QINIU_ACCESS_KEY

    global UEDITOR_QINIU_SECRET_KEY

    global UEDITOR_QINIU_BUCKET_NAME

    global UEDITOR_QINIU_DOMAIN

    UEDITOR_UPLOAD_PATH = app.config.get('UEDITOR_UPLOAD_PATH')

    if UEDITOR_UPLOAD_PATH and not os.path.exists(UEDITOR_UPLOAD_PATH):

        os.mkdir(UEDITOR_UPLOAD_PATH)

    UEDITOR_UPLOAD_TO_QINIU = app.config.get("UEDITOR_UPLOAD_TO_QINIU")

    if UEDITOR_UPLOAD_TO_QINIU:

        try:

            UEDITOR_QINIU_ACCESS_KEY = app.config["UEDITOR_QINIU_ACCESS_KEY"]

            UEDITOR_QINIU_SECRET_KEY = app.config["UEDITOR_QINIU_SECRET_KEY"]

            UEDITOR_QINIU_BUCKET_NAME = app.config["UEDITOR_QINIU_BUCKET_NAME"]

            UEDITOR_QINIU_DOMAIN = app.config["UEDITOR_QINIU_DOMAIN"]

        except Exception as e:

            option = e.args[0]

            raise RuntimeError('请在app.config中配置%s！'%option)

    csrf = app.extensions.get('csrf')

    if csrf:

        csrf.exempt(upload)

def _random_filename(rawfilename):

    letters = string.ascii_letters

    random_filename = str(time.time()) + "".join(random.sample(letters,5))

    filename = hashlib.md5(random_filename.encode('utf-8')).hexdigest()

    subffix = os.path.splitext(rawfilename)[-1]

    return filename + subffix

@bp.route('/upload/',methods=['GET','POST'])

def upload():

    action = request.args.get('action')

    result = {}

    if action == 'config':

        config_path = os.path.join(bp.static_folder or app.static_folder,'ueditor','config.json')

        with open(config_path,'r',encoding='utf-8') as fp:

            result = json.loads(re.sub(r'\/\*.*\*\/','',fp.read()))

    elif action in ['uploadimage','uploadvideo','uploadfile']:

        image = request.files.get("upfile")

        filename = image.filename

        save_filename = _random_filename(filename)

        result = {

            'state': '',

            'url': '',

            'title': '',

            'original': ''

        }

        if UEDITOR_UPLOAD_TO_QINIU:

            if not sys.modules.get('qiniu'):

                raise RuntimeError('没有导入qiniu模块！')

            buffer = BytesIO()

            image.save(buffer)

            buffer.seek(0)

            q = qiniu.Auth(UEDITOR_QINIU_ACCESS_KEY, UEDITOR_QINIU_SECRET_KEY)

            token = q.upload_token(UEDITOR_QINIU_BUCKET_NAME)

            ret,info = qiniu.put_data(token,save_filename,buffer.read())

            if info.ok:

                result['state'] = "SUCCESS"

                result['url'] = parse.urljoin(UEDITOR_QINIU_DOMAIN,ret['key'])

                result['title'] = ret['key']

                result['original'] = ret['key']

        else:

            image.save(os.path.join(UEDITOR_UPLOAD_PATH, save_filename))

            result['state'] = "SUCCESS"

            result['url'] = url_for('ueditor.files',filename=save_filename)

            result['title'] = save_filename,

            result['original'] = image.filename

    elif action == 'uploadscrawl':

        base64data = request.form.get("upfile")

        img = base64.b64decode(base64data)

        filename = _random_filename('xx.png')

        filepath = os.path.join(UEDITOR_UPLOAD_PATH,filename)

        with open(filepath,'wb') as fp:

            fp.write(img)

        result = {

            "state": "SUCCESS",

            "url": url_for('files',filename=filename),

            "title": filename,

            "original": filename

        }

    return jsonify(result)

@bp.route('/files/<filename>/')

def files(filename):

    return send_from_directory(UEDITOR_UPLOAD_PATH,filename)