嘗試自己寫一個Python緩存裝飾器

本文記錄嘗試編寫一個簡單的緩存裝飾器，以學習為目的，實際生產環境建議大家用標準庫。

from hashlib import md5
from pickle import dump, load

# 用pickle進行數據的讀取、寫入
def _dkL(f):
    with open(f,'rb') as file:  return load(file)
def _dkD(o,f):
    with open(f, 'wb') as file:  return dump(o,file)

def cache(ex_time=10,
    start=0,
    have_args=True,
    have_kw=True,
    cache_path="/tmp/pyCache"):

    # 判斷緩存目錄是否存在
    if not path.exists(cache_path): makedirs(cache_path)

    def _func(f):
        def _dec(*args, **kw):

                        # args參數合并成字符串
            _func_args = ''.join([ str(_) for _ in args][start:]) if have_args else ''

            # kw參數合并成字符串
            _func_kw   = ''.join([ "%s-%s" % (i,kw[i]) for i in kw if kw]) if have_kw else ''

            # 用于識別方法名的字符串
            _func_str  = f.__qualname__ + _func_args + _func_kw 

            # 進行md5加密
            _md5 = md5()
            _md5.update(_func_str.encode('utf-8'))
            _func_md5 = _md5.hexdigest()

            # 方法執行的緩存位置
            file_path = path.join(cache_path, _func_md5)

                                # 判斷方法執行結果是否過期
            if path.exists(file_path):
                file_mtime = int(path.getmtime(file_path))
                if int(time()) - file_mtime <= int(ex_time):
                    return _dkL(file_path)
            func_result = f(*args, **kw)
            _dkD(func_result, file_path)
            return func_result
        return _dec
    return _func

使用方法

    @cache(60, have_args=False)
    def search_all(self):
        ....

實測結果

測試兩個相同的請求，第一次沒有緩存，第二次讀取了緩存