代码加行号以及去行号

2019独角兽企业重金招聘Python工程师标准>>>

import fileinput,re,sys
re_pat=re.compile(r'#\s*(\d)*')
scope={}
def replacement(match):
    code=match.group(1)
    try:
        return ''
    except SyntaxError:
        print 'error!'
        break
lines=[]
def minus_number():
    for i in fileinput.input(sys.arg[2]):
        lines.append(i)
    text=''.join(lines)
    print re_pat.sub(replacement,text)

def add_number():
    for i in fileinput.input(sys.arg[2]):
        line=line.restrip()
        num=fileinput.lineno()
        print '%-80s # %2i' % (line,num)
        
def usage():
    print 'add(-a) or minus(-m) + filename'
    
def main():
    if sys.argv[1]=='-a':
        add_number()
    elif sys.argv[1]=='-m':
        minus_number()
    else:
        usage()
        
if __name__ == '__main__':
    main()

转载于:https://my.oschina.net/tigereatspinach/blog/352371