item = queue.next("Host") def failed(queue,item): try: queue.failed(item) except Exception as e: logger.error("Problem with reaching NI, got error: %s", e) def main(): args = CLI().options() try: config = SafeConfigParser() config.readfp(open(args.config)) except IOError as (errno, strerror): logger.error("Config file '%s' is missing", args.config) return None ## ready :) api_user = config.get("NI", "api_user") api_key = config.get("NI", "api_key") queue_url = url_concat(config.get("NI", "url"), "scan_queue/") queue = Queue(queue_url, api_user, api_key) nerds_url = url_concat(config.get("NI", "url"), "nerds/") nerds_api = NerdsApi(nerds_url, api_user, api_key) process_host(queue, nerds_api) if __name__ == "__main__": main()
def test_slash_on_base_and_chunk(self): result = url_concat("https://nordu.net/test/","/some/path") self.assertEqual(result, "https://nordu.net/test/some/path")
def test_no_slashses(self): result = url_concat("https://nordu.net/test","some/path") self.assertEqual(result, "https://nordu.net/test/some/path")
def test_only_chunk(self): result = url_concat("https://nordu.net/test","/some/path") self.assertEqual(result, "https://nordu.net/test/some/path")