核心编程第三章

#readNwriteTextFiles.py

import os
ls = os.linesep

def makeTextFile():
    'makeTextFile.py -- create text file'

    # get filename
    while True:
        fname = raw_input("your filename:")
        try:
            fobj = open(fname)
        except IOError:
            break
        else:
            print "already exsits."
            
    #get file content (text) line
    all = []
    print "\nEnter lines ('.' by itself to quit).\n"

    #loop until user terminates input 
    while True:
        entry = raw_input("what you want to write in:")
        if entry == '.':
            break
        else:
            all.append(entry)
                
    #write lines to file with proper line-ending
    fobj = open(fname, 'w')
    fobj.writelines(['%s%s' % (x, ls) for x in all])
    fobj.close()
    print 'DONE!'



def readTextFile():
    'readTextFile.py -- read and display text file'

    # get filename 
    fname = raw_input('Enter filename:')
    print

    # attempt to open file for reading
    if os.path.exists(fname):
        fobj = open(fname, 'r')
        # display contents to the screen
        for eachLine in fobj:
            eachLine=eachLine.strip(ls)
            print eachLine
        fobj.close()
    else:
        print "ERROR: '%s' already exists." % fname    

if __name__ == '__main__':        
    print """
a.create file 
b.read file
"""
    choose = raw_input("your choice:")    
    if choose == 'a':
        makeTextFile()
    else:
        readTextFile()

转载于:https://my.oschina.net/u/2519674/blog/650130