Python实现ICS文件事件追加功能的解决方案咨询

你的问题在于每次调用ics()函数时，都会创建一个全新的Calendar对象，然后直接覆盖写入文件——这就导致之前添加的事件全部丢失了。要实现追加功能，核心思路是复用已有的Calendar实例（或者从现有文件加载），添加新事件后再重新写入。

下面给你两种实用的实现方案：

方案一：复用同一个Calendar对象（适合批量生成事件） #

如果你的所有事件是在同一个循环/流程里生成的，最高效的方式是先创建一个Calendar对象，循环添加所有事件后，一次性写入文件。

修改后的代码示例： #

from icalendar import Calendar, Event
from datetime import datetime, UTC
import os

# 先初始化日历对象（只做一次）
def init_calendar():
    cal = Calendar()
    cal.add('prodid', 'IVU月度排班工具')
    cal.add('version', '2.0')
    return cal

# 定义添加事件的函数
def add_event(cal, summary, description, dtstart, dtend):
    event = Event()
    event.add('summary', summary)
    event.add('description', description)
    event.add('dtstart', dtstart)
    event.add('dtend', dtend)
    event.add('dtstamp', datetime(2005,4,4,0,10,0,tzinfo=UTC))
    # 生成唯一UID（非常重要！避免日历客户端识别为重复事件）
    event['uid'] = f'{datetime.now().timestamp()}@ivu-scheduler.com'
    event.add('priority', 5)
    cal.add_component(event)

# 保存日历到文件
def save_calendar(cal, file_path):
    with open(file_path, 'wb') as f:
        f.write(cal.to_ical())
    print(f'日历文件已成功保存到 {file_path}')

# 调用示例（循环添加事件）
if __name__ == '__main__':
    myPath = './schedules'
    userfinder = '张三'
    year, month = 2024, 5
    ics_file_path = f'{myPath}/{userfinder}-{month}-{year}.ics'
    
    # 确保目录存在
    os.makedirs(myPath, exist_ok=True)
    
    # 初始化日历
    cal = init_calendar()
    
    # 循环添加事件（这里替换成你的实际数据循环）
    for day in range(1, 5):
        summary = f'{day}号早班'
        description = f'备注: 无, 船舶: 海王星号, 职能: 值班, 班次时长: 8小时'
        dtstart = datetime(year, month, day, 8, 0, 0, tzinfo=UTC)
        dtend = datetime(year, month, day, 16, 0, 0, tzinfo=UTC)
        
        add_event(cal, summary, description, dtstart, dtend)
        print(f'日期: {year}-{month}-{day} 班次: {summary} --> 成功添加')
    
    # 最后一次性保存
    save_calendar(cal, ics_file_path)

方案二：每次调用时加载现有文件（适合多次调用函数） #

如果你的事件不是一次性生成的（比如需要多次调用函数添加事件），可以每次调用时先检查ICS文件是否存在：存在则加载现有日历，添加新事件后重新写入；不存在则新建日历。

修改后的代码示例： #

from icalendar import Calendar, Event
from datetime import datetime, UTC
import os

def add_event_to_ics(summary, description, dtstart, dtend, userfinder, year, month, myPath):
    ics_file_path = f'{myPath}/{userfinder}-{month}-{year}.ics'
    
    # 加载现有日历或新建
    if os.path.exists(ics_file_path):
        with open(ics_file_path, 'rb') as f:
            cal = Calendar.from_ical(f.read())
    else:
        cal = Calendar()
        cal.add('prodid', 'IVU月度排班工具')
        cal.add('version', '2.0')
        # 确保目录存在
        os.makedirs(myPath, exist_ok=True)
    
    # 创建新事件
    event = Event()
    event.add('summary', summary)
    event.add('description', description)
    event.add('dtstart', dtstart)
    event.add('dtend', dtend)
    event.add('dtstamp', datetime(2005,4,4,0,10,0,tzinfo=UTC))
    # 唯一UID必须保证！
    event['uid'] = f'{datetime.now().timestamp()}@ivu-scheduler.com'
    event.add('priority', 5)
    
    # 添加事件并保存
    cal.add_component(event)
    with open(ics_file_path, 'wb') as f:
        f.write(cal.to_ical())
    
    print(f'日期: {dtstart.date()} 班次: {summary} --> 成功添加到ICS文件')

# 调用示例（可以多次调用这个函数）
add_event_to_ics(
    summary='5号晚班',
    description='备注: 加班, 船舶: 天王星号, 职能: 值守, 班次时长: 10小时',
    dtstart=datetime(2024,5,5,18,0,0,tzinfo=UTC),
    dtend=datetime(2024,5,6,4,0,0,tzinfo=UTC),
    userfinder='张三',
    year=2024,
    month=5,
    myPath='./schedules'
)

# 后续可以再次调用添加新事件
add_event_to_ics(
    summary='6号中班',
    description='备注: 正常排班, 船舶: 土星号, 职能: 操作, 班次时长: 8小时',
    dtstart=datetime(2024,5,6,12,0,0,tzinfo=UTC),
    dtend=datetime(2024,5,6,20,0,0,tzinfo=UTC),
    userfinder='张三',
    year=2024,
    month=5,
    myPath='./schedules'
)

关键注意事项： #

• UID唯一性：原代码中使用了固定的UID，这会导致日历客户端将所有事件识别为同一个，从而覆盖或忽略。一定要为每个事件生成唯一的UID（比如用时间戳+自定义域名）。
• 文件权限：确保你的程序有读写目标目录的权限，避免保存失败。
• 时区一致性：dtstart和dtend要使用相同的时区（比如示例中的UTC），否则日历客户端可能会显示错误的时间。

内容的提问来源于stack exchange，提问作者TUX One