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])