def filtering(method): if session.get('handlingfile_filter') is not None: img_path = os.path.join(app.config['UPLOAD_PATH'], session.get('handlingfile_filter')) filtering = Filtering(img_path) #buffered = BytesIO() if method == "mean-filter": img = filtering.mean_filter() elif method == "median-filter": img = filtering.median_filter() elif method == "gaussian-filter": img = filtering.gaussian_filter() elif method == "bilateral-filter": img = filtering.bilateral_filter() session['handlingfile_filter_tmp'] = osutils().add_tmp_in_filename( session.get('handlingfile_filter')) img.save(os.path.join(app.config['UPLOAD_PATH'], session['handlingfile_filter_tmp']), format="PNG") img_str = Basicprocess().img2base64(img) return jsonify({'code': 200, 'img64': img_str}) else: return jsonify({'code': 404})
def basicprocess(method): if session.get('handlingfile_basicprocess') is not None: radio = request.form[method] img_path = os.path.join(app.config['UPLOAD_PATH'], session.get('handlingfile_basicprocess')) #basic_process = #buffered = BytesIO() if method == "contrast": img = Basicprocess().change_contrast(img_path, float(radio)) elif method == "value": img = Basicprocess().change_value_quick(img_path, float(radio)) elif method == "saturation": img = Basicprocess().change_saturation_quick( img_path, float(radio)) elif method == "hue": img = Basicprocess().change_hue_quick(img_path, float(radio)) session['handlingfile_basicprocess_tmp'] = osutils( ).add_tmp_in_filename(session.get('handlingfile_basicprocess')) img.save(os.path.join(app.config['UPLOAD_PATH'], session['handlingfile_basicprocess_tmp']), format="PNG") #img.save(buffered, format="PNG") #img_str = base64.b64encode(buffered.getvalue()).decode('utf-8') img_str = Basicprocess().img2base64(img) return jsonify({'code': 200, 'img64': img_str}) else: return jsonify({'code': 404})
def dehaze(): if session.get('handlingfile_dehaze') is not None: img_path = os.path.join(app.config['UPLOAD_PATH'], session.get('handlingfile_dehaze')) img = Dehaze().deHaze(img_path) #buffered = BytesIO() #img.save(buffered, format="PNG") #img_str = base64.b64encode(buffered.getvalue()).decode('utf-8') session['handlingfile_dehaze_tmp'] = osutils().add_tmp_in_filename( session.get('handlingfile_dehaze')) img.save(os.path.join(app.config['UPLOAD_PATH'], session['handlingfile_dehaze_tmp']), format="PNG") img_str = Basicprocess().img2base64(img) return jsonify({'code': 200, 'img64': img_str}) else: return jsonify({'code': 404})
def save(method): session_item = "handlingfile_" + method + "_tmp" if session.get(session_item) is not None: filename = session.get(session_item) img = Img(filename) db.session.add(img) db.session.commit() upload_filename = str(session.get('userid')) + '_' + str( img.imgid) + osutils().get_extension(filename) upload_path = os.path.join(app.config['UPLOAD_PATH'], session[session_item]) qiniuutils().upload(upload_filename, upload_path) userimg = UserImg(userid=session.get('userid'), imgid=img.imgid, filename=upload_filename) db.session.add(userimg) db.session.commit() return jsonify({'code': 200}) return jsonify({'code': 404})
def colorizer(): if request.method == 'POST': if session.get('handlingfile_colorizer') is not None: imgbase64 = request.values.get('image').split(",")[1] #basicprocess=Basicprocess() mark = Basicprocess().base64_to_rgb(imgbase64) img_path = os.path.join(app.config['UPLOAD_PATH'], session.get('handlingfile_colorizer')) img = cl.colorize(img_path, mark) #buffered=BytesIO() #img.save(buffered, format="PNG") #img_str = base64.b64encode(buffered.getvalue()).decode('utf-8') session['handlingfile_colorizer_tmp'] = osutils( ).add_tmp_in_filename(session.get('handlingfile_colorizer')) img.save(os.path.join(app.config['UPLOAD_PATH'], session['handlingfile_colorizer_tmp']), format="PNG") img_str = Basicprocess().img2base64(img) return jsonify({'code': 200, 'img64': img_str}) else: return jsonify({'code': 404}) if request.method == 'GET': return jsonify({'code': 0, 'progress': cl.get_progress()})