代码改变世界

python去除行首行号

2008-11-20 19:10  ubunoon  阅读(495)  评论(0编辑  收藏  举报
有时候,从网络文档中下载的代码,往往是带有行号的文档,这个时候,我们需要将行首的行号去除,才能够在我们的文档中运行,下面的代码是为去除行首代码模块,python实现:

#!/usr/bin/env python
#
coding=utf-8

# remove line number before copy code from other
#
 file

class Parser():
    
""" 
    Parse the file, and remove the line number 
    of given text,and remove the left and right space
    
"""
    
def __init__(self,raw):
        self.raw 
= raw.splitlines()
        self.result
=[]
        
    
def remove(self):
        
# find lines of the file 
        #lines = len(self.raw)
        number =  1
        
for line in  self.raw:
            line 
= line.lstrip().rstrip().replace(str(number),'')
            self.result.append(line)
            number 
+=1
        
return "\n".join(self.result)

def test():
    testlines 
= "1 print 'hello world' \n\
    2 if __name == 'main': \n\
3   print 'third line '   \n\
"
    print Parser(testlines).remove()

if __name__ == "__main__":
    test()