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)
