import os
from flask import Flask, request, redirect, url_for
from werkzeug import secure_filename
from damatuWeb import DamatuApi
from settings import UPLOAD_FOLDER, ALLOWED_EXTENSIONS, USER, PASSWD
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
def allowed_file(filename):
"""
根据文件名判断是否允许的文件
"""
return '.' in filename and \
filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
@app.route("/", methods=['GET', 'POST'])
def index():
"""
上传图片,交给打码兔打码,返回验证码
上传的图片以内网ip+png命名
"""
try:
if request.method == 'POST':
file = request.files['file']
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
file_l = os.path.join(app.config['UPLOAD_FOLDER'], filename)
res = file.save(file_l)
dmt = DamatuApi(USER, PASSWD)
try:
code = dmt.decode(file_l, 200)
return code
except Exception as error:
return error
else:
return "Server OK!"
except Exception as error:
return error
if __name__ == "__main__":
app.run(host='0.0.0.0', port=12345, debug=True, threaded=True)