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()

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

相关推荐