From 0c56e20c9df4b7fdecf174232f2c0305ae0d48cc Mon Sep 17 00:00:00 2001
From: xaoyaoo <xaoyaoo@163.com>
Date: Sun, 18 Aug 2024 10:50:20 +0800
Subject: [PATCH] rename core_db_type

---
 pywxdump/wx_core/merge_db.py       | 4 ++--
 pywxdump/wx_core/utils/__init__.py | 2 +-
 pywxdump/wx_core/wx_info.py        | 8 ++++----
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/pywxdump/wx_core/merge_db.py b/pywxdump/wx_core/merge_db.py
index ad773ac..e3ff076 100644
--- a/pywxdump/wx_core/merge_db.py
+++ b/pywxdump/wx_core/merge_db.py
@@ -15,7 +15,7 @@ from typing import List
 
 from .decryption import batch_decrypt
 from .wx_info import get_core_db
-from .utils import wx_core_loger, wx_core_error, DB_TYPE_CORE
+from .utils import wx_core_loger, wx_core_error, CORE_DB_TYPE
 
 
 @wx_core_error
@@ -467,7 +467,7 @@ def all_merge_real_time_db(key, wx_path, merge_path: str, real_time_exe_path: st
         from pywxdump import get_core_db
     except ImportError:
         return False, "未找到模块 pywxdump"
-    db_paths = get_core_db(wx_path, DB_TYPE_CORE)
+    db_paths = get_core_db(wx_path, CORE_DB_TYPE)
     if not db_paths[0]:
         return False, db_paths[1]
     db_paths = db_paths[1]
diff --git a/pywxdump/wx_core/utils/__init__.py b/pywxdump/wx_core/utils/__init__.py
index bbfea24..7ae0270 100644
--- a/pywxdump/wx_core/utils/__init__.py
+++ b/pywxdump/wx_core/utils/__init__.py
@@ -12,5 +12,5 @@ from .ctypes_utils import get_process_list, get_memory_maps, get_process_exe_pat
 from .memory_search import search_memory
 from ._loger import wx_core_loger
 
-DB_TYPE_CORE = ["MicroMsg", "MSG", "MediaMSG", "OpenIMContact", "OpenIMMsg", "PublicMsg", "OpenIMMedia",
+CORE_DB_TYPE = ["MicroMsg", "MSG", "MediaMSG", "OpenIMContact", "OpenIMMsg", "PublicMsg", "OpenIMMedia",
                 "Favorite", "Sns"]
diff --git a/pywxdump/wx_core/wx_info.py b/pywxdump/wx_core/wx_info.py
index 24c4071..3a685da 100644
--- a/pywxdump/wx_core/wx_info.py
+++ b/pywxdump/wx_core/wx_info.py
@@ -14,7 +14,7 @@ from typing import List, Union
 from .utils import verify_key, get_exe_bit, wx_core_error
 from .utils import get_process_list, get_memory_maps, get_process_exe_path, get_file_version_info
 from .utils import search_memory
-from .utils import wx_core_loger, DB_TYPE_CORE
+from .utils import wx_core_loger, CORE_DB_TYPE
 import ctypes.wintypes as wintypes
 
 # 定义常量
@@ -415,15 +415,15 @@ def get_core_db(wx_path: str, db_types: list = None) -> [dict]:
     """
     获取聊天消息核心数据库路径
     :param wx_path: 微信文件夹路径 eg：C:\*****\WeChat Files\wxid*******
-    :param db_types: 数据库类型 eg: DB_TYPE_CORE，中选择一个或多个
+    :param db_types: 数据库类型 eg: CORE_DB_TYPE，中选择一个或多个
     :return: 返回数据库路径 eg: [{"wxid": wxid, "db_type": db_type, "db_path": db_path, "wxid_dir": wxid_dir}, ...]
     """
     if not os.path.exists(wx_path):
         return False, f"[-] 目录不存在: {wx_path}"
 
     if not db_types:
-        db_types = DB_TYPE_CORE
-    db_types = [dt for dt in db_types if dt in DB_TYPE_CORE]
+        db_types = CORE_DB_TYPE
+    db_types = [dt for dt in db_types if dt in CORE_DB_TYPE]
     msg_dir = os.path.dirname(wx_path)
     my_wxid = os.path.basename(wx_path)
     wxdbpaths = get_wx_db(msg_dir=msg_dir, db_types=db_types, wxids=my_wxid)