Python一键生成核酸检测日历的操作代码

import pandas as pd
import uiautomation as auto

pane = auto.PaneControl(searchDepth=1, Name="深i您 - 自主申报")
pane.SetActive(waitTime=0.01)
pane.SetTopmost(waitTime=0.01)

tags = pane.GetFirstChildControl().GetFirstChildControl() \
    .GetLastChildControl().GetFirstChildControl() \
    .GetChildren()
result = []
for tag in tags:
    tmp = []
    for item, depth in auto.WalkControl(tag, maxDepth=4):
        if item.ControlType != auto.ControlType.TextControl:
            continue
        tmp.append(item.Name)
    row = {"检测结果": tmp[1]}
    for k, v in zip(tmp[2::2], tmp[3::2]):
        row[k[:-1]] = v
    result.append(row)
pane.SetTopmost(False, waitTime=0.01)
df = pd.DataFrame(result)
df.采样时间 = pd.to_datetime(df.采样时间)
df.检测时间 = pd.to_datetime(df.检测时间)
df.head()

df.to_excel(f"{pd.Timestamp('now').date()}核酸检测记录.xlsx", index=False)

import pandas as pd
import uiautomation as auto

pane = auto.PaneControl(searchDepth=1, Name="深i您 - 自主申报")
pane.SetActive(waitTime=0.01)
pane.SetTopmost(waitTime=0.01)

tag = pane.GetFirstChildControl().GetFirstChildControl() \
    .GetLastChildControl().GetFirstChildControl()
tmp = []
for item, depth in auto.WalkControl(tag, maxDepth=2):
    if item.ControlType != auto.ControlType.TextControl:
        continue
    tmp.append(item.Name)
result = []
tags = tag.GetChildren()
if tmp:
    row = {"检测结果": tmp[1]}
    for k, v in zip(tmp[2::2], tmp[3::2]):
        row[k[:-1]] = v
    result.append(row)
for tag in tags:
    if tag.Name or tag.GetFirstChildControl().Name:
        continue
    tmp = []
    for item, depth in auto.WalkControl(tag, maxDepth=4):
        if item.ControlType != auto.ControlType.TextControl:
            continue
        tmp.append(item.Name)
    row = {"检测结果": tmp[1]}
    for k, v in zip(tmp[2::2], tmp[3::2]):
        row[k[:-1]] = v
    result.append(row)
pane.SetTopmost(False, waitTime=0.01)
df = pd.DataFrame(result)
df.采样时间 = pd.to_datetime(df.采样时间)
df.检测时间 = pd.to_datetime(df.检测时间)
df.head()

import calendar
from PIL import Image, ImageFont, ImageDraw


def create_calendar_img(year, month, days):
    "作者：小小明 https://xxmdmst.blog.csdn.net"
    font = ImageFont.truetype('msyh.ttc', size=20)
    im = Image.new(mode='RGB', size=(505, 251), color="white")
    draw = ImageDraw.Draw(im=im)
    draw.rectangle((0, 0, 504, 40), fill='#418CFA', outline='black', width=1)
    draw.text(xy=(170, 0), text=f"{year}年{month}月", fill=0,
              font=ImageFont.truetype('msyh.ttc', size=30))
    title_datas = ['星期一', '星期二', '星期三', '星期四', '星期五', '星期六', '星期日']
    for i, title_data in enumerate(title_datas):
        draw.rectangle((i*72, 40, (i+1)*72, 70),
                       fill='lightblue', outline='black', width=1)
        draw.text(xy=(i*72+7, 40), text=title_data, fill=0,
                  font=font)

    # 第一天是星期几和一个月的天数
    weekday, day_num = calendar.monthrange(year, month)
    col, row = weekday, 1
    for i in range(1, day_num+1):
        if col >= 7:
            col = 0
            row += 1
        fill = "#009B3C" if i in days else None
        draw.rectangle((col*72, 40+30*row, (col+1)*72, 40+30*(row+1)),
                       fill=fill, outline='black', width=1)
        draw.text(xy=(col*72+24, 40+30*row), text=str(i).zfill(2), fill=0,
                  font=font)
        col += 1
    return im

dates = df.采样时间.dt.date.sort_values().astype(str)
for month, date_split in dates.groupby(dates.str[:7]):
    year, month = map(int, month.split("-"))
    days = date_split.str[-2:].astype(int)
    im = create_calendar_img(year, month, days.values)
    display(im)

im.save(f"{year}年{month}月检测记录.jpg")