bhouse_backstage/backstage/upload/routes.py

from backstage.config import UPLOAD_PATH_MAP
import os
from flask import render_template, Blueprint, request, redirect, url_for
import flask
from flask_cors import CORS, cross_origin
from flask.helpers import make_response, send_file, send_from_directory
from werkzeug.wrappers import Response
import requests
from PIL import Image
from urllib.parse import urlparse
import uuid


#from backstage.utils.routes import update_manage_table

upload_app = Blueprint('upload', __name__)


@upload_app.route('/backstage/upload/<path:filepath>', methods=['POST'])
def upload_post(filepath):
    #bdata = request.stream.read()
    #aa = request.get_data()
    if request.method == 'POST':

        # check if the post request has the file part
        # print(request.full_path)
        """ if 'file' not in request.files:
            #flash('No file part')
            return redirect(request.url) """

        file = request.files['image']

        # If the user does not select a file, the browser submits an
        # empty file without a filename.
        """ if file.filename == '':
            #flash('No selected file')
            return redirect(request.url) """

        if file:
            #filename = secure_filename(file.filename)

            #fname = str(uuid.uuid4()) + file.filename[file.filename.rfind("."):]
            #sitepath = UPLOAD_PATH_MAP[0][0] + filepath[filepath.rfind("/")+1:] + "/img/"
            sitepath = UPLOAD_PATH_MAP[0][0] + filepath + "/img/"
            oimgtype = file.filename[file.filename.rfind(".")+1:]
            oimgtypeName = oimgtype
            if oimgtype.lower() == 'jpg':
                oimgtypeName = 'jpeg'

            wfname = str(uuid.uuid4()) + ".webp"
            owfname = wfname.replace('webp', oimgtype)
            #fullpath = UPLOAD_PATH_MAP[0][0] + filepath[filepath.rfind("/")+1:] + "\\img\\" + fname
            #print(UPLOAD_PATH_MAP[0][1] + filepath[filepath.rfind("/")+1:] + "/img/" + fname)
            # file.save(fullpath)
            #image_object = Image.open(fullpath)
            #print(request.args.get('url', type=str))
            wpath = os.getcwd() + "/backstage/upload/" + filepath+"/img/"
            owpath = os.getcwd() + "/backstage/upload/" + filepath+"/img/orig/"
            #fullpath = wpath + fname
            if not os.path.exists(wpath):
                os.makedirs(wpath)
            if not os.path.exists(owpath):
                os.makedirs(owpath)
            if not os.path.exists(sitepath):
                os.makedirs(sitepath)
            # file.save(fullpath)
            image_object = Image.open(file)
            image_object.save(owpath+owfname, oimgtypeName)
            if image_object.size[0] > 1000:
                image_object.thumbnail(size=((800, 800)))
            image_object.save(wpath+wfname, 'webp')
            image_object.save(sitepath+wfname, 'webp')
            #file.save(os.getcwd()+ "/backstage/upload/img/"+ fname)
            # return redirect(url_for('download_file', name=file.filename))
            aa = {"success": 1, "file": {"url": "/backstage/upload/" +
                                         filepath+"/img/" + wfname, "width": image_object.width, "height": image_object.height}}
            # aa = {"success": 1, "file": {"url": UPLOAD_PATH_MAP[0][1] + filepath[filepath.rfind(
            #    "/")+1:] + "/img/" + fname, "width": image_object.width, "height": image_object.height}}
            return aa
    if request.method == 'GET':
        print('GET')
        # print(request.files)
        # print(request.form)

    # print(requests.post("/backstage/upload"))
    aa = {"success": 1, "file": {"url": "http://www.choozmo.com/images/logo%20%281%29.webp", }}
    return aa


@upload_app.route('/backstage/upload/<path:filepath>', methods=['GET'])
def upload_get(filepath):
    # print(filepath)
    #print(os.getcwd() + "/backstage/upload/"+filepath, filepath[filepath.rfind("/")+1:])
    #aa = {"success" : 1,"file": { "url" : "https://www.tesla.com/tesla_theme/assets/img/_vehicle_redesign/roadster_and_semi/roadster/hero.jpg", } }
    # return redirect(url_for('upload.upload_get',filename=filename), code=301)
    # return send_from_directory(os.getcwd() + "/backstage/upload/"+filepath, filepath[filepath.rfind("/")+1:])

    return send_file(os.getcwd() + "/backstage/upload/"+filepath)


@upload_app.route('/backstage/getimage/<path:filepath>', methods=['POST', 'GET'])
def get_image(filepath):

    # print(request.get_json()['url'])
    sitepath = UPLOAD_PATH_MAP[0][0] + filepath + "/img/"
    oimgtype = str(request.get_json()['url'])[str(request.get_json()['url']).rfind(".")+1:]
    oimgtypeName = oimgtype
    if oimgtype.lower() == 'jpg':
        oimgtypeName = 'jpeg'
    # fname = str(uuid.uuid4()) + str(request.get_json()
    #                                ['url'])[str(request.get_json()['url']).rfind("."):]
    wfname = str(uuid.uuid4()) + ".webp"
    owfname = wfname.replace('webp', oimgtype)
    wpath = os.getcwd() + "/backstage/upload/" + filepath+"/img/"
    owpath = os.getcwd() + "/backstage/upload/" + filepath+"/img/orig/"
    """ fullpath = wpath + fname
    if not os.path.exists(wpath):
        os.makedirs(wpath)
    f = open(fullpath, 'wb')
    f.write(requests.get(request.get_json()['url']).content)
    f.close() """
    if not os.path.exists(wpath):
        os.makedirs(wpath)
    if not os.path.exists(owpath):
        os.makedirs(owpath)
    if not os.path.exists(sitepath):
        os.makedirs(sitepath)
    image_object = Image.open(requests.get(request.get_json()['url'], stream=True).raw)
    image_object.save(owpath+owfname, oimgtypeName)
    if image_object.size[0] > 1000:
        image_object.thumbnail(size=((1600, 1600)))
    image_object.save(wpath+wfname, 'webp')
    image_object.save(sitepath+wfname, 'webp')
    # send_file()
    #aa = {"success" : 1,"file": { "url" : "http://localhost:9000/backstage/upload/avatar1.jpg", } }
    #resp = make_response(open(os.getcwd()+ "/backstage/upload/" + fname, 'br').read(), 301)
    #resp.content_type = "image/jpeg"
    #resp.content_encoding = "Unicode"
    # return redirect(request.get_json()['url'], code=301)
    aa = {"success": 1, "file": {"url": "/backstage/upload/" +
                                 filepath+"/img/" + wfname, "width": image_object.width, "height": image_object.height}}
    return aa