Skip to content

Commit

Permalink
Enable multiple input for -f and -o option
Browse files Browse the repository at this point in the history
Signed-off-by: Wonjae Park <j.wonjae.park@gmail.com>
  • Loading branch information
JustinWonjaePark committed Jul 9, 2024
1 parent 5fd88ab commit cfff62d
Show file tree
Hide file tree
Showing 3 changed files with 66 additions and 47 deletions.
85 changes: 50 additions & 35 deletions src/fosslight_source/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from fosslight_util.timer_thread import TimerThread
from ._help import print_version, print_help_msg_source_scanner
from ._license_matched import get_license_list_to_print
from fosslight_util.output_format import check_output_format, write_output_file
from fosslight_util.output_format import check_output_formats, write_output_file
from fosslight_util.correct import correct_with_yaml
from .run_scancode import run_scan
from .run_scanoss import run_scanoss_py
Expand Down Expand Up @@ -48,7 +48,7 @@ def main():
write_json_file = False
output_file_name = ""
print_matched_text = False
format = ""
formats = ""
selected_scanner = ""
correct_mode = True

Expand All @@ -62,7 +62,7 @@ def main():
parser.add_argument('-j', '--json', action='store_true', required=False)
parser.add_argument('-o', '--output', nargs=1, type=str, required=False, default="")
parser.add_argument('-m', '--matched', action='store_true', required=False)
parser.add_argument('-f', '--format', nargs=1, type=str, required=False)
parser.add_argument('-f', '--formats', nargs='*', type=str, required=False)
parser.add_argument('-s', '--scanner', nargs=1, type=str, required=False, default='all')
parser.add_argument('-t', '--timeout', type=int, required=False, default=120)
parser.add_argument('-c', '--cores', type=int, required=False, default=-1)
Expand All @@ -87,8 +87,8 @@ def main():
output_file_name = ''.join(args.output)
if args.matched:
print_matched_text = True
if args.format:
format = ''.join(args.format)
if args.formats:
formats = list(args.formats)
if args.scanner:
selected_scanner = ''.join(args.scanner)
if args.no_correction:
Expand All @@ -107,8 +107,9 @@ def main():
if os.path.isdir(path_to_scan):
result = []
result = run_scanners(path_to_scan, output_file_name, write_json_file, core, True,
print_matched_text, format, time_out, correct_mode, correct_filepath,
print_matched_text, formats, time_out, correct_mode, correct_filepath,
selected_scanner, path_to_exclude)

_result_log["Scan Result"] = result[1]

try:
Expand Down Expand Up @@ -138,7 +139,7 @@ def count_files(path_to_scan, path_to_exclude):


