moods.py

来自「pyLJclient是一个跨平台的livejournal客户端」· Python 代码 · 共 52 行

import pickle, os

MOODS_CACHE = 'mood_cache'

class Mood:
    def __init__(self, id='',name=''):
	self.id = id
	self.name = name

class Moods:
    def __init__(self, mood_init_list=[]):
	self.list = self.read_mood_cache()
	if mood_init_list:
	    for mood in mood_init_list:
		self.list.append(mood)
		self.write_mood_cache()
	  
    def add_mood(self, mood):
	self.list.append(mood)
	self.write_mood_cache()

    def __getitem__(self, i):
	return self.list[i]

    def update(self, update_list):
	for i in update_list:
	    self.list.append(i)
	self.write_mood_cache()

    def read_mood_cache(self):
	try:
	    f = open(os.path.join(os.getcwd(), MOODS_CACHE), 'r')
	except:
	    return []
	else:
	    return pickle.load(f)

    def write_mood_cache(self):
	try:
	    f = open(os.path.join(os.getcwd(), MOODS_CACHE), 'w')
	except:
	    print "Cannot open (%s) file to write mood cache file"%(os.path.join(os.getcwd(), MOODS_CACHE))
	else:
	    pickle.dump(self.list, f)
	    f.close()
    
    def getMoods(self):
	return self.list

    def getmoodcount(self):
	return len(self.list)