def addToken(self, params): try: params = json.loads(params) params.update(dict(token=app.config['TOKEN'])) params = dumps(params) except: pass return params
def prepare(self, sql, case): if sql: obj = MysqlDb(app.config['DBNAME']) value = getattr(obj, sql)() pre_data = case.get('pre_data') params = json.loads(case.get('params')) requestData = params.get('requestData') requestData.update({pre_data: value}) params.update(dict(requestData=requestData)) case.update(dict(params=dumps(params))) del obj
def transfer_data(self, case): try: params = self.addToken(case.get('params')) enable = case.get('enable') code = int(case.get('code')) headers = case.get('headers') expected = self.get_response(case.get('expected')) if params != '' and isinstance(params, str): params = self.form_to_dict( params) if 'form' in headers else dumps( self.data_to_json(params)) else: params = dumps(params) headers = self.data_to_json(headers) case.update( dict(headers=headers, params=params, enable=enable, code=code, expected=expected)) except Exception as err: logging.error("{} error: {}".format(Operator.get_current_func(), str(err))) return case