def post(self):
name = request.data.decode('utf-8')
if current_user.is_sper == False:
return reponse(message=MessageEnum.user_not_permision.value[1],
code=MessageEnum.user_not_permision.value[0])
if name == '':
return reponse(code=MessageEnum.project_cannot_empty.value[0],
message=MessageEnum.project_cannot_empty.value[1])
projec = Project.query.filter_by(project_name=name,
status=False).first()
if projec:
return reponse(code=MessageEnum.project_only_one.value[0],
message=MessageEnum.project_only_one.value[1])
new_moel = Project(project_name=name, project_user_id=current_user.id)
try:
db.session.add(new_moel)
db.session.commit()
# testgroup = TestGroup(adduser=current_user.id,
# addtime=datetime.datetime.now(),
# updatetime=datetime.datetime.now(),
# updateuser=current_user.id,
# name='黑名单', projectid=new_moel.id)
# db.session.add(testgroup)
db.session.commit()
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1])
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(code=MessageEnum.project_add_error.value[0],
message=MessageEnum.project_add_error.value[1])
def dispatch_request(self):
if chckuserpermisson() is False:
return reponse(code=MessageEnum.permiss_is_ness.value[0],
message=MessageEnum.permiss_is_ness.value[1],
data='')
userjobnum = request.get_json()
try:
id = int(userjobnum['id'])
job_num = int(userjobnum['jobnum'])
except Exception as e:
logger.exception(e)
return reponse(code=MessageEnum.activ_is_int.value[0],
message=MessageEnum.activ_is_int.value[1])
user = User.query.filter_by(id=id, status=False).first()
if not user:
return reponse(
code=MessageEnum.login_user_not_exict_message.value[0],
message=MessageEnum.login_user_not_exict_message.value[1])
try:
user_job = User.query.filter_by(jobnum=job_num).first()
if user_job:
return reponse(code=MessageEnum.activi_user_jobnum.value[0],
message=MessageEnum.activi_user_jobnum.value[1])
except Exception as e:
logger.exception(e)
pass
if (user.jobnum is None or user.jobnum == "None"):
user.jobnum = job_num
db.session.add(user)
db.session.commit()
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1],
data='')
return reponse(code=MessageEnum.activi_user_jobnum_is.value[0],
message=MessageEnum.activi_user_jobnum_is.value[1])
def testapi(self):
try:
if self.fangsh == 'POST' or self.fangsh == 'post':
response, spend = self.requ.post(url=self.url,
params=self.param,
headers=self.headers)
elif self.fangsh == 'GET' or self.fangsh == 'get':
response, spend = self.requ.get(url=self.url,
headers=self.headers,
parms=self.param)
elif self.fangsh == 'PUT' or self.fangsh == 'put':
response, spend = self.requ.putfile(url=self.url,
params=self.param,
headers=self.headers)
elif self.fangsh == 'DELETE' or self.fangsh == 'delete':
response, spend = self.requ.delfile(url=self.url,
params=self.param,
headers=self.headers)
else:
response = ""
spend = ""
return response, spend
except Exception as e:
print(e)
logger.exception(e)
response = "请求出错了"
spend = "错误"
return response, spend
def dispatch_request(self, id):
if chckuserpermisson() is False:
flash(MessageEnum.permiss_is_ness.value[1])
return redirect(request.headers.get('Referer'))
user = User.query.filter_by(username=session.get('username')).first()
if user.is_sper != 1:
flash(MessageEnum.permiss_is_ness.value[1])
return redirect(request.headers.get('Referer'))
new_ad = User.query.filter_by(id=id).first()
if new_ad.status is True:
flash(MessageEnum.free_is_again.value[1])
return redirect(url_for('home.adminuser'))
if new_ad == user:
flash(MessageEnum.ower_cannot_free_me.value[1])
return redirect(url_for('home.adminuser'))
new_ad.status = True
try:
db.session.commit()
flash(MessageEnum.free_is_success.value[1])
return redirect(url_for('home.adminuser'))
except Exception as e:
logger.exception(e)
db.session.rollback()
flash(MessageEnum.free_user_error.value[1])
return redirect(url_for('home.adminuser'))
def dispatch_request(self, id):
if chckuserpermisson() is False:
flash(MessageEnum.permiss_is_ness.value[1])
return redirect(request.headers.get('Referer'))
user = User.query.filter_by(username=session.get('username')).first()
new_ad = User.query.filter_by(id=id).first()
if new_ad != user:
if user.is_sper == 1:
new_ad.set_password('111111')
try:
db.session.commit()
msg = Message(u"密码修改通知",
sender=user.email,
recipients=user.email)
msg.body = u"密码修改成功, 你的用户名：%s，你的密码是：%s" % (user.username,
"111111")
msg.html = '<a href="http://127.0.0.1:5000/login">去登录</a>'
mail = Mail()
mail.send(msg)
flash(MessageEnum.reset_success_message.value[1])
return redirect(url_for('home.adminuser'))
except Exception as e:
logger.exception(e)
db.session.rollback()
flash(MessageEnum.user_reset_error.value[1])
return redirect(url_for('home.adminuser'))
flash(MessageEnum.user_reset_isnot_amin.value[1])
return redirect(url_for('home.adminuser'))
flash(MessageEnum.user_reset_owner.value[1])
return redirect(url_for('home.adminuser'))
def put(self):
data = request.data.decode('utf-8')
json_data = json.loads(data)
id = json_data['id']
name = json_data['name']
projec = json_data['project']
edit_mode = Model.query.filter_by(id=id, status=False).first()
if projec == '请选择':
project_one = None
else:
project_one = Project.query.filter_by(
status=False, project_name=projec).first().id
if not edit_mode:
mew = Model(model_name=name, model_user_id=current_user.id)
db.session.add(mew)
db.session.commit()
return reponse(message=MessageEnum.successs.value[1],
code=MessageEnum.successs.value[0])
edit_mode.model_name = name
edit_mode.project = project_one
try:
db.session.commit()
return reponse(message=MessageEnum.successs.value[1],
code=MessageEnum.successs.value[0])
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(message=MessageEnum.edit_model_error.value[1],
code=MessageEnum.edit_model_error.value[0])
def get(self, id):
task = Task.query.filter_by(id=id).first()
if len(task.interface.all()) <= 1:
flash(MessageEnum.task_must_be_mulite_case.value[1])
return redirect(url_for('home.timingtask'))
try:
time_start = eval(task.taskstart)
day_week = time_start['day_of_week']
hour = time_start['hour']
mindes = time_start['minute']
sched.add_job(func=addtask,
id=str(id),
args=[str(id)],
trigger='cron',
day_of_week=day_week,
hour=hour,
minute=mindes,
jobstore='redis',
replace_existing=True)
task.yunxing_status = '启动'
db.session.commit()
flash(MessageEnum.task_start_success.value[1])
return redirect(url_for('home.timingtask'))
except Exception as e:
logger.exception(e)
flash(MessageEnum.task_start_eeor.value[1])
return redirect(url_for('home.timingtask'))
def excemysql(conne, Sqlmy):
try:
with conne.cursor() as conn:
conn.execute(Sqlmy)
result = conn.fetchall()
return {'code': 1, 'result': result}
except Exception as e:
logger.exception(e)
return {'code': 0, 'error': e}
def cursemsql(host, port, user, password, database):
try:
conne = connect(host=host,
port=port,
user=user,
password=password,
db=database)
return {'code': 1, 'conne': conne}
except Exception as e:
logger.exception(e)
return {'code': 0, 'error': e}
def register():
form = RegFrom()
if request.method == 'POST':
usernmae = request.form.get('username')
pasword = request.form.get('password')
setpasswod = request.form.get('se_password')
email = request.form.get('email')
jobnum = request.form.get("jobnum")
if email == "" or email is None:
flash(MessageEnum.user_email_not_none.value[1])
return render_template('home/register.html', form=form)
try:
if (str(email.split("@")[1]) != email_type):
flash(MessageEnum.email_geshi_error.value[1])
return render_template('home/register.html', form=form)
except Exception as e:
logger.exception(e)
flash(MessageEnum.user_email_error.value[1])
return render_template('home/register.html', form=form)
job_num = User.query.filter_by(jobnum=jobnum).first()
if job_num:
flash(MessageEnum.jobnum_oblg_reg_one.value[1])
return render_template('home/register.html', form=form)
if pasword != setpasswod:
flash(MessageEnum.password_not_same.value[1])
return render_template('home/register.html', form=form)
user = User.query.filter_by(username=usernmae).first()
if user:
flash(MessageEnum.user_exict.value[1])
return render_template('home/register.html', form=form)
emai = User.query.filter_by(user_email=email).first()
if emai:
flash(MessageEnum.email_exict.value[1])
return render_template('home/register.html', form=form)
new_user = User(username=usernmae, user_email=email, jobnum=job_num)
new_user.set_password(pasword)
db.session.add(new_user)
try:
db.session.commit()
# 需要邮箱发送的方法
msg = Message(u"你好", sender=email, recipients=email)
msg.body = u"欢迎你注册, 你的用户名：%s，你的密码是：%s" % (usernmae, pasword)
msg.html = '<a href="http://127.0.0.1:5000/login">去登录</a>'
mail = Mail()
mail.send(msg)
return redirect(url_for('home.login'))
except Exception as e:
logger.exception(e)
db.session.rollback()
flash(MessageEnum.user_register_error.value[1])
return render_template('home/register.html', form=form)
return render_template('home/register.html', form=form)
def post(self):
password = request.data.decode('utf-8')
user = User.query.filter_by(username=session.get('username')).first()
user.set_password(password)
try:
db.session.commit()
return reponse(code=MessageEnum.change_password_success.value[0],
data=MessageEnum.change_password_success.value[1])
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(code=MessageEnum.change_password_error.value[0],
data=MessageEnum.change_password_error.value[1])
def post(self, id):
mock = Mockserver.query.filter_by(id=id, status=False).first()
if not mock:
flash(MessageEnum.mock_check_again.value[1])
return redirect(url_for('home.mockserver'))
name = request.form['name']
desc = request.form['desc']
path = request.form['path']
methods = request.form['meth']
types = request.form['type']
headers = request.form['headers']
parm = request.form['parm']
back = request.form['back']
is_check = request.form['checkout']
is_headers = request.form['checkouheaders']
run_is = request.form['kaiqi']
if is_check == u'是':
is_check = True
else:
is_check = False
if is_headers == u'是':
is_headers = True
else:
is_headers = False
if run_is == u'是':
is_start = True
else:
is_start = False
mock.make_uers = current_user.id
mock.path = path
mock.methods = methods
mock.headers = headers
mock.description = desc
mock.fanhui = back
mock.name = name
mock.params = parm
mock.rebacktype = types
mock.status = is_start
mock.ischeck = is_check
mock.is_headers = is_headers
mock.update_time = datetime.datetime.now()
try:
db.session.commit()
flash(MessageEnum.successs.value[1])
return redirect(url_for('home.mockserver'))
except Exception as e:
logger.exception(e)
db.session.rollback()
flash(MessageEnum.mock_edit_fail.value[1])
return render_template('edit/editmock.html', mock=mock)
def get(self, id):
next = request.headers.get('Referer')
task = Task.query.filter_by(id=id).first()
try:
sched.remove_job(str(id))
task.yunxing_status = u'关闭'
db.session.commit()
flash(MessageEnum.remove_success.value[1])
return redirect(next or url_for('home.timingtask'))
except Exception as e:
logger.exception(e)
task.yunxing_status = u'创建'
db.session.commit()
flash(MessageEnum.remove_fail.value[1])
return redirect(next or url_for('home.timingtask'))
def get(self, id):
next = request.headers.get('Referer')
start = Mockserver.query.filter_by(id=id).first()
if start:
start.status = False
try:
db.session.commit()
flash(MessageEnum.mock_close_success.value[1])
return redirect(next or url_for('home.mockserver'))
except Exception as e:
logger.exception(e)
flash(MessageEnum.mock_server_close_fail.value[1])
return redirect(next or url_for('home.mockserver'))
flash(MessageEnum.mock_stop_fail.value[1])
return redirect(next or url_for('mockserver'))
def post(self, id, inte_id):
pasrm = Parameter.query.filter_by(id=int(id)).first()
interface_one = Interface.query.filter_by(id=inte_id,
status=False).first()
if interface_one is None:
flash(MessageEnum.Interface_not_exict.value[1])
return redirect(url_for('home.interface'))
if pasrm is None:
flash(MessageEnum.parame_is_not_exict.value[1])
return redirect(url_for('interface.interface_detail', id=inte_id))
type = request.form.get('type')
nuss = request.form.get('nussu')
typec = request.form.get('typechu')
desec = request.form.get('desec')
shili = request.form.get('shili')
name = request.form.get('name')
if name is None or name == '':
flash(MessageEnum.parame_name_not_empty.value[1])
return render_template('edit/edtiparmes.html',
pasrm=pasrm,
interface_one=interface_one)
if nuss == '是':
if_nuss = True
else:
if_nuss = False
if typec == '出参':
is_chu = 1
else:
is_chu = 0
if type is None or type == '':
flash(MessageEnum.parame_type_is_not_empty.value[1])
return render_template('edit/edtiparmes.html',
pasrm=pasrm,
interface_one=interface_one)
pasrm.type = is_chu
pasrm.necessary = if_nuss
pasrm.desc = desec
pasrm.default = shili
try:
flash(MessageEnum.successs.value[1])
db.session.commit()
return redirect(url_for('interface.interface_detail', id=inte_id))
except Exception as e:
logger.exception(e)
flash(MessageEnum.edit_fial.value[1])
return render_template('edit/edtiparmes.html',
pasrm=pasrm,
interface_one=interface_one)
def delete(self):
data = request.data.decode('utf-8')
proje = Project.query.filter_by(id=data, status=False).first()
if not proje:
return reponse(message=MessageEnum.project_not_exict.value[1],
code=MessageEnum.project_not_exict.value[0])
proje.status = True
try:
db.session.commit()
return reponse(message=MessageEnum.successs.value[1],
code=MessageEnum.successs.value[0])
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(message=MessageEnum.delete_fail.value[1],
code=MessageEnum.delete_fail.value[0])
def post(self):
data = request.get_json()
name = data['name']
password = data['password']
work = data['work']
project = data['project']
email = data['email']
use = User.query.filter_by(username=name).first()
if use:
return reponse(message=MessageEnum.user_is_exict.value[1],
code=MessageEnum.user_is_exict.value[0])
emai = User.query.filter_by(user_email=str(email)).first()
if emai:
return reponse(message=MessageEnum.email_only_one.value[1],
code=MessageEnum.email_only_one.value[0])
wrok = Work.query.filter_by(name=work).first()
new_user = User(username=name, user_email=email)
new_user.set_password(password)
new_user.work_id = wrok.id
db.session.add(new_user)
try:
db.session.commit()
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(message=MessageEnum.model_edit_fial.value[1],
code=MessageEnum.model_edit_fial.value[0])
if len(project) <= 0:
return reponse(message=MessageEnum.successs.value[1],
code=MessageEnum.successs.value[0])
else:
try:
user_id = User.query.filter_by(username=name).first()
for proj in project:
project_one = Project.query.filter_by(
project_name=proj).first()
quanxian = Quanxian(project=project_one.id, rose=1)
quanxian.user.append(user_id)
db.session.add(quanxian)
db.session.commit()
return reponse(message=MessageEnum.successs.value[1],
code=MessageEnum.successs.value[0])
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(message=MessageEnum.model_edit_fial.value[1],
code=MessageEnum.model_edit_fial.value[0])
def put(self):
data = request.get_json()
json_data = data
project = json_data['project']
id = json_data['id']
url = json_data['url']
desc = json_data['desc']
name = json_data['name']
host = json_data['host']
port = json_data['port']
usernmae = json_data['username']
password = json_data['password']
project = Project.query.filter_by(project_name=project).first()
event = Interfacehuan.query.filter_by(id=id).first()
if not event:
newevent = Interfacehuan(url=url,
desc=desc,
project=project.id,
database=name,
databaseuser=usernmae,
databasepassword=password,
dbhost=host,
dbport=port,
make_user=current_user.id)
db.session.add(newevent)
db.session.commit()
return reponse(message=MessageEnum.successs.value[1],
code=MessageEnum.successs.value[0])
event.url = url
event.desc = desc
event.database = name
event.databaseuser = usernmae
event.datebasepassword = password
event.dbhost = host
event.dbport = port
event.project = project.id
event.make_user = current_user.id
try:
db.session.commit()
return reponse(message=MessageEnum.successs.value[1],
code=MessageEnum.successs.value[0])
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(message=MessageEnum.edit_mock_error.value[1],
code=MessageEnum.edit_mock_error.value[0])
def get(self, page=1):
if current_user.is_sper is True:
project = Project.query.filter_by(status=False).all()
else:
project = []
for projec in current_user.quanxians:
project.append(projec.projects)
projects_lsit = fenye_list(Ob_list=project, split=PageShow)
pages = range(1, len(projects_lsit) + 1)
try:
pyth_post1 = projects_lsit[int(page) - 1]
return render_template('home/test_result.html',
projects=pyth_post1,
pages=pages)
except Exception as e:
logger.exception(e)
return redirect(url_for('home.test_result'))
def post(self):
data = request.get_json()
task_time = {
'type': 'cron',
'day_of_week': data['week'],
'hour': data['hour'],
'minute': data['minx']
}
taskname_is = Task.query.filter_by(taskname=data['taskname']).first()
testevent = Interfacehuan.query.filter_by(
url=data['testevent']).first()
if not testevent:
return reponse(code=MessageEnum.task_event_not_exict.value[0],
message=MessageEnum.task_event_not_exict.value[1],
data='')
if taskname_is:
return reponse(code=MessageEnum.task_name_not_same.value[0],
message=MessageEnum.task_name_not_same.value[1],
data='')
procjt = Project.query.filter_by(project_name=data['projects'],
status=False).first()
if not procjt:
return reponse(
code=MessageEnum.task_project_is_not_exict.value[0],
message=MessageEnum.task_project_is_not_exict.value[1],
data='')
new_task = Task(taskname=data['taskname'],
taskstart=str(task_time),
taskrepor_to=data['to_email'],
taskrepor_cao=data['cao_email'],
task_make_email=data['weihu'],
makeuser=current_user.id,
prject=procjt.id,
testevent=testevent.id)
db.session.add(new_task)
try:
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1],
data='')
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(code=MessageEnum.task_add_fail.value[0],
message=MessageEnum.task_add_fail.value[1],
data='')
def get(self, id):
passem = Parameter.query.filter_by(id=id, status=False).first()
if not passem:
flash(MessageEnum.parame_is_not_exict.value[1])
return redirect(
url_for('interface.interface_detail', id=passem.interfaces.id))
passem.status = True
try:
db.session.commit()
flash(MessageEnum.successs.value[1])
return redirect(
url_for('interface.interface_detail', id=passem.interfaces.id))
except Exception as e:
logger.exception(e)
db.session.rollback()
flash(MessageEnum.delete_fail.value[1])
return redirect(
url_for('interface.interface_detail', id=passem.interfaces.id))
def get(self, id):
next = request.headers.get('Referer')
task_one = Task.query.filter_by(id=id).first()
if not task_one:
flash(MessageEnum.delete_not_exict.value[1])
return redirect(next or url_for('home.timingtask'))
if task_one.status is True:
flash(MessageEnum.task_is_delete.value[1])
return redirect(next or url_for('home.timingtask'))
task_one.status = True
try:
db.session.commit()
flash(MessageEnum.task_delete_success.value[1])
return redirect(next or url_for('home.timingtask'))
except Exception as e:
logger.exception(e)
db.session.rollback()
flash(MessageEnum.task_delete_fail.value[1])
return redirect(next or url_for('home.timingtask'))
def post(self, id):
if current_user.is_sper is True:
projects = Project.query.filter_by(status=False).all()
else:
projects = []
id = []
for i in current_user.quanxians:
if (i.projects in id) is False:
if i.projects.status is False:
projects.append(i.projects)
id.append(i.projects)
task_one = Task.query.filter_by(id=id).first()
procject_test = request.form.get('project')
if procject_test == '':
flash(MessageEnum.not_add_project.value[1])
return render_template('add/addtestcasefortask.html',
task_one=task_one,
procjets=projects)
test_yongli = request.form.getlist('testyongli')
if test_yongli == '':
flash(MessageEnum.project_not_case.value[1])
return render_template('add/addtestcasefortask.html',
task_one=task_one,
procjets=projects)
for oldtask in task_one.interface.all():
task_one.interface.remove(oldtask)
for yongli in test_yongli:
task_yong = InterfaceTest.query.filter_by(id=yongli).first()
if task_yong.status is True:
continue
else:
task_one.interface.append(task_yong)
task_one.prject = procject_test
db.session.add(task_one)
try:
db.session.commit()
flash(MessageEnum.task_update_case.value[1])
return redirect(url_for('home.timingtask'))
except Exception as e:
logger.exception(e)
flash(MessageEnum.task_update_case_error.value[1])
return redirect(url_for('home.timingtask'))
def post(self, id):
task_one = Task.query.filter_by(id=id).first()
taskname = request.form['taskname']
to_email_data = request.form['to_email']
cao_email = request.form['cao_email']
week = request.form['week']
hour = request.form['hours']
minute = request.form['minute']
task_time = {
'type': 'cron',
'day_of_week': week,
'hour': hour,
'minute': minute
}
weihu = request.form['weihu']
if taskname == '':
flash(MessageEnum.task_name_not_none.value[1])
return render_template('add/addtimingtasks.html')
if to_email_data == '':
flash(MessageEnum.task_recevier.value[1])
return render_template('add/addtimingtasks.html')
if weihu == '':
flash(MessageEnum.task_user.value[1])
return render_template('add/addtimingtasks.html')
task_one.taskname = taskname
task_one.taskrepor_to = to_email_data
task_one.taskrepor_cao = cao_email
task_one.task_make_email = weihu
task_one.makeuser = current_user.id
task_one.taskstart = str(task_time)
try:
db.session.commit()
flash(MessageEnum.task_edit.value[1])
return redirect(url_for('home.timingtask'))
except Exception as e:
logger.exception(e)
db.session.rollback()
flash(MessageEnum.task_edit_fail.value[1])
return redirect(url_for('home.timingtask'))
def put(self):
data = request.data.decode('utf-8')
json_data = json.loads(data)
id = json_data['id']
name = json_data['name']
prohect = Project.query.filter_by(id=id).first()
if not prohect:
new = Project(project_name=name, project_user_id=current_user.id)
db.session.add(new)
db.session.commit()
return reponse(message=MessageEnum.successs.value[1],
code=MessageEnum.successs.value[0])
prohect.project_name = name
try:
db.session.commit()
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1])
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(code=MessageEnum.eidt_excption.value[0],
message=MessageEnum.eidt_excption.value[1])
def get(self, url, headers, parms): # get消息
try:
self.r = requests.get(url, headers=headers, params=parms, timeout=Interface_Time_Out)
self.r.encoding = 'UTF-8'
spend = self.r.elapsed.total_seconds()
json_response = json.loads(self.r.text)
return json_response, spend
except exceptions.Timeout:
logger.exception(exceptions.Timeout)
return {'get请求出错': "请求超时"}
except exceptions.InvalidURL:
logger.exception(exceptions.InvalidURL)
return {'get请求出错': "非法url"}
except exceptions.HTTPError:
logger.exception(exceptions.HTTPError)
return {'get请求出错': "http请求错误"}
except Exception as e:
logger.exception(e)
return {'get请求出错': "错误原因:%s" % e}
def putfile(self, url, params, headers): # put请求
try:
self.rdata = json.dumps(params)
me = requests.put(url, self.rdata, headers=headers,
timeout=Interface_Time_Out)
json_response = json.loads(me.text)
spend = me.elapsed.total_seconds()
return json_response, spend
except exceptions.Timeout:
logger.exception(exceptions.Timeout)
return {'put请求出错': "请求超时"}
except exceptions.InvalidURL:
logger.exception(exceptions.InvalidURL)
return {'put请求出错': "非法url"}
except exceptions.HTTPError:
logger.exception(exceptions.HTTPError)
return {'put请求出错': "http请求错误"}
except Exception as e:
logger.exception(e)
return {'put请求出错': "错误原因:%s" % e}
def dispatch_request(self):
if chckuserpermisson() is False:
return reponse(code=MessageEnum.permiss_is_ness.value[0],
message=MessageEnum.permiss_is_ness.value[1],
data='')
projec = request.get_json()
try:
username = projec['username']
por = projec['url']
if por == '':
return reponse(code=MessageEnum.select_project_not.value[0],
message=MessageEnum.select_project_not.value[1],
data='')
pan_user = User.query.filter_by(username=username).first()
if not pan_user:
return reponse(
code=MessageEnum.login_user_not_exict_message.value[0],
message=MessageEnum.login_user_not_exict_message.value[1],
data='')
if pan_user.is_sper is True:
return reponse(
code=MessageEnum.super_admin_not_set_project.value[0],
message=MessageEnum.super_admin_not_set_project.value[1],
data='')
pand_por = Project.query.filter_by(project_name=por).first()
if not pand_por:
return reponse(
code=MessageEnum.set_project_bot_exict.value[0],
message=MessageEnum.set_project_bot_exict.value[1],
data='')
pro_per = Quanxian.query.filter_by(project=pand_por.id).all()
oneadmin = []
for i in pro_per:
if i.rose == 2:
oneadmin.append(i.user.all())
if [pan_user] in oneadmin:
return reponse(code=MessageEnum.set_is_admin.value[0],
message=MessageEnum.set_is_admin.value[1])
if (len(oneadmin)) > OneAdminCount:
return reponse(code=MessageEnum.project_admin_many.value[0],
message=MessageEnum.project_admin_many.value[1])
for roses in pan_user.quanxians:
if roses.project == pand_por.id:
roses.rose = 2
try:
db.session.commit()
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1])
except Exception as e:
logger.exception(e)
db.session.rollback()
return reponse(code=MessageEnum.set_fail.value[0],
message=MessageEnum.set_fail.value[1],
data='')
except Exception as e:
logger.exception(e)
return reponse(
code=MessageEnum.set_project_admin_exception.value[0],
message=MessageEnum.set_project_admin_exception.value[1] +
'原因是：%s' % e,
data='')
def get_token_data(path):
huoqupath = Mockserver.query.filter_by(path=path, status=True).first()
if not huoqupath:
abort(404)
heders = request.headers
method = request.method
if method.lower() != huoqupath.methods:
return jsonreponse(code=MessageEnum.request_method.value[0],
message=MessageEnum.request_method.value[1])
try:
token = heders['token']
if token == 'Fetext_token_system':
paerm = request.values.to_dict()
if dict_par(paerm, huoqupath.params) == True:
if huoqupath.rebacktype == 'json':
try:
json_fan = json.dumps(huoqupath.fanhui)
return jsonreponse(
code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1],
data=json_fan)
except Exception as e:
logger.exception(e)
return jsonreponse(
code=MessageEnum.resquest_return_not_json.value[0],
message=MessageEnum.resquest_return_not_json.
value[1])
else:
return jsonreponse(
code=MessageEnum.request_method_not_supprot.value[0],
message=MessageEnum.request_method_not_supprot.value[1]
)
else:
return jsonreponse(
code=MessageEnum.method_parame_not_right.value[0],
message=MessageEnum.method_parame_not_right.value[1])
except Exception as e:
logger.exception(e)
if huoqupath.is_headers == True:
if comp_dict(heders, huoqupath.headers) == True:
if huoqupath.ischeck == True:
paerm = request.values.to_dict()
if dict_par(paerm, huoqupath.params) == True:
if huoqupath.rebacktype == 'json':
try:
json_return = json.dumps(huoqupath.fanhui)
return jsonreponse(
code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1],
data=json_return)
except Exception as e:
logger.exception(e)
return jsonreponse(
code=MessageEnum.resquest_return_not_json.
value[0],
message=MessageEnum.
resquest_return_not_json.value[1])
elif huoqupath.rebacktype == 'xml':
response = make_response(huoqupath.fanhui)
response.content_type = 'application/xml'
return response
else:
return jsonreponse(
code=MessageEnum.request_method_not_supprot.
value[0],
message=MessageEnum.request_method_not_supprot.
value[1])
else:
return jsonreponse(
code=MessageEnum.method_parame_not_right.value[0],
message=MessageEnum.method_parame_not_right.
value[1])
else:
if huoqupath.rebacktype == 'json':
try:
json_return = json.dumps(huoqupath.fanhui)
return jsonreponse(
code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1],
data=json_return)
except Exception as e:
logger.exception(e)
return jsonreponse(
code=MessageEnum.resquest_return_not_json.
value[0],
message=MessageEnum.resquest_return_not_json.
value[1])
elif huoqupath.rebacktype == 'xml':
response = make_response(huoqupath.fanhui)
response.content_type = 'application/xml'
return response
return jsonreponse(
code=MessageEnum.request_method_not_supprot.value[0],
message=MessageEnum.request_method_not_supprot.value[1]
)
return jsonreponse(code=MessageEnum.request_scre.value[0],
message=MessageEnum.request_scre.value[1])
else:
if huoqupath.ischeck == True:
paerm = request.values.to_dict()
if dict_par(paerm, huoqupath.params) == True:
if huoqupath.rebacktype == 'json':
try:
json_return = json.dumps(huoqupath.fanhui)
return jsonreponse(
code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1],
data=json_return)
except Exception as e:
logger.exception(e)
return jsonreponse(
code=MessageEnum.resquest_return_not_json.
value[0],
message=MessageEnum.resquest_return_not_json.
value[1])
elif huoqupath.rebacktype == 'xml':
response = make_response(huoqupath.fanhui)
response.content_type = 'application/xml'
return response
else:
return jsonreponse(code=MessageEnum.
request_method_not_supprot.value[0],
message=MessageEnum.
request_method_not_supprot.value[1])
return jsonreponse(
code=MessageEnum.method_parame_not_right.value[0],
message=MessageEnum.method_parame_not_right.value[1])
if huoqupath.rebacktype == 'json':
try:
json_return = json.dumps(huoqupath.fanhui)
return jsonreponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1],
data=json_return)
except Exception as e:
logger.exception(e)
return jsonreponse(
code=MessageEnum.resquest_return_not_json.value[0],
message=MessageEnum.resquest_return_not_json.value[1])
elif huoqupath.rebacktype == 'xml':
response = make_response(huoqupath.fanhui)
response.content_type = 'application/xml'
return response
else:
return jsonreponse(
code=MessageEnum.request_method_not_supprot.value[0],
message=MessageEnum.request_method_not_supprot.value[1])
def post(self):
try:
data = request.get_json()
config = GeneralConfiguration.query.filter_by(
name=data['name']).first()
if config:
return reponse(code=MessageEnum.common_is_same.value[0],
message=MessageEnum.common_is_same.value[1],
data='')
if data['type'] == "key-value":
newconfig = GeneralConfiguration(user=current_user,
style=0,
key=data["key"],
name=data['name'])
db.session.add(newconfig)
db.session.commit()
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1])
elif data['type'] == 'token':
newconfig = GeneralConfiguration(user=current_user,
style=1,
name=data['name'],
token_method=data['method'],
token_parame=data['parame'],
token_url=data['url'])
db.session.add(newconfig)
db.session.commit()
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1])
elif data['type'] == 'sql':
testevnet = Interfacehuan.query.filter_by(
id=int(data['eventid'])).first()
if not testevnet:
return reponse(
code=MessageEnum.testeveirment_not_exict.value[0],
message=MessageEnum.testeveirment_not_exict.value[1])
newconfig = GeneralConfiguration(user=current_user,
style=1,
name=data['name'],
testevent=testevnet,
sqlurl=data['sql'])
db.session.add(newconfig)
db.session.commit()
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1])
elif data['type'] == 'http请求':
newconfig = GeneralConfiguration(user=current_user,
style=1,
name=data['name'],
request_method=data['method'],
request_parame=data['parame'],
request_url=data['url'])
db.session.add(newconfig)
db.session.commit()
return reponse(code=MessageEnum.successs.value[0],
message=MessageEnum.successs.value[1])
else:
return reponse(
code=MessageEnum.common_gene_not_support.value[0],
message=MessageEnum.common_gene_not_support.value[1],
data='')
except Exception as e:
logger.exception(e)
return reponse(code=MessageEnum.parames_not_null.value[0],
data=MessageEnum.parames_not_null.value[1])
