在Python中使用optparse定义变量

时间:2012-05-04 14:58:27

标签: python optparse optionparser

我很好奇如何使用optparse设置变量。我这样运行程序;

programname.py -d c:\ users \\ etc \ etc \ etc

我希望能够使用-d C:\ Users \\ etc \ etc来填充一个名为“path”的变量,我稍后会在程序中使用它。可以这样做吗?这是我的optionparser代码。

我稍后调用Path变量,用于填充字典。

我得到的错误是:

E:> japp_id.py -d“C:\ Users \\ AppData \ Roaming \ Microsoft \ Windows \ Recent \ AutomaticDestinations” Traceback(最近一次调用最后一次):   文件“E:\ japp_id.py”,第30行,in     对于os.listdir(路径)中的ids: NameError:名称'path'未定义

try:
    import os
    import sys
    import urllib2
    from BeautifulSoup import BeautifulSoup
    from optparse import OptionParser
except ImportError:
    print 'Imports not installed'
    sys.exit()

def main():
usage = "usage: %prog [options] arg1 arg2"
parser = OptionParser()
parser.add_option("-d", "--default", action="callback", type="string", dest="dpath")

(opts, args) = parser.parse_args()


if opts.dpath == None:
    parser.print_help()
    parser.error("You must supply a -d for dpath")
if not os.path.isfile(opts.dpath):
    parser.error("%s does not exist" % opts.dpath)
    if __name__ == "__main__":
        main()

appIDlist = []
for ids in os.listdir(path):
        appid = "%s" % (ids).rstrip('.automaticDestinations-ms')
        appIDlist.append(str(appid))

f = urllib2.urlopen("http://www.forensicswiki.org/wiki/List_of_Jump_List_IDs")
s = f.read()
soup = BeautifulSoup(''.join(s))
rows = soup.findAll('tr')

appIDdictionary = dict()        #create an empty dictionary to allow lookups {'appID':'processName'}
for tr in rows:                 #iterate through every row in the table
        cols = tr.findAll('td', limit=2)        #get the first two data chunks (<td>'s)
        appID = str(cols[0]).rstrip('</td>').lstrip('<td>')     #clean up formatting
        processName = str(cols[1]).rstrip('</td>').lstrip('<td>')       #clean up formatting
        appIDdictionary[appID] = processName     #add the pair to the dictionary

#iterate through list of appids pulled from windows user profile, look them up in the dictionary, and print the result
for id in appIDlist:
        if id in appIDdictionary:
                print appIDdictionary[id]# + " is " + ids.rstrip('.automaticDestinations-ms')
        else:
                print 'Unable to find ' + id + ' in dictionary.'
f.close()

3 个答案:

答案 0 :(得分:1)

From the python docs

parser.add_option("-f", "--file", dest="filename",
                  help="write report to FILE", metavar="FILE")

dest参数是路径存储到的变量的名称。随后使用opts.filename访问它。

答案 1 :(得分:0)

您的意思是path = opts.dpath吗?

然后os.listdir(path) ...

答案 2 :(得分:0)

您可能没有将其传入:您需要使用opts调用函数并访问opts.dpath或执行myfunc(opts.dpath)

也许。您的代码实际上并没有告诉我们问题所在。

<强>更新

是的,你想在第30行附近for ids in os.listdir(opts.dpath)