Python编程实现对CodeSys中ST代码的自动排版(二)
由于刚学Python，对很多知识都还是一知半解，欢迎各位大侠给出宝贵意见。在上篇随笔中的代码基础上，又做了一些修改，完善了一些功能。（1）能够缩进IF..ELSIF..ELSE..END_IF语句（2）能够缩进WHILE....END_WHILE语句（3）能够缩进FOR...END_FOR语句（4）能够简单的缩进CASE...END_CASE语句，由于没有很好的办法判断CASE项，就暂时对CASE之后的代码进行了4个字符的缩进，以上三条都是缩进2个字符。（5）对输入输出的文件名进行了判断。测试方法是把源文件和生成的文件都放在了Python文件的同一目录下。
上代码（每行代码写的太长了，好像需要滚动条.... :(）
#FormatCodeSysST.py
#coding:gb2312
import string
getFile = raw_input("请输入要转换的文件名称(包括后缀名):")
try:
file_in= open(getFile,'r')
Get_Flag = True
except IOError:
print "★文件读取失败★"
raw_input("按回车键返回...")
Get_Flag = False
if Get_Flag:
outputFile = raw_input("请输入转换后保存的文件名称(包括后缀名):")
try:
file_out = open(outputFile,'w')
Out_Flag = True
except:
print "★文件保存失败★"
raw_input("按回车键返回...")
Out_Flag = False
if Out_Flag:
print "正在处理中，请稍候...."
text1 = file_in.readline()
lst1 = []
#-----------------------------------------------------------------------------
#删除行首和行末的空格
while (text1 != ''):
lst1.append(string.strip(text1))
text1 = file_in.readline()
indent = 0
#-----------------------------------------------------------------------------
#判断每一行的开头字符串，并根据不同的字符串进行不同的缩进
for i in lst1:
if (i[:3] == 'IF ') or (i[:3] == 'IF(') or (i[:len(i)]=='IF'): #if开头的行
i = ' '*indent + i
indent += 2
elif (i[:5] == 'ELSE ')or (i[:len(i)] == 'ELSE') or (i[:5] == 'ELSE(') : #else开头的行
indent -= 2
i = ' '*indent + i
indent += 2
elif (i[:len(i)] == 'ELSIF')or (i[:6] == 'ELSIF ') or (i[:6] == 'ELSIF('): #elsif 开头的行
indent -= 2
i = ' '*indent + i
indent += 2
elif (i[:7] =='END_IF ') or (i[:len(i)] =='END_IF') or (i[:7] =='END_IF('): #end_if开头的行
indent -= 2
i = ' '*indent + i
elif i[:6] == 'WHILE ' or i[:len(i)] == 'WHILE': #while开头的行
i = ' '*indent + i
indent += 2
elif (i[:len(i)] == 'END_WHILE') or i[:10] == 'END_WHILE(' or i[:10] == 'END_WHILE ' :#end_while开头的行
indent -= 2
i = ' '*indent + i
elif i[:4] == 'FOR ': #for开头的行
i = ' '*indent + i
indent += 2
elif (i[:len(i)] == 'END_FOR') or (i[:8] == 'END_FOR(')or (i[:8] == 'END_FOR '): #end_for开头的行
indent -= 2
i = ' '*indent + i
elif (i[:5]=='CASE '): #CASE开头的行
i = ' '*indent + i
indent += 4
elif (i[:len(i)]=='END_CASE') or (i[:9]=='END_CASE(') or (i[:9]=='END_CASE '):#END_CASE开头的行
indent -= 4
i = ' '*indent + i
else:
i = ' '*indent + i
file_out.write(i + '\n')
print '转换完成！'
file_in.close()
file_out.close()
raw_input("按回车键返回...")
好像不能用呢
页:
[1]