Python argument de ligne de commande affecter à une variable

J'ai de stocker l'argument de ligne de commande dans une variable ou d'attribuer une valeur par défaut pour elle.

Ce que je suis en train est le ci-dessous

import sys
Var=sys.argv[1] or "somevalue"

J'obtiens l'erreur de l'indice de si je ne spécifiez pas d'argument. Comment résoudre ce problème?

  • Peut-être utiliser une approche plus structurée, voir stackoverflow.com/questions/1009860/...
  • En principe oui, mais c'est overkill pour ce scénario, vous ne pensez pas?
  • Peut-être. Nous ne voyons pas l'ensemble du contexte.
InformationsquelleAutor pikapika | 2014-02-11
  1. 7
    Var=sys.argv[1] if len(sys.argv) > 1 else "somevalue"
    • Nice one-liner, je n'ai commencé à l'aide de cette 'perl-comme' méthode il y a quelques semaines.
    InformationsquelleAutor Michael Schlottke-Lakemper
  2. 2

    Bonne question.

    Je pense que la meilleure solution serait de faire

    try:
    var = sys.argv[1]
except IndexError:
    var = "somevalue"
    InformationsquelleAutor Tim Pietzcker
  3. 2

    La builtin argparse module est prévu pour ces sortes de tâches:

    import argparse

# Set up argument parser
ap = argparse.ArgumentParser()

# Single positional argument, nargs makes it optional
ap.add_argument("thingy", nargs='?', default="blah")

# Do parsing
a = ap.parse_args()

# Use argument
print a.thingy

    Ou, si vous êtes coincé avec Python 2.6 ou version antérieure, et ne souhaitez pas ajouter une exigence sur le rétroportage module argparse, vous pouvez faire la même chose manuellement comme suit:

    import optparse

opter = optparse.OptionParser()
# opter.add_option("-v", "--verbose") etc
opts, args = opter.parse_args()

if len(args) == 0:
    var = "somevalue"
elif len(args) == 1:
    var = args[0]
else:
    opter.error("Only one argument expected, got %d" % len(args))

print var
    • pourquoi ne pas directement opter.add_option('--var', default="somevalue")? Voir ma réponse.
    InformationsquelleAutor dbr
  4. 0

    Essayez ce qui suit une ligne de commande-traitement modèle:

    def process_command_line(argv):
    ...
    # add your option here
    parser.add_option('--var',
        default="somevalue",
        help="your help text")

def main(argv=None):
    settings, args = process_command_line(argv)
    ...
    print settings, args # <- print your settings and args

    De course ./your_script.py avec le modèle ci-dessous et de vos modifications ci-dessus imprime {'var': 'somevalue'} []

    Pour un exemple de ligne de commande-traitement modèle de voir un exemple dans Code Comme un Pythonista: Idiomatiques Python (http://python.net/~goodger/projets/pycon/2007/idiomatiques/document.html#ligne de commande-traitement):

    #!/usr/bin/env python

"""
Module docstring.
"""

import sys
import optparse

def process_command_line(argv):
    """
    Return a 2-tuple: (settings object, args list).
    `argv` is a list of arguments, or `None` for ``sys.argv[1:]``.
    """
    if argv is None:
        argv = sys.argv[1:]

    # initialize the parser object:
    parser = optparse.OptionParser(
        formatter=optparse.TitledHelpFormatter(width=78),
        add_help_option=None)

    # define options here:
    parser.add_option(      # customized description; put --help last
        '-h', '--help', action='help',
        help='Show this help message and exit.')

    settings, args = parser.parse_args(argv)

    # check number of arguments, verify values, etc.:
    if args:
        parser.error('program takes no command-line arguments; '
                     '"%s" ignored.' % (args,))

    # further process settings & args if necessary

    return settings, args

def main(argv=None):
    settings, args = process_command_line(argv)
    # application code here, like:
    # run(settings, args)
    return 0        # success

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