123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161 |
- /**
- * class ArgumentParser
- *
- * Object for parsing command line strings into js objects.
- *
- * Inherited from [[ActionContainer]]
- **/
- 'use strict';
- var util = require('util');
- var format = require('util').format;
- var Path = require('path');
- var sprintf = require('sprintf-js').sprintf;
- // Constants
- var c = require('./const');
- var $$ = require('./utils');
- var ActionContainer = require('./action_container');
- // Errors
- var argumentErrorHelper = require('./argument/error');
- var HelpFormatter = require('./help/formatter');
- var Namespace = require('./namespace');
- /**
- * new ArgumentParser(options)
- *
- * Create a new ArgumentParser object.
- *
- * ##### Options:
- * - `prog` The name of the program (default: Path.basename(process.argv[1]))
- * - `usage` A usage message (default: auto-generated from arguments)
- * - `description` A description of what the program does
- * - `epilog` Text following the argument descriptions
- * - `parents` Parsers whose arguments should be copied into this one
- * - `formatterClass` HelpFormatter class for printing help messages
- * - `prefixChars` Characters that prefix optional arguments
- * - `fromfilePrefixChars` Characters that prefix files containing additional arguments
- * - `argumentDefault` The default value for all arguments
- * - `addHelp` Add a -h/-help option
- * - `conflictHandler` Specifies how to handle conflicting argument names
- * - `debug` Enable debug mode. Argument errors throw exception in
- * debug mode and process.exit in normal. Used for development and
- * testing (default: false)
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#argumentparser-objects
- **/
- function ArgumentParser(options) {
- if (!(this instanceof ArgumentParser)) {
- return new ArgumentParser(options);
- }
- var self = this;
- options = options || {};
- options.description = (options.description || null);
- options.argumentDefault = (options.argumentDefault || null);
- options.prefixChars = (options.prefixChars || '-');
- options.conflictHandler = (options.conflictHandler || 'error');
- ActionContainer.call(this, options);
- options.addHelp = typeof options.addHelp === 'undefined' || !!options.addHelp;
- options.parents = options.parents || [];
- // default program name
- options.prog = (options.prog || Path.basename(process.argv[1]));
- this.prog = options.prog;
- this.usage = options.usage;
- this.epilog = options.epilog;
- this.version = options.version;
- this.debug = (options.debug === true);
- this.formatterClass = (options.formatterClass || HelpFormatter);
- this.fromfilePrefixChars = options.fromfilePrefixChars || null;
- this._positionals = this.addArgumentGroup({ title: 'Positional arguments' });
- this._optionals = this.addArgumentGroup({ title: 'Optional arguments' });
- this._subparsers = null;
- // register types
- function FUNCTION_IDENTITY(o) {
- return o;
- }
- this.register('type', 'auto', FUNCTION_IDENTITY);
- this.register('type', null, FUNCTION_IDENTITY);
- this.register('type', 'int', function (x) {
- var result = parseInt(x, 10);
- if (isNaN(result)) {
- throw new Error(x + ' is not a valid integer.');
- }
- return result;
- });
- this.register('type', 'float', function (x) {
- var result = parseFloat(x);
- if (isNaN(result)) {
- throw new Error(x + ' is not a valid float.');
- }
- return result;
- });
- this.register('type', 'string', function (x) {
- return '' + x;
- });
- // add help and version arguments if necessary
- var defaultPrefix = (this.prefixChars.indexOf('-') > -1) ? '-' : this.prefixChars[0];
- if (options.addHelp) {
- this.addArgument(
- [ defaultPrefix + 'h', defaultPrefix + defaultPrefix + 'help' ],
- {
- action: 'help',
- defaultValue: c.SUPPRESS,
- help: 'Show this help message and exit.'
- }
- );
- }
- if (typeof this.version !== 'undefined') {
- this.addArgument(
- [ defaultPrefix + 'v', defaultPrefix + defaultPrefix + 'version' ],
- {
- action: 'version',
- version: this.version,
- defaultValue: c.SUPPRESS,
- help: "Show program's version number and exit."
- }
- );
- }
- // add parent arguments and defaults
- options.parents.forEach(function (parent) {
- self._addContainerActions(parent);
- if (typeof parent._defaults !== 'undefined') {
- for (var defaultKey in parent._defaults) {
- if (parent._defaults.hasOwnProperty(defaultKey)) {
- self._defaults[defaultKey] = parent._defaults[defaultKey];
- }
- }
- }
- });
- }
- util.inherits(ArgumentParser, ActionContainer);
- /**
- * ArgumentParser#addSubparsers(options) -> [[ActionSubparsers]]
- * - options (object): hash of options see [[ActionSubparsers.new]]
- *
- * See also [subcommands][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#sub-commands
- **/
- ArgumentParser.prototype.addSubparsers = function (options) {
- if (this._subparsers) {
- this.error('Cannot have multiple subparser arguments.');
- }
- options = options || {};
- options.debug = (this.debug === true);
- options.optionStrings = [];
- options.parserClass = (options.parserClass || ArgumentParser);
- if (!!options.title || !!options.description) {
- this._subparsers = this.addArgumentGroup({
- title: (options.title || 'subcommands'),
- description: options.description
- });
- delete options.title;
- delete options.description;
- } else {
- this._subparsers = this._positionals;
- }
- // prog defaults to the usage message of this parser, skipping
- // optional arguments and with no "usage:" prefix
- if (!options.prog) {
- var formatter = this._getFormatter();
- var positionals = this._getPositionalActions();
- var groups = this._mutuallyExclusiveGroups;
- formatter.addUsage(this.usage, positionals, groups, '');
- options.prog = formatter.formatHelp().trim();
- }
- // create the parsers action and add it to the positionals list
- var ParsersClass = this._popActionClass(options, 'parsers');
- var action = new ParsersClass(options);
- this._subparsers._addAction(action);
- // return the created parsers action
- return action;
- };
- ArgumentParser.prototype._addAction = function (action) {
- if (action.isOptional()) {
- this._optionals._addAction(action);
- } else {
- this._positionals._addAction(action);
- }
- return action;
- };
- ArgumentParser.prototype._getOptionalActions = function () {
- return this._actions.filter(function (action) {
- return action.isOptional();
- });
- };
- ArgumentParser.prototype._getPositionalActions = function () {
- return this._actions.filter(function (action) {
- return action.isPositional();
- });
- };
- /**
- * ArgumentParser#parseArgs(args, namespace) -> Namespace|Object
- * - args (array): input elements
- * - namespace (Namespace|Object): result object
- *
- * Parsed args and throws error if some arguments are not recognized
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#the-parse-args-method
- **/
- ArgumentParser.prototype.parseArgs = function (args, namespace) {
- var argv;
- var result = this.parseKnownArgs(args, namespace);
- args = result[0];
- argv = result[1];
- if (argv && argv.length > 0) {
- this.error(
- format('Unrecognized arguments: %s.', argv.join(' '))
- );
- }
- return args;
- };
- /**
- * ArgumentParser#parseKnownArgs(args, namespace) -> array
- * - args (array): input options
- * - namespace (Namespace|Object): result object
- *
- * Parse known arguments and return tuple of result object
- * and unknown args
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#partial-parsing
- **/
- ArgumentParser.prototype.parseKnownArgs = function (args, namespace) {
- var self = this;
- // args default to the system args
- args = args || process.argv.slice(2);
- // default Namespace built from parser defaults
- namespace = namespace || new Namespace();
- self._actions.forEach(function (action) {
- if (action.dest !== c.SUPPRESS) {
- if (!$$.has(namespace, action.dest)) {
- if (action.defaultValue !== c.SUPPRESS) {
- var defaultValue = action.defaultValue;
- if (typeof action.defaultValue === 'string') {
- defaultValue = self._getValue(action, defaultValue);
- }
- namespace[action.dest] = defaultValue;
- }
- }
- }
- });
- Object.keys(self._defaults).forEach(function (dest) {
- namespace[dest] = self._defaults[dest];
- });
- // parse the arguments and exit if there are any errors
- try {
- var res = this._parseKnownArgs(args, namespace);
- namespace = res[0];
- args = res[1];
- if ($$.has(namespace, c._UNRECOGNIZED_ARGS_ATTR)) {
- args = $$.arrayUnion(args, namespace[c._UNRECOGNIZED_ARGS_ATTR]);
- delete namespace[c._UNRECOGNIZED_ARGS_ATTR];
- }
- return [ namespace, args ];
- } catch (e) {
- this.error(e);
- }
- };
- ArgumentParser.prototype._parseKnownArgs = function (argStrings, namespace) {
- var self = this;
- var extras = [];
- // replace arg strings that are file references
- if (this.fromfilePrefixChars !== null) {
- argStrings = this._readArgsFromFiles(argStrings);
- }
- // map all mutually exclusive arguments to the other arguments
- // they can't occur with
- // Python has 'conflicts = action_conflicts.setdefault(mutex_action, [])'
- // though I can't conceive of a way in which an action could be a member
- // of two different mutually exclusive groups.
- function actionHash(action) {
- // some sort of hashable key for this action
- // action itself cannot be a key in actionConflicts
- // I think getName() (join of optionStrings) is unique enough
- return action.getName();
- }
- var conflicts, key;
- var actionConflicts = {};
- this._mutuallyExclusiveGroups.forEach(function (mutexGroup) {
- mutexGroup._groupActions.forEach(function (mutexAction, i, groupActions) {
- key = actionHash(mutexAction);
- if (!$$.has(actionConflicts, key)) {
- actionConflicts[key] = [];
- }
- conflicts = actionConflicts[key];
- conflicts.push.apply(conflicts, groupActions.slice(0, i));
- conflicts.push.apply(conflicts, groupActions.slice(i + 1));
- });
- });
- // find all option indices, and determine the arg_string_pattern
- // which has an 'O' if there is an option at an index,
- // an 'A' if there is an argument, or a '-' if there is a '--'
- var optionStringIndices = {};
- var argStringPatternParts = [];
- argStrings.forEach(function (argString, argStringIndex) {
- if (argString === '--') {
- argStringPatternParts.push('-');
- while (argStringIndex < argStrings.length) {
- argStringPatternParts.push('A');
- argStringIndex++;
- }
- } else {
- // otherwise, add the arg to the arg strings
- // and note the index if it was an option
- var pattern;
- var optionTuple = self._parseOptional(argString);
- if (!optionTuple) {
- pattern = 'A';
- } else {
- optionStringIndices[argStringIndex] = optionTuple;
- pattern = 'O';
- }
- argStringPatternParts.push(pattern);
- }
- });
- var argStringsPattern = argStringPatternParts.join('');
- var seenActions = [];
- var seenNonDefaultActions = [];
- function takeAction(action, argumentStrings, optionString) {
- seenActions.push(action);
- var argumentValues = self._getValues(action, argumentStrings);
- // error if this argument is not allowed with other previously
- // seen arguments, assuming that actions that use the default
- // value don't really count as "present"
- if (argumentValues !== action.defaultValue) {
- seenNonDefaultActions.push(action);
- if (actionConflicts[actionHash(action)]) {
- actionConflicts[actionHash(action)].forEach(function (actionConflict) {
- if (seenNonDefaultActions.indexOf(actionConflict) >= 0) {
- throw argumentErrorHelper(
- action,
- format('Not allowed with argument "%s".', actionConflict.getName())
- );
- }
- });
- }
- }
- if (argumentValues !== c.SUPPRESS) {
- action.call(self, namespace, argumentValues, optionString);
- }
- }
- function consumeOptional(startIndex) {
- // get the optional identified at this index
- var optionTuple = optionStringIndices[startIndex];
- var action = optionTuple[0];
- var optionString = optionTuple[1];
- var explicitArg = optionTuple[2];
- // identify additional optionals in the same arg string
- // (e.g. -xyz is the same as -x -y -z if no args are required)
- var actionTuples = [];
- var args, argCount, start, stop;
- for (;;) {
- if (!action) {
- extras.push(argStrings[startIndex]);
- return startIndex + 1;
- }
- if (explicitArg) {
- argCount = self._matchArgument(action, 'A');
- // if the action is a single-dash option and takes no
- // arguments, try to parse more single-dash options out
- // of the tail of the option string
- var chars = self.prefixChars;
- if (argCount === 0 && chars.indexOf(optionString[1]) < 0) {
- actionTuples.push([ action, [], optionString ]);
- optionString = optionString[0] + explicitArg[0];
- var newExplicitArg = explicitArg.slice(1) || null;
- var optionalsMap = self._optionStringActions;
- if (Object.keys(optionalsMap).indexOf(optionString) >= 0) {
- action = optionalsMap[optionString];
- explicitArg = newExplicitArg;
- } else {
- throw argumentErrorHelper(action, sprintf('ignored explicit argument %r', explicitArg));
- }
- } else if (argCount === 1) {
- // if the action expect exactly one argument, we've
- // successfully matched the option; exit the loop
- stop = startIndex + 1;
- args = [ explicitArg ];
- actionTuples.push([ action, args, optionString ]);
- break;
- } else {
- // error if a double-dash option did not use the
- // explicit argument
- throw argumentErrorHelper(action, sprintf('ignored explicit argument %r', explicitArg));
- }
- } else {
- // if there is no explicit argument, try to match the
- // optional's string arguments with the following strings
- // if successful, exit the loop
- start = startIndex + 1;
- var selectedPatterns = argStringsPattern.substr(start);
- argCount = self._matchArgument(action, selectedPatterns);
- stop = start + argCount;
- args = argStrings.slice(start, stop);
- actionTuples.push([ action, args, optionString ]);
- break;
- }
- }
- // add the Optional to the list and return the index at which
- // the Optional's string args stopped
- if (actionTuples.length < 1) {
- throw new Error('length should be > 0');
- }
- for (var i = 0; i < actionTuples.length; i++) {
- takeAction.apply(self, actionTuples[i]);
- }
- return stop;
- }
- // the list of Positionals left to be parsed; this is modified
- // by consume_positionals()
- var positionals = self._getPositionalActions();
- function consumePositionals(startIndex) {
- // match as many Positionals as possible
- var selectedPattern = argStringsPattern.substr(startIndex);
- var argCounts = self._matchArgumentsPartial(positionals, selectedPattern);
- // slice off the appropriate arg strings for each Positional
- // and add the Positional and its args to the list
- for (var i = 0; i < positionals.length; i++) {
- var action = positionals[i];
- var argCount = argCounts[i];
- if (typeof argCount === 'undefined') {
- continue;
- }
- var args = argStrings.slice(startIndex, startIndex + argCount);
- startIndex += argCount;
- takeAction(action, args);
- }
- // slice off the Positionals that we just parsed and return the
- // index at which the Positionals' string args stopped
- positionals = positionals.slice(argCounts.length);
- return startIndex;
- }
- // consume Positionals and Optionals alternately, until we have
- // passed the last option string
- var startIndex = 0;
- var position;
- var maxOptionStringIndex = -1;
- Object.keys(optionStringIndices).forEach(function (position) {
- maxOptionStringIndex = Math.max(maxOptionStringIndex, parseInt(position, 10));
- });
- var positionalsEndIndex, nextOptionStringIndex;
- while (startIndex <= maxOptionStringIndex) {
- // consume any Positionals preceding the next option
- nextOptionStringIndex = null;
- for (position in optionStringIndices) {
- if (!optionStringIndices.hasOwnProperty(position)) { continue; }
- position = parseInt(position, 10);
- if (position >= startIndex) {
- if (nextOptionStringIndex !== null) {
- nextOptionStringIndex = Math.min(nextOptionStringIndex, position);
- } else {
- nextOptionStringIndex = position;
- }
- }
- }
- if (startIndex !== nextOptionStringIndex) {
- positionalsEndIndex = consumePositionals(startIndex);
- // only try to parse the next optional if we didn't consume
- // the option string during the positionals parsing
- if (positionalsEndIndex > startIndex) {
- startIndex = positionalsEndIndex;
- continue;
- } else {
- startIndex = positionalsEndIndex;
- }
- }
- // if we consumed all the positionals we could and we're not
- // at the index of an option string, there were extra arguments
- if (!optionStringIndices[startIndex]) {
- var strings = argStrings.slice(startIndex, nextOptionStringIndex);
- extras = extras.concat(strings);
- startIndex = nextOptionStringIndex;
- }
- // consume the next optional and any arguments for it
- startIndex = consumeOptional(startIndex);
- }
- // consume any positionals following the last Optional
- var stopIndex = consumePositionals(startIndex);
- // if we didn't consume all the argument strings, there were extras
- extras = extras.concat(argStrings.slice(stopIndex));
- // if we didn't use all the Positional objects, there were too few
- // arg strings supplied.
- if (positionals.length > 0) {
- self.error('too few arguments');
- }
- // make sure all required actions were present
- self._actions.forEach(function (action) {
- if (action.required) {
- if (seenActions.indexOf(action) < 0) {
- self.error(format('Argument "%s" is required', action.getName()));
- }
- }
- });
- // make sure all required groups have one option present
- var actionUsed = false;
- self._mutuallyExclusiveGroups.forEach(function (group) {
- if (group.required) {
- actionUsed = group._groupActions.some(function (action) {
- return seenNonDefaultActions.indexOf(action) !== -1;
- });
- // if no actions were used, report the error
- if (!actionUsed) {
- var names = [];
- group._groupActions.forEach(function (action) {
- if (action.help !== c.SUPPRESS) {
- names.push(action.getName());
- }
- });
- names = names.join(' ');
- var msg = 'one of the arguments ' + names + ' is required';
- self.error(msg);
- }
- }
- });
- // return the updated namespace and the extra arguments
- return [ namespace, extras ];
- };
- ArgumentParser.prototype._readArgsFromFiles = function (argStrings) {
- // expand arguments referencing files
- var self = this;
- var fs = require('fs');
- var newArgStrings = [];
- argStrings.forEach(function (argString) {
- if (self.fromfilePrefixChars.indexOf(argString[0]) < 0) {
- // for regular arguments, just add them back into the list
- newArgStrings.push(argString);
- } else {
- // replace arguments referencing files with the file content
- try {
- var argstrs = [];
- var filename = argString.slice(1);
- var content = fs.readFileSync(filename, 'utf8');
- content = content.trim().split('\n');
- content.forEach(function (argLine) {
- self.convertArgLineToArgs(argLine).forEach(function (arg) {
- argstrs.push(arg);
- });
- argstrs = self._readArgsFromFiles(argstrs);
- });
- newArgStrings.push.apply(newArgStrings, argstrs);
- } catch (error) {
- return self.error(error.message);
- }
- }
- });
- return newArgStrings;
- };
- ArgumentParser.prototype.convertArgLineToArgs = function (argLine) {
- return [ argLine ];
- };
- ArgumentParser.prototype._matchArgument = function (action, regexpArgStrings) {
- // match the pattern for this action to the arg strings
- var regexpNargs = new RegExp('^' + this._getNargsPattern(action));
- var matches = regexpArgStrings.match(regexpNargs);
- var message;
- // throw an exception if we weren't able to find a match
- if (!matches) {
- switch (action.nargs) {
- /*eslint-disable no-undefined*/
- case undefined:
- case null:
- message = 'Expected one argument.';
- break;
- case c.OPTIONAL:
- message = 'Expected at most one argument.';
- break;
- case c.ONE_OR_MORE:
- message = 'Expected at least one argument.';
- break;
- default:
- message = 'Expected %s argument(s)';
- }
- throw argumentErrorHelper(
- action,
- format(message, action.nargs)
- );
- }
- // return the number of arguments matched
- return matches[1].length;
- };
- ArgumentParser.prototype._matchArgumentsPartial = function (actions, regexpArgStrings) {
- // progressively shorten the actions list by slicing off the
- // final actions until we find a match
- var self = this;
- var result = [];
- var actionSlice, pattern, matches;
- var i, j;
- function getLength(string) {
- return string.length;
- }
- for (i = actions.length; i > 0; i--) {
- pattern = '';
- actionSlice = actions.slice(0, i);
- for (j = 0; j < actionSlice.length; j++) {
- pattern += self._getNargsPattern(actionSlice[j]);
- }
- pattern = new RegExp('^' + pattern);
- matches = regexpArgStrings.match(pattern);
- if (matches && matches.length > 0) {
- // need only groups
- matches = matches.splice(1);
- result = result.concat(matches.map(getLength));
- break;
- }
- }
- // return the list of arg string counts
- return result;
- };
- ArgumentParser.prototype._parseOptional = function (argString) {
- var action, optionString, argExplicit, optionTuples;
- // if it's an empty string, it was meant to be a positional
- if (!argString) {
- return null;
- }
- // if it doesn't start with a prefix, it was meant to be positional
- if (this.prefixChars.indexOf(argString[0]) < 0) {
- return null;
- }
- // if the option string is present in the parser, return the action
- if (this._optionStringActions[argString]) {
- return [ this._optionStringActions[argString], argString, null ];
- }
- // if it's just a single character, it was meant to be positional
- if (argString.length === 1) {
- return null;
- }
- // if the option string before the "=" is present, return the action
- if (argString.indexOf('=') >= 0) {
- optionString = argString.split('=', 1)[0];
- argExplicit = argString.slice(optionString.length + 1);
- if (this._optionStringActions[optionString]) {
- action = this._optionStringActions[optionString];
- return [ action, optionString, argExplicit ];
- }
- }
- // search through all possible prefixes of the option string
- // and all actions in the parser for possible interpretations
- optionTuples = this._getOptionTuples(argString);
- // if multiple actions match, the option string was ambiguous
- if (optionTuples.length > 1) {
- var optionStrings = optionTuples.map(function (optionTuple) {
- return optionTuple[1];
- });
- this.error(format(
- 'Ambiguous option: "%s" could match %s.',
- argString, optionStrings.join(', ')
- ));
- // if exactly one action matched, this segmentation is good,
- // so return the parsed action
- } else if (optionTuples.length === 1) {
- return optionTuples[0];
- }
- // if it was not found as an option, but it looks like a negative
- // number, it was meant to be positional
- // unless there are negative-number-like options
- if (argString.match(this._regexpNegativeNumber)) {
- if (!this._hasNegativeNumberOptionals.some(Boolean)) {
- return null;
- }
- }
- // if it contains a space, it was meant to be a positional
- if (argString.search(' ') >= 0) {
- return null;
- }
- // it was meant to be an optional but there is no such option
- // in this parser (though it might be a valid option in a subparser)
- return [ null, argString, null ];
- };
- ArgumentParser.prototype._getOptionTuples = function (optionString) {
- var result = [];
- var chars = this.prefixChars;
- var optionPrefix;
- var argExplicit;
- var action;
- var actionOptionString;
- // option strings starting with two prefix characters are only split at
- // the '='
- if (chars.indexOf(optionString[0]) >= 0 && chars.indexOf(optionString[1]) >= 0) {
- if (optionString.indexOf('=') >= 0) {
- var optionStringSplit = optionString.split('=', 1);
- optionPrefix = optionStringSplit[0];
- argExplicit = optionStringSplit[1];
- } else {
- optionPrefix = optionString;
- argExplicit = null;
- }
- for (actionOptionString in this._optionStringActions) {
- if (actionOptionString.substr(0, optionPrefix.length) === optionPrefix) {
- action = this._optionStringActions[actionOptionString];
- result.push([ action, actionOptionString, argExplicit ]);
- }
- }
- // single character options can be concatenated with their arguments
- // but multiple character options always have to have their argument
- // separate
- } else if (chars.indexOf(optionString[0]) >= 0 && chars.indexOf(optionString[1]) < 0) {
- optionPrefix = optionString;
- argExplicit = null;
- var optionPrefixShort = optionString.substr(0, 2);
- var argExplicitShort = optionString.substr(2);
- for (actionOptionString in this._optionStringActions) {
- if (!$$.has(this._optionStringActions, actionOptionString)) continue;
- action = this._optionStringActions[actionOptionString];
- if (actionOptionString === optionPrefixShort) {
- result.push([ action, actionOptionString, argExplicitShort ]);
- } else if (actionOptionString.substr(0, optionPrefix.length) === optionPrefix) {
- result.push([ action, actionOptionString, argExplicit ]);
- }
- }
- // shouldn't ever get here
- } else {
- throw new Error(format('Unexpected option string: %s.', optionString));
- }
- // return the collected option tuples
- return result;
- };
- ArgumentParser.prototype._getNargsPattern = function (action) {
- // in all examples below, we have to allow for '--' args
- // which are represented as '-' in the pattern
- var regexpNargs;
- switch (action.nargs) {
- // the default (null) is assumed to be a single argument
- case undefined:
- case null:
- regexpNargs = '(-*A-*)';
- break;
- // allow zero or more arguments
- case c.OPTIONAL:
- regexpNargs = '(-*A?-*)';
- break;
- // allow zero or more arguments
- case c.ZERO_OR_MORE:
- regexpNargs = '(-*[A-]*)';
- break;
- // allow one or more arguments
- case c.ONE_OR_MORE:
- regexpNargs = '(-*A[A-]*)';
- break;
- // allow any number of options or arguments
- case c.REMAINDER:
- regexpNargs = '([-AO]*)';
- break;
- // allow one argument followed by any number of options or arguments
- case c.PARSER:
- regexpNargs = '(-*A[-AO]*)';
- break;
- // all others should be integers
- default:
- regexpNargs = '(-*' + $$.repeat('-*A', action.nargs) + '-*)';
- }
- // if this is an optional action, -- is not allowed
- if (action.isOptional()) {
- regexpNargs = regexpNargs.replace(/-\*/g, '');
- regexpNargs = regexpNargs.replace(/-/g, '');
- }
- // return the pattern
- return regexpNargs;
- };
- //
- // Value conversion methods
- //
- ArgumentParser.prototype._getValues = function (action, argStrings) {
- var self = this;
- // for everything but PARSER args, strip out '--'
- if (action.nargs !== c.PARSER && action.nargs !== c.REMAINDER) {
- argStrings = argStrings.filter(function (arrayElement) {
- return arrayElement !== '--';
- });
- }
- var value, argString;
- // optional argument produces a default when not present
- if (argStrings.length === 0 && action.nargs === c.OPTIONAL) {
- value = (action.isOptional()) ? action.constant : action.defaultValue;
- if (typeof (value) === 'string') {
- value = this._getValue(action, value);
- this._checkValue(action, value);
- }
- // when nargs='*' on a positional, if there were no command-line
- // args, use the default if it is anything other than None
- } else if (argStrings.length === 0 && action.nargs === c.ZERO_OR_MORE &&
- action.optionStrings.length === 0) {
- value = (action.defaultValue || argStrings);
- this._checkValue(action, value);
- // single argument or optional argument produces a single value
- } else if (argStrings.length === 1 &&
- (!action.nargs || action.nargs === c.OPTIONAL)) {
- argString = argStrings[0];
- value = this._getValue(action, argString);
- this._checkValue(action, value);
- // REMAINDER arguments convert all values, checking none
- } else if (action.nargs === c.REMAINDER) {
- value = argStrings.map(function (v) {
- return self._getValue(action, v);
- });
- // PARSER arguments convert all values, but check only the first
- } else if (action.nargs === c.PARSER) {
- value = argStrings.map(function (v) {
- return self._getValue(action, v);
- });
- this._checkValue(action, value[0]);
- // all other types of nargs produce a list
- } else {
- value = argStrings.map(function (v) {
- return self._getValue(action, v);
- });
- value.forEach(function (v) {
- self._checkValue(action, v);
- });
- }
- // return the converted value
- return value;
- };
- ArgumentParser.prototype._getValue = function (action, argString) {
- var result;
- var typeFunction = this._registryGet('type', action.type, action.type);
- if (typeof typeFunction !== 'function') {
- var message = format('%s is not callable', typeFunction);
- throw argumentErrorHelper(action, message);
- }
- // convert the value to the appropriate type
- try {
- result = typeFunction(argString);
- // ArgumentTypeErrors indicate errors
- // If action.type is not a registered string, it is a function
- // Try to deduce its name for inclusion in the error message
- // Failing that, include the error message it raised.
- } catch (e) {
- var name = null;
- if (typeof action.type === 'string') {
- name = action.type;
- } else {
- name = action.type.name || action.type.displayName || '<function>';
- }
- var msg = format('Invalid %s value: %s', name, argString);
- if (name === '<function>') { msg += '\n' + e.message; }
- throw argumentErrorHelper(action, msg);
- }
- // return the converted value
- return result;
- };
- ArgumentParser.prototype._checkValue = function (action, value) {
- // converted value must be one of the choices (if specified)
- var choices = action.choices;
- if (choices) {
- // choise for argument can by array or string
- if ((typeof choices === 'string' || Array.isArray(choices)) &&
- choices.indexOf(value) !== -1) {
- return;
- }
- // choise for subparsers can by only hash
- if (typeof choices === 'object' && !Array.isArray(choices) && choices[value]) {
- return;
- }
- if (typeof choices === 'string') {
- choices = choices.split('').join(', ');
- } else if (Array.isArray(choices)) {
- choices = choices.join(', ');
- } else {
- choices = Object.keys(choices).join(', ');
- }
- var message = format('Invalid choice: %s (choose from [%s])', value, choices);
- throw argumentErrorHelper(action, message);
- }
- };
- //
- // Help formatting methods
- //
- /**
- * ArgumentParser#formatUsage -> string
- *
- * Return usage string
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#printing-help
- **/
- ArgumentParser.prototype.formatUsage = function () {
- var formatter = this._getFormatter();
- formatter.addUsage(this.usage, this._actions, this._mutuallyExclusiveGroups);
- return formatter.formatHelp();
- };
- /**
- * ArgumentParser#formatHelp -> string
- *
- * Return help
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#printing-help
- **/
- ArgumentParser.prototype.formatHelp = function () {
- var formatter = this._getFormatter();
- // usage
- formatter.addUsage(this.usage, this._actions, this._mutuallyExclusiveGroups);
- // description
- formatter.addText(this.description);
- // positionals, optionals and user-defined groups
- this._actionGroups.forEach(function (actionGroup) {
- formatter.startSection(actionGroup.title);
- formatter.addText(actionGroup.description);
- formatter.addArguments(actionGroup._groupActions);
- formatter.endSection();
- });
- // epilog
- formatter.addText(this.epilog);
- // determine help from format above
- return formatter.formatHelp();
- };
- ArgumentParser.prototype._getFormatter = function () {
- var FormatterClass = this.formatterClass;
- var formatter = new FormatterClass({ prog: this.prog });
- return formatter;
- };
- //
- // Print functions
- //
- /**
- * ArgumentParser#printUsage() -> Void
- *
- * Print usage
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#printing-help
- **/
- ArgumentParser.prototype.printUsage = function () {
- this._printMessage(this.formatUsage());
- };
- /**
- * ArgumentParser#printHelp() -> Void
- *
- * Print help
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#printing-help
- **/
- ArgumentParser.prototype.printHelp = function () {
- this._printMessage(this.formatHelp());
- };
- ArgumentParser.prototype._printMessage = function (message, stream) {
- if (!stream) {
- stream = process.stdout;
- }
- if (message) {
- stream.write('' + message);
- }
- };
- //
- // Exit functions
- //
- /**
- * ArgumentParser#exit(status=0, message) -> Void
- * - status (int): exit status
- * - message (string): message
- *
- * Print message in stderr/stdout and exit program
- **/
- ArgumentParser.prototype.exit = function (status, message) {
- if (message) {
- if (status === 0) {
- this._printMessage(message);
- } else {
- this._printMessage(message, process.stderr);
- }
- }
- process.exit(status);
- };
- /**
- * ArgumentParser#error(message) -> Void
- * - err (Error|string): message
- *
- * Error method Prints a usage message incorporating the message to stderr and
- * exits. If you override this in a subclass,
- * it should not return -- it should
- * either exit or throw an exception.
- *
- **/
- ArgumentParser.prototype.error = function (err) {
- var message;
- if (err instanceof Error) {
- if (this.debug === true) {
- throw err;
- }
- message = err.message;
- } else {
- message = err;
- }
- var msg = format('%s: error: %s', this.prog, message) + c.EOL;
- if (this.debug === true) {
- throw new Error(msg);
- }
- this.printUsage(process.stderr);
- return this.exit(2, msg);
- };
- module.exports = ArgumentParser;
|