Python脚本示例[命令行参数，函数，判定，退出等]

mmdbcn

　　第一次根据需求写脚本
　　第一个版本，用于通用的数据转换
　　原数据为需要构造目标格式里面的几个字段，用某分隔符分开
　　目标数据为用指定分隔符分割的字段，源文件字段填充其间，其他字段为0
　　主要涉及命令行参数的处理和文件操作
#!/usr/bin/python# -*- coding: utf-8 -*-#dataformat.py#this script change data from your source to the dest data formatimport os,getopt,sys#read file ,return lines of the filedef read_file(path):f = open(path,"r")lines = f.readlines()f.close()return lines#process one line#now according to the order of to and the source file line order#change to a more flexable waydef one_line_proc(parts,total,to,outsp):toindex = 0outline=""for i in range(1,total+1):if toindex!=len(to) and i==to[toindex]:outline+=parts[toindex]toindex+=1else:outline+="0"if i!=total:outline+=outspreturn outline#hand from inpath to the outpathdef process(inpath,total,to,outpath,insp="\t",outsp="\t"):lines = read_file(inpath)f = open(outpath,"w")result=[]for line in lines:parts = line.strip("\n").split(insp)if len(parts) == len(to):outline = one_line_proc(parts,total,to,outsp)result.append(outline+"\n")f.writelines(result)f.close()def main():try:opts,args = getopt.getopt(sys.argv[1:],"F:P:t:a:i:o:")if len(opts) < 3:print("the mount of params must great equal than 3")sys.exit(1)for op,value in opts:if op == "-i":inpath = valueelif op == "-o":outpath = valueelif op == "-t":total = int(value)elif op == "-a":to = value.split(",")elif op == "-F":insp = value.decode("string_escape")elif op == "-P":outsp = value.decode("string_escape")#print(opts)#print(args)except getopt.GetoptError:print("params are not defined well!")if 'outpath' not in dir():outpath = inpath+".dist"if 'inpath' not in dir():print("-i param is needed,input file path must define!")sys.exit(1)if 'total' not in dir(): print("-t param is needed,the fields of result file must define!")sys.exit(1)if 'to' not in dir():print("-a param is needed,must assign the field to put !")sys.exit(1)if not os.path.exists(inpath):print("file : %s is not exists"%inpath)sys.exit(1)tmp=[]for st in to:tmp.append(int(st))to=tmpif 'insp' in dir() and 'outsp' in dir():#print("path a")process(inpath,total,to,outpath,insp,outsp)elif 'insp' in dir():#print("path b")process(inpath,total,to,outpath,insp)elif 'outsp' in dir():#print("path c")process(inpath,total,to,outpath,outsp=outsp)else:#print("path d")process(inpath,total,to,outpath)#if __name__ =="__main__":main()