python MongoCache

class MongoCache(object):
    """
    数据库缓存
    """
    def __init__(self,client=None,expires=timedelta(days=30)):
        self.client = MongoClient("localhost",27017)
        self.db = self.client.cache
        self.db.webpage.create_index('timestamp',expireAfterSeconds=expires.total_seconds())


    def __setitem__(self, key, value):
        record = {"result":Binary(zlib.compress(pickle.dumps(value))),"timestamp":datetime.utcnow()}
        self.db.webpage.update({"_id":key},{'$set':record},upsert=True)

    def __getitem__(self, item):
        record = self.db.webpage.find_one({"_id":item})
        if record:
            return pickle.loads(zlib.decompress(record["result"]))
        else:
            raise KeyError(item + "does not exist")

    def __contains__(self, item):
        try:
            self[item]#调用getitem方法

        except KeyError:
            return False
        else:
            return True

    def clear(self):
        self.db.webpage.drop()
    
cai_url ="http://www.runoob.com/mongodb/mongodb-insert.html"

cai_mongo = MongoCache()

cai_mongo[cai_url]=requests.get(url=cai_url).content#调用__setitem__

print(cai_mongo[cai_url].decode('utf-8'))#调用__getitem__