GnoteBook 【Python版】|程序设计 - 精英盒子

import os

print "........................................."
print "欢迎使用GnoteBook"
print "输入1来新建文档"
print "输入2来编辑文档"
print "输入3来退出程序"
print "........................................."

while 1:
    mode = int(raw_input("请选择模式：")) #选择模式
    file_name = ""

    if mode == 1:
        while 1:
            file_name = raw_input("给你的便签起个名字吧：")
            if os.path.exists(file_name):
                print "文件已存在，换个名字吧！"
            else:
                break
        all = []
        print "开始编辑，单独输入'.'来退出编辑"
        while 1:
            file_string = raw_input('>')
            if file_string == '.':
                break
            else:
                all.append(file_string)
        ls = os.linesep
        file_obj = open(file_name+".txt", 'w')
        file_obj.writelines(["%s%s" % (x, ls) for x in all])
        file_obj.close()
        print "写入完成！"
    elif mode == 2:
        while 1:
            file_name = raw_input("你的便签叫什么？")
            if os.path.exists(file_name+".txt"):
                break
            else:
                print "文件不存在！！"
        print "文件内容如下："
        file_obj = open(file_name+".txt", 'r')
        for line in file_obj:
            print line,
        file_obj.close()
    elif mode == 3:
        break