diff --git a/auto_cache.py b/auto_cache.py
index 34a32db..28cb3c4 100644
--- a/auto_cache.py
+++ b/auto_cache.py
@@ -6,14 +6,14 @@ import redis
 import urllib.request
 import influxdb_api
 import msgpack
-from datetime import datetime
+import datetime
 
 # 将 Query的信息 序列号到 redis/json， 默认不支持datetime，需要自定义
 # 自定义序列化函数
 # 序列化处理器
 def encode_datetime(obj):
     """将datetime转换为可序列化的字典结构"""
-    if isinstance(obj, datetime):
+    if isinstance(obj, datetime.datetime):
         return {
             '__datetime__': True,
             'as_str': obj.strftime("%Y%m%dT%H:%M:%S.%f")
@@ -24,7 +24,7 @@ def encode_datetime(obj):
 def decode_datetime(obj):
     """将字典还原为datetime对象"""
     if '__datetime__' in obj:
-        return datetime.strptime(
+        return datetime.datetime.strptime(
             obj['as_str'], "%Y%m%dT%H:%M:%S.%f"
         )
     return obj