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

第一次根据需求写脚本

第一个版本，用于通用的数据转换

原数据为需要构造目标格式里面的几个字段，用某分隔符分开

目标数据为用指定分隔符分割的字段，源文件字段填充其间，其他字段为0

主要涉及命令行参数的处理和文件操作

#!/usr/bin/python # -*- coding: utf-8 -*- #dataformat.py #this script change data from your source to the dest data format import os,getopt,sys #read file ,return lines of the file def 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 way def one_line_proc(parts,total,to,outsp): toindex = 0 outline="" for i in range(1,total+1): if toindex!=len(to) and i==to[toindex]: outline+=parts[toindex] toindex+=1 else: outline+="0" if i!=total: outline+=outsp return outline #hand from inpath to the outpath def 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 = value elif op == "-o": outpath = value elif 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=tmp if '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()