Python parse_args Examples

Example #1

File: stix2_validator.py Project: ChengLawson/cti-stix-validator

def main():
    # Parse command line arguments
    options = parse_args(sys.argv[1:], is_script=True)

    # Only print prompt if script is run on cmdline and no input is piped in
    if options.files == sys.stdin and os.isatty(0):
        print('Input STIX content, then press Ctrl+D: ')

    try:
        # Validate input documents
        results = run_validation(options)

        # Print validation results
        print_results(results)

        # Determine exit status code and exit.
        code = codes.get_code(results)
        sys.exit(code)

    except (ValidationError, IOError) as ex:
        output.error("Validation error occurred: '%s'" % str(ex),
                     codes.EXIT_VALIDATION_ERROR)
    except Exception:
        logger.exception("Fatal error occurred")
        sys.exit(codes.EXIT_FAILURE)

Example #2

File: stix_stepper.py Project: cyware-labs/cti-stix-elevator

def main():
    stepper_arg_parser = _get_arg_parser()
    stepper_args = stepper_arg_parser.parse_args()
    validator_options = stix2validator.parse_args(
        shlex.split(stepper_args.validator_args))

    stix2validator.output.set_level(validator_options.verbose)
    stix2validator.output.set_silent(validator_options.silent)

    step_file(stepper_args.file_, validator_options)

Example #3

File: test_stepper.py Project: cyware-labs/cti-stix-elevator

def idiom_stepper_mappings(before_file_path, stored_json):
    """Test fresh conversion from XML to JSON matches stored JSON samples."""
    validator_options = stix2validator.parse_args("")

    stix2validator.output.set_level(validator_options.verbose)
    stix2validator.output.set_silent(validator_options.silent)

    print("Checking - " + before_file_path)
    print("With Master - " + stored_json["id"])

    converted_json = step_file(before_file_path, validator_options)
    converted_json = json.loads(converted_json)
    return idiom_mappings(converted_json, stored_json, _IGNORE)

Example #4

def main():
    stepper_arg_parser = _get_arg_parser()
    stepper_args = stepper_arg_parser.parse_args()
    initialize_options()
    set_option_value("missing_policy", "use-extensions")
    set_option_value("custom_property_prefix",
                     stepper_args.custom_property_prefix)
    validator_options = stix2validator.parse_args(
        shlex.split(stepper_args.validator_args))

    stix2validator.output.set_level(validator_options.verbose)
    stix2validator.output.set_silent(validator_options.silent)

    step_file(stepper_args.file_, validator_options)

Example #5

File: test_stepper.py Project: secureonelabs/cti-stix-elevator

def idiom_stepper_mappings(before_file_path, stored_json):
    """Test fresh conversion from XML to JSON matches stored JSON samples."""
    validator_options = stix2validator.parse_args("")
    initialize_options()
    set_option_value("missing_policy", "use-extensions")
    set_option_value("custom_property_prefix", "elevator")

    stix2validator.output.set_level(validator_options.verbose)
    stix2validator.output.set_silent(validator_options.silent)

    print("Checking - " + before_file_path)
    print("With Master - " + stored_json["id"])

    converted_json = step_file(before_file_path, validator_options)
    converted_json = json.loads(converted_json)
    return idiom_mappings(converted_json, stored_json, _IGNORE)

Example #6

def main():
    # Parse command line arguments
    options = parse_args(sys.argv[1:], is_script=True)

    try:
        # Validate input documents
        results = run_validation(options)

        # Print validation results
        print_results(results)

        # Determine exit status code and exit.
        code = codes.get_code(results)
        sys.exit(code)

    except (ValidationError, IOError) as ex:
        output.error(
            "Validation error occurred: '%s'" % str(ex),
            codes.EXIT_VALIDATION_ERROR
        )
    except Exception:
        logger.exception("Fatal error occurred")
        sys.exit(codes.EXIT_FAILURE)