作者:水果jia | 来源:互联网 | 2023-09-24 19:46
本文由编程笔记#小编为大家整理,主要介绍了python Python2示例处理选项和参数。相关的知识,希望对你有一定的参考价值。
import sys, os, getopt
TARGET_TYPE = ".log"
def process_file(path):
''' Process a file. '''
print path
def process_dir(path):
''' Process a directory. '''
file_list = []
files = os.listdir(path)
for file in files:
file = os.path.join(path, file)
root, ext = os.path.splitext(os.path.basename(file))
if os.path.isfile(file) and ext == TARGET_TYPE:
process_file(file)
def main():
if len(sys.argv) <2:
print "Arguments should be at least 2."
print "python get_blockid.py -f [FILE]"
print "python get_blockid.py -d [DIRECTORY]"
exit(1)
try:
opts, args = getopt.getopt(sys.argv[1:], "f:d:", ["file=", "directory="])
for arg, value in opts:
if arg in (&#39;-f&#39;, &#39;--file&#39;):
root, ext = os.path.basename(value)
if ext == &#39;TARGET_TYPE&#39;:
process_file(value)
elif arg in (&#39;-d&#39;, &#39;--directory&#39;):
process_dir(value)
else:
print "Argument error. %s" % arg
exit(1)
except getopt.GetoptError as e:
print e
exit(1)
if __name__ == &#39;__main__&#39;:
main()