Python练习2--文件读取

文件读取

# -*- encoding:utf-8 -*-
import os

def WriteFunc():
    # 换行'\r\n'
    ls = os.linesep
    filename = raw_input('Enter filename')
    while True:
        #文件处理，判断文件是否存在
        try:
            fobj = open(filename, 'w')
        except IOError,e:
            print "%s already exists"%filename
        else:
            break
    all = []
    print "\nEnter lines（'.' by itself to quit）.\n"
    while True:
        #读取内容，每一行为列表的一个元素
        entry = raw_input('>')
        if entry == '.':
            break
        else:
            all.append(entry)
    # 将列表每个元素之间用'\r\n'间隔，写入文件
    fobj.writelines(['%s%s'%(x,ls) for x in all])
    fobj.close()
    print 'DONE'


def ReadFunc():
    fname = raw_input('Enter filename:')
    # 文件处理
    if not os.path.exists(fname):
        print  "*** file open error, not exist:"
    else:
        fobj = open(fname, 'r')
        for eachLine in fobj:
            print  eachLine
        fobj.close()

if __name__ == '__main__':
    while True:
        ch = raw_input('1.Write a file:\n2.Read a file\n0.退出\nYou choose:')
        if ch == '1':
            WriteFunc()
        elif ch == '2':
            ReadFunc()
        elif ch == '0':
            break
        else:
            print '输入无效选择，请重新输入!\n'