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()})