def setobject_action(self, dbid, action): api_dict = commen.get_api("/DDL.json") api_dict = api_dict["oracle"]["objectpermission"] print(api_dict["url"]) try: api_request(api_dict['url'] % (dbid, action), api_dict['header'], api_dict['method']) except Exception as e: assert e
def clear(self): payload = {} res = api_request(api_dict["clear"]["url"], api_dict["clear"]["header"], api_dict["clear"]["method"], payload=payload) return res
def diy_rule_add(self, dbType=None, name=None, risk_level=None, status=None, ruleType=None): param = self.param_dict['add_diy_rule'] if dbType != None: if dbType == "dm": param['body']['dbType'] = "dameng" else: param['body']['dbType'] = dbType if name != None: param['body']['chsName'] = name if risk_level != None: param['body']['riskLevel'] = risk_level if ruleType != None: param['body']['ruleType'] = ruleType if status != None: param['body']['vpStatus'] = status try: response = api_request(api_url=param['url'], headers=param['header'], method=param['method'], payload=param['body']) return response except Exception as e: LOG.info(e)
def nopage(self): try: response = api_request(api_url=self.param_dict['nopage']['uri'], method=self.param_dict['nopage']['method']) return response except Exception as e: LOG.info(e)
def update_access(self, id=None, action=None, cyc=None, rate=None, riskLevel=None, audit=None): self.param_dict['update']['body']['dbIds'][0] = int(id) self.param_dict['update']['body']['access']['status'] = 1 if action != None: self.param_dict['update']['body']['access']['action'] = action if cyc != None: self.param_dict['update']['body']['access']['cyc'] = cyc if rate != None: self.param_dict['update']['body']['access']['rate'] = rate if riskLevel != None: self.param_dict['update']['body']['access'][ 'riskLevel'] = riskLevel if audit != None: self.param_dict['update']['body']['access']['audit'] = audit response = api_request(api_url=self.param_dict['update']['url'], headers=self.param_dict['update']['header'], method=self.param_dict['update']['method'], payload=self.param_dict['update']['body']) return response
def update(self, dispose=None, featuresStatus=None, virStatus=None, id=None, dataMaskStatusOM=None): try: if dispose != None: self.param_dict['update']['body']['rule']['dispose'] = dispose if featuresStatus != None: self.param_dict['update']['body'][ 'featuresStatus'] = featuresStatus if dataMaskStatusOM != None: self.param_dict['update']['body'][ 'dataMaskStatusOM'] = dataMaskStatusOM if virStatus != None: self.param_dict['update']['body']['virStatus'] = virStatus self.param_dict['update']['body']['dbIds'][0] = int(id) response = api_request(api_url=self.param_dict['update']['url'], headers=self.param_dict['update']['header'], method=self.param_dict['update']['method'], payload=self.param_dict['update']['body']) return response except Exception as e: LOG.info(e)
def state_change(self, state, IDlist): payload = {"cmdType": state, "ids": IDlist} res = api_request(api_dict["status"]["url"], api_dict["status"]["header"], api_dict["status"]["method"], payload=payload) res = json.loads(res.content.decode("utf-8")) return res
def select_rule(self, byparam, param): try: Param = self.param_dict['select_diy_rule'] response = api_request(api_url=Param['urlParam'][byparam] % param, headers=Param['header'], method=Param['method']) return response except Exception as e: LOG.info(e)
def view_rule(self, id): try: param = self.param_dict['view_diy_rule'] response = api_request(api_url=param['url'] + str(id), headers=param['header'], method=param['method']) return response except Exception as e: LOG.info(e)
def setrisk(self, dbid=1, auditLevel=3, status=0, dangerownerId=1): payload = { "dbid": dbid, "assetOwner": "", "ownerType": "database account", "additionRule": "", "auditLevel": auditLevel, "status": status, "dangerownerId": dangerownerId, "part": "danger" } api_dict = commen.get_api("/DDL.json") api_dict = api_dict["riskconfig"] try: api_request( api_dict['url'] % (payload["dbid"], payload["ownerType"], payload["auditLevel"], payload["status"], payload["dangerownerId"], payload["part"]), api_dict['header'], api_dict['method']) except Exception as e: assert e
def operate_rule(self, operate, id): try: self.param_dict['mutiOperate_diy_rule']['body'][operate][ 'ids'].append(id) param = self.param_dict['mutiOperate_diy_rule'] response = api_request(api_url=param['url'], headers=param['header'], method=param['method'], payload=param['body'][operate]) return response except Exception as e: LOG.info(e) pass
def select(self, dbname=None, featuresStatus=None, virStatus=None): response = None try: if dbname != None and featuresStatus == None and virStatus == None: response = api_request( api_url=self.param_dict['select']['uri']['dbname'] + dbname, method=self.param_dict['select']['method']) if featuresStatus != None and dbname == None and virStatus == None: response = api_request( api_url=self.param_dict['select']['uri']['featuresStatus'] + str(featuresStatus), method=self.param_dict['select']['method']) if virStatus != None and featuresStatus == None and dbname == None: response = api_request( api_url=self.param_dict['select']['uri']['virStatus'] + str(virStatus), method=self.param_dict['select']['method']) return response except Exception as e: LOG.info(e)
def getlist(self): res = api_request(api_dict["whitelist"]["url"], api_dict["whitelist"]["header"], api_dict["whitelist"]["method"]) res = json.loads(res.content) return res
def addruler(self, payload): res = api_request(api_dict["increase"]["url"], api_dict["increase"]["header"], api_dict["increase"]["method"], payload) res = json.loads(res.content) assert res['message'] == "success", "%s,新增白名单失败" % res['message']