python 寫個簡單TUI日記本

logfunction.py:

#!/usr/bin/env python
#-*- coding:utf-8 -*-


import sys
import os
import time

def quit():
        sys.exit(0)

def writelog():
        name = time.strftime('%Y-%m-%d',time.localtime(time.time()))
        
        try:
                fd = open(name, 'w')
        except IOError:
                print u'文件不存在啊...'
        endio = True
        print u"輸入日記:"
        while endio:
                line = raw_input()
                if line == 'end':
                        break  
                fd.write(line + '\n')
        fd.close()          
    
def readlog():
        print u'輸入日記名:'
        name = raw_input()
        try:
                fd = open(name, 'r')
        except IOError:
                print u'文件不存在啊...'
        lines = fd.readlines()
        for line in lines:
                print line
        fd.close()

log.py:

#!/usr/bin/env python
#-*- coding:utf-8 -*-

from logfunction import *


FuntionAct = {'w':writelog, 'r':readlog, 'q':quit}

end = True

while end:
        
        print u'寫日記(w)'
        print u'讀日記(r)'
        print u'退出(q)'
        
        choice = raw_input('輸入操作:')
        
        FuntionAct[choice]()