Hardware-Sniffer/Hardware-Sniffer-CLI.py

import argparse
import HardwareSniffer
import platform
import sys
import traceback

EXIT_SUCCESS = 0
EXIT_INVALID_ARGS = 1
EXIT_UNSUPPORTED_OS = 2
EXIT_HARDWARE_COLLECTION_ERROR = 3
EXIT_REPORT_GENERATION_ERROR = 4
EXIT_ACPI_DUMP_ERROR = 5

def main():
    os_name = platform.system()

    parser = argparse.ArgumentParser()
    parser.add_argument("-e", "--export", action="store_true", help="export system report")
    parser.add_argument("-o", "--output-dir", default="SysReport", help="custom output directory to save system report, default to SysReport")
    args = parser.parse_args()

    if not args.export:
        parser.print_help()
        return EXIT_INVALID_ARGS

    if os_name != "Windows":
        print(f"Unsupported operating system: {os_name}", file=sys.stderr)
        return EXIT_UNSUPPORTED_OS

    try:
        h = HardwareSniffer.HardwareSniffer(args.output_dir, rich_format=False)

        h.hardware_info.hardware_collector()
    except Exception as e:
        print(f"Error during hardware collection: {e}", file=sys.stderr)
        traceback.print_exc()
        return EXIT_HARDWARE_COLLECTION_ERROR

    try:
        h.export_hardware_report()
    except Exception as e:
        print(f"Error saving report: {e}", file=sys.stderr)
        traceback.print_exc()
        return EXIT_REPORT_GENERATION_ERROR

    try:
        h.dump_acpi_tables()
    except Exception as e:
        print(f"Error dumping ACPI tables: {e}", file=sys.stderr)
        traceback.print_exc()
        return EXIT_ACPI_DUMP_ERROR

    h.u.head("Hardware Sniffer")
    print("")
    print("Done! Please check the report in `{}`".format(h.result_dir))
    return EXIT_SUCCESS

if __name__ == '__main__':
    sys.exit(main())