Function process_command_line

openVulnQuery/_library/cli_api.py:356–416

Interpret parameters given in command line.

(string_list=None)

def process_command_line(string_list=None):
    """Interpret parameters given in command line."""

    parser = parser_factory()

    args = parser.parse_args(args=string_list)

    if args.api_resource[0] not in constants.ALLOWS_FILTER:
        if args.first_published or args.last_published:
            parser.error(
                'Only {} based filter can have additional first_published or'
                ' last_published filter'.format(constants.ALLOWS_FILTER))

    if args.api_resource[0] not in constants.SUPPORTED_PLATFORMS_ALIAS:
        if args.platformAlias:
            parser.error(
                'Only {} based filter can have additional platformAlias filter'
                .format(constants.SUPPORTED_PLATFORMS_ALIAS))

    if args.api_resource[0] in constants.NON_ADVISORY_QUERY:
        if args.count or args.fields:
            parser.error(
                '{} do not support fields or count options'.format(constants.NON_ADVISORY_QUERY))

        if args.api_resource[0] == "OS":
            if args.api_resource[1] not in constants.SUPPORTED_PLATFORMS_VERSION:
                parser.error(
                    'Only network operating system types for OS type query are: {}'.format(constants.SUPPORTED_PLATFORMS_VERSION))
        elif args.api_resource[0] == "platform":
            if args.api_resource[1] not in constants.SUPPORTED_PLATFORMS_ALIAS:
                parser.error(
                    'Only network operating system types for platform type query are: {}'.format(constants.SUPPORTED_PLATFORMS_ALIAS))

    if not args.json_config_path:
        # Try next environment variables are set, then config.py, or fail:
        keys_required = ('CLIENT_ID', 'CLIENT_SECRET')
        env_config = {k: os.getenv(k, None) for k in keys_required}

        if all([v for v in env_config.values()]):  # OK, take env values:
            for key in keys_required:
                setattr(config, key, env_config[key])
        elif all([getattr(config, k, None) for k in keys_required]):
            pass  # Fallback to the credentials in config.py (non-empty!)
        else:
            parser.error(
                ' --conf <FILE> ? Missing configuration file (credentials)')
    else:
        if not os.path.isfile(args.json_config_path):
            parser.error(
                'Configuration file not found at %s' % args.json_config_path)
        else:
            parsed_config = json.load(open(args.json_config_path))
            keys_required = ('CLIENT_ID', 'CLIENT_SECRET')
            for key in keys_required:
                setattr(config, key, parsed_config[key])
            keys_optional = ('REQUEST_TOKEN_URL', 'API_URL')
            for key in keys_optional:
                if key in parsed_config: