Exemplo n.º 1
0
 def _handle_response_success(self):
     # pylint: disable=no-member
     if self.response.status_code in [codes.ok]:
         json_data = self.response.json()
         count = json_data.get('count', 0)
         results = json_data.get('results', [])
         if count == 0:
             # if GET is used for single scan job,
             # count doesn't exist and will be 0
             if 'id' in self.args and self.args.id:
                 data = pretty_print(json_data)
                 print(data)
             else:
                 print(_(messages.SCAN_LIST_NO_SCANS))
                 sys.exit(1)
         else:
             data = pretty_print(results)
             print(data)
         if json_data.get('next'):
             next_link = json_data.get('next')
             params = urlparse.parse_qs(urlparse.urlparse(next_link).query)
             page = params.get('page', ['1'])[0]
             if self.req_params:
                 self.req_params['page'] = page
             else:
                 self.req_params = {'page': page}
             input(_(messages.NEXT_RESULTS))
             self._do_command()
     else:
         print(_(messages.SCAN_LIST_NO_SCANS))
         sys.exit(1)
Exemplo n.º 2
0
 def _handle_response_success(self):
     json_data = self.response.json()
     if json_data == []:
         print(_(messages.SOURCE_LIST_NO_SOURCES))
     else:
         data = pretty_print(json_data)
         print(data)
Exemplo n.º 3
0
 def _handle_response_success(self):
     json_data = self.response.json()
     if len(json_data) == 1:
         cred_entry = json_data[0]
         data = pretty_print(cred_entry)
         print(data)
     else:
         print(_(messages.SOURCE_DOES_NOT_EXIST % self.args.name))
         sys.exit(1)
Exemplo n.º 4
0
 def _handle_response_success(self):
     json_data = self.response.json()
     count = json_data.get('count', 0)
     if count == 1:
         cred_entry = json_data.get('results')[0]
         data = pretty_print(cred_entry)
         print(data)
     else:
         print(_(messages.CRED_DOES_NOT_EXIST % self.args.name))
         sys.exit(1)
Exemplo n.º 5
0
 def _handle_response_success(self):
     json_data = self.response.json()
     status = pretty_print(json_data)
     if self.args.path:
         try:
             write_file(self.args.path, status)
             print(_(messages.STATUS_SUCCESSFULLY_WRITTEN))
         except EnvironmentError as err:
             err_msg = _(messages.WRITE_FILE_ERROR % (self.args.path, err))
             print(err_msg)
     else:
         print(status)
Exemplo n.º 6
0
    def _handle_response_success(self):
        file_content = None
        if self.args.output_json:
            file_content = self.response.json()
            file_content = pretty_print(file_content)
        else:
            file_content = self.response.text

        try:
            write_file(self.args.path, file_content)
            print(_(messages.REPORT_SUCCESSFULLY_WRITTEN))
        except EnvironmentError as err:
            err_msg = _(messages.WRITE_FILE_ERROR % (self.args.path, err))
            print(err_msg)
Exemplo n.º 7
0
    def _handle_response_success(self):
        json_data = self.response.json()
        count = json_data.get('count', 0)
        results = json_data.get('results', [])
        if count == 0:
            print(_(messages.SOURCE_LIST_NO_SOURCES))
        else:
            data = pretty_print(results)
            print(data)

        if json_data.get('next'):
            next_link = json_data.get('next')
            params = urlparse.parse_qs(urlparse.urlparse(next_link).query)
            page = params.get('page', ['1'])[0]
            if self.req_params:
                self.req_params['page'] = page
            else:
                self.req_params = {'page': page}
            input(_(messages.NEXT_RESULTS))
            self._do_command()
Exemplo n.º 8
0
 def _handle_response_success(self):
     json_data = self.response.json()
     data = pretty_print(json_data)
     print(data)