python 删除给定字符串的空格或新行

Posted 2021-05-09

#!/usr/bin/python

# Works at least for python 2.7.3

import sys
import os
import re

# testStr is for test usage, when they are assinged to the variable s in main() below
testStr1="a b\n c \n d "
#testStr2='/path/to/white_space/file'

def doStr(s):
    pattern = re.compile(r'\s+')
    strg = re.sub(pattern, '', s)
    print 'Ridded result:\n', strg

def doFile(fPath):
    with open(fPath, 'r') as f:
        readed = [line.rstrip() for line in f]
    strReaded = ''.join(readed)
    print strReaded
    doStr(strReaded)

def main():
    print 'Args: ',str(sys.argv) 
    #s = testStr1
    s = ''
    if len(sys.argv) > 1:
        s=sys.argv[1]

    if os.path.isfile(s):
        doFile(s)
    elif len(s) > 0:
        doStr(s)
    else:
        print "Nothing I can do"

main()