From 4f749852957befe494cc202846d9dbaba7e45bd0 Mon Sep 17 00:00:00 2001
From: b3nguang <1127786222@qq.com>
Date: Fri, 13 Sep 2024 10:22:03 +0800
Subject: [PATCH] Update cache.py

---
 core/cache.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/core/cache.py b/core/cache.py
index d6c377f..85e7c9e 100644
--- a/core/cache.py
+++ b/core/cache.py
@@ -17,19 +17,17 @@ def set(self, key, value):
         self.cache[key] = (value, time.time())
 
     def get(self, key):
-        if key in self.cache:
+        item = self.cache.get(key)
+        if item:
             if not self._is_expired(key):
-                return self.cache[key][0]  # 返回值
+                return item[0]  # 返回值
             else:
                 del self.cache[key]  # 如果过期，删除该key
         return None
 
     def cleanup(self):
         # 清除过期的缓存
-        keys_to_delete = []
-        for key in list(self.cache):
-            if self._is_expired(key):
-                keys_to_delete.append(key)
+        keys_to_delete = [key for key in list(self.cache) if self._is_expired(key)]
         for key in keys_to_delete:
             del self.cache[key]