怎么用python判断，文件中的等于号“=”,小于号“<”的前后是否含有空格，没有的话，就添加空格

Question

比如文本中有如下代码：
...
for(i=1;i<10;i++)
{
...
}
改为：
for(i = 1;i < 10;i++)
...
如何实现？如果有空格则不变。

Answer 1

import re
code=open('test.c').read()
newcode=re.sub(r"(?<! )(=|<)",r" \1",code)
newcode=re.sub(r"(=|<)(?! )",r"\1 ",newcode)
newfile=open('test.c','w')
newfile.write(newcode)
newfile.close()