11

core/function_basic.py

@@ -4,7 +4,7 @@ import matplotlib.pyplot as plt
 from matplotlib.font_manager import FontProperties
 from io import BytesIO
 import base64
-from datetime import datetime
+from datetime import datetime,date
 import calendar
 
 def load_template(template_name):
@@ -169,37 +169,28 @@ def plot_line_chart_base64(data, x_field, y_field, title=None, reverse_x=False):
     return f"data:image/png;base64,{img_base64}"
 
 
-def build_calendar_items(start_date_str, data):
-    start_date = datetime.datetime.strptime(start_date_str, "%Y-%m-%d").date()
-    weekday = start_date.weekday()  # 周一0 ~ 周日6
-    weekday = (weekday + 1) % 7     # 调整：周日=0, 周一=1 ... 周六=6
+def week_to_num(week :str):
+    week_map = {
+    "一": 0,
+    "二": 1,
+    "三": 2,
+    "四": 3,
+    "五": 4,
+    "六": 5,
+    "日": 6,  
+    }
+    return week_map.get(week,None)
 
-    result = []
-    row = []
+def compare_date_str(date_str: str) -> str:
+    """
+    date_str 格式：YYYY-MM-DD
+    """
+    d = datetime.strptime(date_str, "%Y-%m-%d").date()
+    today = date.today()
 
-    # 第一行补空
-    for _ in range(weekday):
-        row.append({"date": "", "items": [], "is_today": False})
-
-    # 排数据
-    for i, day in enumerate(data):
-        # 识别今天
-        is_today = (day["date"] == start_date_str)
-
-        row.append({
-            "date": day["day"],
-            "items": [day],
-            "is_today": is_today,
-        })
-
-        if len(row) == 7:
-            result.append(row)
-            row = []
-
-    # 最后一行剩余补空
-    if row:
-        while len(row) < 7:
-            row.append({"date": "", "items": [], "is_today": False})
-        result.append(row)
-
-    return result
+    if d < today:
+        return "过去"
+    elif d == today:
+        return "今天"
+    else:
+        return "将来"