12

我正在尝试optparse，这是我的初始脚本。

#!/usr/bin/env python

import os, sys
from optparse import OptionParser

parser = OptionParser()
usage = "usage: %prog [options] arg1 arg2"

parser.add_option("-d", "--dir", type="string",
                  help="List of directory",
                  dest="inDir", default=".")

parser.add_option("-m", "--month", type="int",
                  help="Numeric value of the month", 
                  dest="mon")

options, arguments = parser.parse_args()

if options.inDir:
    print os.listdir(options.inDir)

if options.mon:
    print options.mon

def no_opt()
    print "No option has been given!!"

现在，这就是我想要做的：

  1. 如果该选项没有给出参数，它将采用“默认”值。iemyScript.py -d将仅列出当前目录或-m不带任何参数将当前月份作为参数。
  2. 对于“--month”，仅允许 01 到 12 作为参数
  3. 想要组合多个选项来执行不同的任务，myScript.py -d this_dir -m 02即将做与 -d 和 -m 不同的事情。
  4. 只有当脚本没有提供选项时，它才会打印“没有给出选项！！ ”。

这些可行吗？我确实访问了 doc.python.org 站点以获取可能的答案，但作为一个 python 初学者，我发现自己迷失在页面中。非常感谢您的帮助；提前致谢。干杯！！

更新：16/01/11

我想我仍然缺少一些东西。这就是我现在脚本中的内容。 

parser = OptionParser()
usage = "usage: %prog [options] arg1 arg2"

parser.add_option("-m", "--month", type="string",
                  help="select month from  01|02|...|12",
                  dest="mon", default=strftime("%m"))

parser.add_option("-v", "--vo", type="string",
                  help="select one of the supported VOs",
                  dest="vos")

options, arguments = parser.parse_args()

这些是我的目标：

  1. 不带任何选项运行脚本，将返回option.mon[ working ]
  2. 使用 -m 选项运行脚本，返回option.mon[ working ]
  3. 使用 ONLY -v 选项运行脚本，只会返回option.vos[根本不工作]
  4. 运行带有 -m 和 -v 选项的脚本，会做不同的事情[还没有说到点子上]

当我只使用 -m 选项运行脚本时，它会option.mon先打印，然后再打印option.vos，这是我根本不想要的。如果有人能把我引向正确的方向，我真的很感激。干杯！！

第三次更新

    #!/bin/env python

    from time import strftime
    from calendar import month_abbr
    from optparse import OptionParser

    # Set the CL options 
    parser = OptionParser()
    usage = "usage: %prog [options] arg1 arg2"

    parser.add_option("-m", "--month", type="string",
                      help="select month from  01|02|...|12", 
              dest="mon", default=strftime("%m"))

    parser.add_option("-u", "--user", type="string",
                      help="name of the user", 
              dest="vos")

    options, arguments = parser.parse_args()

    abbrMonth = tuple(month_abbr)[int(options.mon)]

    if options.mon:
        print "The month is: %s" % abbrMonth 

    if options.vos:
        print "My name is: %s" % options.vos 

    if options.mon and options.vos:
        print "I'm '%s' and this month is '%s'" % (options.vos,abbrMonth)

这是脚本在使用各种选项运行时返回的内容： 

# ./test.py
The month is: Feb
#
# ./test.py -m 12
The month is: Dec
#
# ./test.py -m 3 -u Mac
The month is: Mar
My name is: Mac
I'm 'Mac' and this month is 'Mar'
#
# ./test.py -u Mac
The month is: Feb
My name is: Mac
I'm 'Mac' and this month is 'Feb'

我只喜欢看：

 1. `I'm 'Mac' and this month is 'Mar'` - as *result #3*  
 2. `My name is: Mac` - as *result #4*

我究竟做错了什么？干杯！！

第四次更新：

对自己的回答：这样我可以得到我想要的东西，但我仍然没有留下深刻的印象。

#!/bin/env python

import os, sys
from time import strftime
from calendar import month_abbr
from optparse import OptionParser

def abbrMonth(m):
    mn = tuple(month_abbr)[int(m)]
    return mn

# Set the CL options 
parser = OptionParser()
usage = "usage: %prog [options] arg1 arg2"

parser.add_option("-m", "--month", type="string",
                  help="select month from  01|02|...|12",
                  dest="mon")

parser.add_option("-u", "--user", type="string",
                  help="name of the user",
                  dest="vos")

(options, args) = parser.parse_args()

if options.mon and options.vos:
    thisMonth = abbrMonth(options.mon)
    print "I'm '%s' and this month is '%s'" % (options.vos, thisMonth)
    sys.exit(0)

if not options.mon and not options.vos:
    options.mon = strftime("%m")

if options.mon:
    thisMonth = abbrMonth(options.mon)
    print "The month is: %s" % thisMonth

if options.vos:
    print "My name is: %s" % options.vos

现在这给了我我正在寻找的东西：

# ./test.py 
The month is: Feb

# ./test.py -m 09
The month is: Sep

# ./test.py -u Mac
My name is: Mac

# ./test.py -m 3 -u Mac
I'm 'Mac' and this month is 'Mar'

这是唯一的方法吗？对我来说看起来不是“最好的方式”。干杯！！

4

3 回答 3

3

optparse已弃用；你应该argparse在 python2 和 python3 中使用

http://docs.python.org/library/argparse.html#module-argparse

于 2011-06-02T23:29:11.487 回答
2

你的解决方案在我看来是合理的。注释：

  • 我不明白你为什么month_abbr变成一个元组；没有它应该可以正常工作tuple()
  • 我建议检查无效的月份值（raise OptionValueError如果您发现问题）
  • 如果您真的希望用户准确输入“01”、“02”、...或“12”，则可以使用“选择”选项类型；请参阅选项类型文档
于 2014-05-11T00:32:58.417 回答
1

只是为了说明 argparse.ArgumentParser 的 add_argument() 方法的选择选项：

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
from argparse import ArgumentParser
from datetime import date

parser = ArgumentParser()

parser.add_argument("-u", "--user", default="Max Power", help="Username")
parser.add_argument("-m", "--month", default="{:02d}".format(date.today().month),
                    choices=["01","02","03","04","05","06",
                             "07","08","09","10","11","12"],
                    help="Numeric value of the month")

try:
    args = parser.parse_args()
except:
    parser.error("Invalid Month.")
    sys.exit(0) 

print  "The month is {} and the User is {}".format(args.month, args.user)
于 2015-08-13T13:35:56.083 回答