def setup_reports(self): if self.arguments.output_file: output_file = FileUtils.get_abs_path(self.arguments.output_file) self.output.output_file(output_file) else: if self.batch: file_name = "BATCH" file_name += self.get_output_extension() directory_path = self.batch_directory_path else: local_requester = Requester(self.url_list[0]) file_name = ("{}_".format( local_requester.base_path.replace(os.path.sep, ".")[:-1])) file_name += time.strftime("%y-%m-%d_%H-%M-%S") file_name += self.get_output_extension() directory_path = FileUtils.build_path(self.save_path, local_requester.host) output_file = FileUtils.build_path(directory_path, file_name) if FileUtils.exists(output_file): i = 2 while FileUtils.exists(output_file + "_" + str(i)): i += 1 output_file += "_" + str(i) if not FileUtils.exists(directory_path): FileUtils.create_directory(directory_path) if not FileUtils.exists(directory_path): self.output.error( "Couldn't create the reports folder at {}".format( directory_path)) sys.exit(1) self.output.output_file(output_file) if self.arguments.output_file and self.arguments.output_format: self.report_manager = ReportManager(self.arguments.output_format, self.arguments.output_file) elif self.arguments.output_format: self.report_manager = ReportManager(self.arguments.output_format, output_file) else: self.report_manager = ReportManager("plain", output_file)
def setupReports(self): if self.arguments.outputFile is not None: outputFile = FileUtils.get_abs_path(self.arguments.outputFile) self.output.outputFile(outputFile) else: if self.batch: fileName = "BATCH" fileName += self.getOutputExtension() directoryPath = self.batchDirectoryPath else: localRequester = Requester(self.urlList[0]) fileName = ('{}_'.format( localRequester.basePath.replace(os.path.sep, ".")[:-1])) fileName += time.strftime('%y-%m-%d_%H-%M-%S') fileName += self.getOutputExtension() directoryPath = FileUtils.build_path(self.savePath, 'reports', localRequester.host) outputFile = FileUtils.build_path(directoryPath, fileName) if FileUtils.exists(outputFile): i = 2 while FileUtils.exists(outputFile + "_" + str(i)): i += 1 outputFile += "_" + str(i) if not FileUtils.exists(directoryPath): FileUtils.create_directory(directoryPath) if not FileUtils.exists(directoryPath): self.output.error( "Couldn't create the reports folder at {}".format( directoryPath)) sys.exit(1) self.output.outputFile(outputFile) if self.arguments.outputFile and self.arguments.outputFormat: self.reportManager = ReportManager(self.arguments.outputFormat, self.arguments.outputFile) elif self.arguments.outputFormat: self.reportManager = ReportManager(self.arguments.outputFormat, outputFile) else: self.reportManager = ReportManager("plain", outputFile)