新增/修改部分用例

utils/data_utils/data_handle.py

@@ -11,6 +11,7 @@ import re, uuid
 import copy
 import os
 import base64
+from datetime import datetime, timedelta
 # 第三方库导入
 from loguru import logger
 from string import Template
@@ -307,6 +308,35 @@ def update_wiki_sidebar(sidebar_content, new_page_name):
     return string_to_base64(new_sidebar_content)
 
 
+def get_current_week(start_or_end="start"):
+    """
+    获取当前日期，并根据参数返回本周的开始或结束日期。
+
+    参数:
+    - start_or_end: 字符串，指定返回本周的开始日期（"start"）还是结束日期（"end"）。
+
+    返回:
+    - 本周开始或结束日期的字符串表示，格式为"月日"（例如："01月01日"）。
+    """
+    # 获取当前日期
+    today = datetime.today()
+    # 计算今天是本周的第几天（0代表周一，1代表周二，以此类推）
+    current_weekday = today.weekday()
+
+    if start_or_end == "start":
+        # 计算本周的周一
+        res = today - timedelta(days=current_weekday)
+    elif start_or_end == "end":
+        # 计算本周的周日
+        res = today - timedelta(days=current_weekday) + timedelta(days=6)
+    else:
+        # 如果参数非法，返回当前日期的周一
+        logger.error(f"Invalid value for start_or_end: {start_or_end}. Defaulting to 'start'.")
+        res = today - timedelta(days=current_weekday)
+
+    return res.strftime("%m月%d日")
+
+
 # 声明data_handle方法，这样外部就可以直接import data_handle来使用了
 data_handle = DataHandle().data_handle