理解选项部分

2024-04-20 11:19:36 发布

您现在位置:Python中文网/ 问答频道 /正文

我在尝试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. 如果选项中没有参数,它将采用“默认”值。 i、 emyScript.py -d将只列出当前目录,或者不带任何参数的-m将以当前月份为参数。
  2. 对于“-month”,仅允许01到12作为参数
  3. 想要组合多个选项来执行不同的任务,即myScript.py -d this_dir -m 02将做不同的事情,而不是作为单个的-d和-m。
  4. 它会打印“没有选项!!“仅当脚本未提供任何选项时。

这些是可行的吗?我确实访问了doc.python.org网站寻找可能的答案,但是作为一个python初学者,我发现自己迷失在了页面中。非常感谢你的帮助,提前谢谢。干杯!!


更新:11年1月16日

我想我还是少了点什么。这就是我剧本里的内容。

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. 只使用-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'

这是唯一的办法吗?在我看来不是最好的办法。干杯!!


Tags: andfrompyimportparserifismac
3条回答

只是为了演示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)

你的解决方案在我看来是合理的。评论:

  • 我不明白为什么要把month_abbr变成元组;如果没有tuple(),它应该可以正常工作
  • 我建议检查无效的月值(raise OptionValueError如果您发现问题)
  • 如果您真的希望用户准确地输入“01”、“02”、“…”或“12”,则可以使用“choice”选项类型;请参见option types documentation

optparse已弃用;您应该在python2和python3中使用argparse

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

相关问题 更多 >