✨ 更新工具类

1 files changed, 6 insertions, 0 deletions

diff --git a/ATRI/utils/__init__.py b/ATRI/utils/__init__.py
index 0fc4dbb..9d92977 100644
--- a/ATRI/utils/__init__.py
+++ b/ATRI/utils/__init__.py
@@ -1,10 +1,12 @@
 import os
 import re
+import string
 import pytz
 import yaml
 import aiofiles
 import time
 from pathlib import Path
+from random import sample
 from datetime import datetime
 from PIL import Image, ImageFile
 from aiofiles.threadpool.text import AsyncTextIOWrapper
@@ -44,6 +46,10 @@ def safe_string(value):
     return str(value)
 
 
+def gen_random_str(k: int) -> str:
+    return str().join(sample(string.ascii_letters + string.digits, k))
+
+
 class ListDealer:
     """
     对列表进行处理