import os, time, md5 from sugar import env from sugar import util class BuddyIconCache(object): """Caches icons on disk and finds them based on md5 hash.""" def __init__(self): ppath = env.get_profile_path() self._cachepath = os.path.join(ppath, "cache", "buddy-icons") if not os.path.exists(self._cachepath): os.makedirs(self._cachepath) self._cache = {} # Read all cached icons and their sums for fname in os.listdir(self._cachepath): m = md5.new() data = self._get_icon_data(fname) if len(data) == 0: continue m.update(data) printable_hash = util.printable_hash(m.digest()) self._cache[printable_hash] = fname del m def _get_icon_data(self, fname): fd = open(os.path.join(self._cachepath, fname), "r") data = fd.read() fd.close() del fd return data def get_icon(self, printable_hash): if type(printable_hash) != type(u""): raise RuntimeError("printable_hash must be a unicode string.") try: fname = self._cache[printable_hash] return self._get_icon_data(fname) except KeyError: pass return None def add_icon(self, icon_data): if len(icon_data) == 0: return m = md5.new() m.update(icon_data) printable_hash = util.printable_hash(m.digest()) if self._cache.has_key(printable_hash): del m return # Write the icon to disk and add an entry to our cache for it m.update(time.asctime()) fname = util.printable_hash(m.digest()) fd = open(os.path.join(self._cachepath, fname), "w") fd.write(icon_data) fd.close() self._cache[printable_hash] = fname del m