def create_report_file(_start_time, merged_result, license_list, scanoss_result, selected_scanner, need_license=False,
output_path="", output_file="", output_extension="", correct_mode=True, correct_filepath="",
output_path="", output_files=[], output_extensions=[], correct_mode=True, correct_filepath="",
path_to_scan="", path_to_exclude=[]):
"""
Create report files for given scanned result.
Expand All @@ -156,22 +157,30 @@ def create_report_file(_start_time, merged_result, license_list, scanoss_result,
output_path = os.getcwd()
else:
output_path = os.path.abspath(output_path)

if output_files and output_files[0]:
# If -o contains file name, set all output file name and dont' use default name
output_files = [output_files[0] for _ in range(len(output_extensions))]
else:
# If -o does not contains file name, set default name
while len(output_files) < len(output_extensions):
output_files.append(None)
for i, output_extension in enumerate(output_extensions):
if output_files[i] is None or output_files[i] == "":
if output_extension == _json_ext:
output_files[i] = f"fosslight_opossum_src_{_start_time}"
else:
output_files[i] = f"fosslight_report_src_{_start_time}"

if not correct_filepath:
correct_filepath = path_to_scan

if output_file == "":
if output_extension == _json_ext:
output_file = f"fosslight_opossum_src_{_start_time}"
else:
output_file = f"fosslight_report_src_{_start_time}"

cover = CoverItem(tool_name=_PKG_NAME,
start_time=_start_time,
input_path=path_to_scan,
exclude_path=path_to_exclude)
files_count, removed_files_count = count_files(path_to_scan, path_to_exclude)
cover.comment = f"Total number of files / removed files: {files_count} / {removed_files_count}"

if len(merged_result) == 0:
if files_count < 1:
cover.comment += "(No file detected.)"
Expand All @@ -190,7 +199,7 @@ def create_report_file(_start_time, merged_result, license_list, scanoss_result,
extended_header = MERGED_HEADER

if need_license:
if selected_scanner == 'scancode' or output_extension == _json_ext:
if selected_scanner == 'scancode':
sheet_list["scancode_reference"] = get_license_list_to_print(license_list)
elif selected_scanner == 'scanoss':
sheet_list["scanoss_reference"] = get_scanoss_extra_info(scanoss_result)
Expand All @@ -206,17 +215,23 @@ def create_report_file(_start_time, merged_result, license_list, scanoss_result,
sheet_list = correct_list
logger.info("Success to correct with yaml.")

output_file_without_ext = os.path.join(output_path, output_file)
success_to_write, writing_msg, result_file = write_output_file(output_file_without_ext, output_extension,
sheet_list, extended_header, "", cover)
if success_to_write:
if result_file:
logger.info(f"Output file:{result_file}")
logger.info(f'{cover.comment}')
combined_paths_and_files = [os.path.join(output_path, file) for file in output_files]
results = []
for combined_path_and_file, output_extension in zip(combined_paths_and_files, output_extensions):
if need_license:
if output_extension == _json_ext:
sheet_list["scancode_reference"] = get_license_list_to_print(license_list)
elif selected_scanner == 'scanoss':
if "scancode_reference" in sheet_list:
del sheet_list["scancode_reference"]
results.append(write_output_file(combined_path_and_file, output_extension, sheet_list, extended_header, "", cover))
for success, msg, result_file in results:
if success:
logger.info(f"Output file: {result_file}")
if cover:
logger.info(f'{cover.comment}')
else:
logger.warning(f"{writing_msg}")
else:
logger.error(f"Fail to generate result file. msg:({writing_msg})")
logger.error(f"Fail to generate result file {result_file}. msg:({msg})")


def merge_results(scancode_result=[], scanoss_result=[], spdx_downloads={}):
Expand Down Expand Up @@ -246,7 +261,7 @@ def merge_results(scancode_result=[], scanoss_result=[], spdx_downloads={}):


def run_scanners(path_to_scan, output_file_name="", write_json_file=False, num_cores=-1, called_by_cli=True,
print_matched_text=False, format="", time_out=120, correct_mode=True, correct_filepath="",
print_matched_text=False, formats=[], time_out=120, correct_mode=True, correct_filepath="",
selected_scanner='all', path_to_exclude=[]):
"""
Run Scancode and scanoss.py for the given path.
Expand All @@ -273,27 +288,28 @@ def run_scanners(path_to_scan, output_file_name="", write_json_file=False, num_c
spdx_downloads = {}
result_log = {}

success, msg, output_path, output_file, output_extension = check_output_format(output_file_name, format)
success, msg, output_path, output_files, output_extensions = check_output_formats(output_file_name, formats)

logger, result_log = init_log(os.path.join(output_path, f"fosslight_log_src_{start_time}.txt"),
True, logging.INFO, logging.DEBUG, _PKG_NAME, path_to_scan, path_to_exclude)
if output_extension != '.xlsx' and output_extension and print_matched_text:

if '.xlsx' not in output_extensions and print_matched_text:
logger.warning("-m option is only available for excel.")
print_matched_text = False

if success:
if selected_scanner == 'scancode' or selected_scanner == 'all' or selected_scanner == '':
success, result_log[RESULT_KEY], scancode_result, license_list = run_scan(path_to_scan, output_file_name,
write_json_file, num_cores, True,
print_matched_text, format, called_by_cli,
time_out, correct_mode, correct_filepath,
path_to_exclude)
print_matched_text, formats, called_by_cli,
time_out, correct_mode, correct_filepath)
if selected_scanner == 'scanoss' or selected_scanner == 'all' or selected_scanner == '':
scanoss_result = run_scanoss_py(path_to_scan, output_file_name, format, True,
write_json_file, num_cores, path_to_exclude)
scanoss_result = run_scanoss_py(path_to_scan, output_file_name, formats, True, write_json_file, num_cores, path_to_exclude)
if selected_scanner in SCANNER_TYPE:
spdx_downloads = get_spdx_downloads(path_to_scan, path_to_exclude)
merged_result = merge_results(scancode_result, scanoss_result, spdx_downloads)
create_report_file(start_time, merged_result, license_list, scanoss_result, selected_scanner,
print_matched_text, output_path, output_file, output_extension, correct_mode,
print_matched_text, output_path, output_files, output_extensions, correct_mode,
correct_filepath, path_to_scan, path_to_exclude)
else:
print_help_msg_source_scanner()
Expand All @@ -302,7 +318,6 @@ def run_scanners(path_to_scan, output_file_name="", write_json_file=False, num_c
else:
result_log[RESULT_KEY] = f"Format error. {msg}"
success = False

return success, result_log.get(RESULT_KEY, ""), merged_result, license_list, scanoss_result


Expand Down
24 changes: 14 additions & 10 deletions src/fosslight_source/run_scancode.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from ._parsing_scancode_file_item import parsing_file_item
from ._parsing_scancode_file_item import get_error_from_header
from ._license_matched import get_license_list_to_print
from fosslight_util.output_format import check_output_format
from fosslight_util.output_format import check_output_formats
from fosslight_binary.binary_analysis import check_binary

logger = logging.getLogger(constant.LOGGER_NAME)
Expand All @@ -24,7 +24,7 @@


def run_scan(path_to_scan, output_file_name="",
_write_json_file=False, num_cores=-1, return_results=False, need_license=False, format="",
_write_json_file=False, num_cores=-1, return_results=False, need_license=False, formats=[],
called_by_cli=False, time_out=120, correct_mode=True, correct_filepath="", path_to_exclude=[]):
if not called_by_cli:
global logger
Expand All @@ -41,19 +41,24 @@ def run_scan(path_to_scan, output_file_name="",
if not correct_filepath:
correct_filepath = path_to_scan

success, msg, output_path, output_file, output_extension = check_output_format(output_file_name, format)
success, msg, output_path, output_files, output_extensions = check_output_formats(output_file_name, formats)
# print("output_path | output_files | output_extensions == ", output_path, " | ", output_files, " | ", output_extensions)
# print(type(output_files), type(output_extensions))
# print("!@#$ WORKING FINE SO FAR !@#$ [RUN_SCAN]")
if success:
if output_path == "": # if json output with _write_json_file not used, output_path won't be needed.
output_path = os.getcwd()
else:
output_path = os.path.abspath(output_path)

if not called_by_cli:
if output_file == "":
if output_extension == _json_ext:
output_file = f"fosslight_opossum_src_{_start_time}"
else:
output_file = f"fosslight_report_src_{_start_time}"
while len(output_files) < len(output_extensions):
output_files.append(None)
for i, output_extension in enumerate(output_extensions):
if output_files[i] is None or output_files[i] == "":
if output_extension == _json_ext:
output_files[i] = f"fosslight_opossum_src_{_start_time}"
else:
output_files[i] = f"fosslight_report_src_{_start_time}"

if _write_json_file:
output_json_file = os.path.join(output_path, "scancode_raw_result.json")
Expand All @@ -63,7 +68,6 @@ def run_scan(path_to_scan, output_file_name="",
if not called_by_cli:
logger, _result_log = init_log(os.path.join(output_path, f"fosslight_log_src_{_start_time}.txt"),
True, logging.INFO, logging.DEBUG, _PKG_NAME, path_to_scan, path_to_exclude)

num_cores = multiprocessing.cpu_count() - 1 if num_cores < 0 else num_cores

if os.path.isdir(path_to_scan):
Expand Down
4 changes: 2 additions & 2 deletions src/fosslight_source/run_scanoss.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from datetime import datetime
import fosslight_util.constant as constant
from fosslight_util.set_log import init_log
from fosslight_util.output_format import check_output_format # , write_output_file
from fosslight_util.output_format import check_output_formats # , write_output_file
from ._parsing_scanoss_file import parsing_scanResult # scanoss
from ._parsing_scanoss_file import parsing_extraInfo # scanoss
import shutil
Expand Down Expand Up @@ -41,7 +41,7 @@ def run_scanoss_py(path_to_scan, output_file_name="", format="", called_by_cli=F
:param write_json_file: if requested, keep the raw files.
:return scanoss_file_list: list of ScanItem (scanned result by files).
"""
success, msg, output_path, output_file, output_extension = check_output_format(output_file_name, format)
success, msg, output_path, output_files, output_extensions = check_output_formats(output_file_name, format)

if not called_by_cli:
global logger
Expand Down

0 comments on commit cfff62d

Please sign in to comment.