From a4932af6b7cb07de1280948d88d481f35572ebaf Mon Sep 17 00:00:00 2001
From: xaoyo <xaoyaoo@qq.com>
Date: Tue, 24 Oct 2023 18:44:13 +0800
Subject: [PATCH] add auto get bias addr ,not need input key or wx folder path.

---
 pywxdump/decrypted/decrypt.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pywxdump/decrypted/decrypt.py b/pywxdump/decrypted/decrypt.py
index d823970..116b17d 100644
--- a/pywxdump/decrypted/decrypt.py
+++ b/pywxdump/decrypted/decrypt.py
@@ -2,7 +2,7 @@ import argparse
 import hmac
 import hashlib
 import os
-
+from typing import Union, List
 from Cryptodome.Cipher import AES
 
 # from Crypto.Cipher import AES # 如果上面的导入失败，可以尝试使用这个
@@ -55,7 +55,7 @@ def decrypt(key: str, db_path, out_path):
     return [True, db_path, out_path, key]
 
 
-def batch_decrypt(key: str, db_path: [str | list], out_path: str):
+def batch_decrypt(key: str, db_path: Union[str, List[str]], out_path: str):
     if not isinstance(key, str) or not isinstance(out_path, str) or not os.path.exists(out_path) or len(key) != 64:
         return f"[-] (key:'{key}' or out_path:'{out_path}') Error!"