olivebot/agent/agent_service.py

import sqlite3
import threading
import datetime
import time
from agent.fay_agent import FayAgentCore

scheduled_tasks = {}
agent_running = False
agent = FayAgentCore()

# 数据库初始化
def init_db():
    conn = sqlite3.connect('timer.db')
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS timer (
            time TEXT NOT NULL,
            repeat_rule TEXT NOT NULL,
            content TEXT NOT NULL
        )
    ''')
    conn.commit()
    conn.close()

    

# 插入测试数据
def insert_test_data():
    conn = sqlite3.connect('timer.db')
    cursor = conn.cursor()
    cursor.execute("INSERT INTO timer (time, repeat_rule, content) VALUES (?, ?, ?)", ('16:20', '1010001', 'Meeting Reminder'))
    conn.commit()
    conn.close()

# 解析重复规则
def parse_repeat_rule(rule, task_time):
    today = datetime.datetime.now()
    if rule == '0000000':  # 不重复
        task_datetime = datetime.datetime.combine(today.date(), task_time)
        if task_datetime > today:
            return task_datetime
        else:
            return None
    for i, day in enumerate(rule):
        if day == '1' and today.weekday() == i:
            task_datetime = datetime.datetime.combine(today.date(), task_time)
            if task_datetime > today:
                return task_datetime
    return None

# 执行任务
def execute_task(task_time, content):
    print(f"Executing task scheduled for {task_time}: {content} --> {agent.run(content)}")
    # 如果不重复，执行后删除记录
    conn = sqlite3.connect('timer.db')
    cursor = conn.cursor()
    cursor.execute("DELETE FROM timer WHERE repeat_rule = '0000000' AND time = ? AND content = ?", (task_time.strftime('%H:%M'), content))
    conn.commit()
    conn.close()


# 检查并执行任务
def check_and_execute():
    while agent_running:
        conn = sqlite3.connect('timer.db')
        cursor = conn.cursor()
        cursor.execute("SELECT time, repeat_rule, content FROM timer")
        rows = cursor.fetchall()

        for row in rows:
            task_time_str, repeat_rule, content = row
            task_time = datetime.datetime.strptime(task_time_str, '%H:%M').time()
            next_execution = parse_repeat_rule(repeat_rule, task_time)

            if next_execution and (task_time_str not in scheduled_tasks or scheduled_tasks[task_time_str] < next_execution):
                # 更新线程
                if task_time_str in scheduled_tasks:
                    scheduled_tasks[task_time_str].cancel()
                timer_thread = threading.Timer((next_execution - datetime.datetime.now()).total_seconds(), execute_task, [next_execution, content])
                timer_thread.start()
                scheduled_tasks[task_time_str] = next_execution

        conn.close()
        time.sleep(30)  # 30秒扫描一次

# agent启动
def agent_start():
    global agent_running
    agent_running = True
    init_db()
    # insert_test_data()
    check_and_execute_thread = threading.Thread(target=check_and_execute)
    check_and_execute_thread.start()

    #初始计划
    agent.run("""
        请为我一个个时间设置初始计划,注意在安排计划时请先确定这定时任务是否已经存在：
        1、在每天12:30到13:30安静不影响主人休息;
        2、每天12点提醒主人吃饭;  
        3、在星期一到星期五13:30提醒主人开始工作; 
        4、在星期一到星期五15:15提醒主人冲咖啡; 
        5、在星期一、星期三的11:15提醒主人开会; 
        6、在星期五17:30提醒主人开会;
        7、在星期一到星期五18:00提醒主人下班;
        8、在每天21点陪主人聊聊天;  
        9、在每天晚上10:30会跟据天气预报信息和前一天的运行情况，检查iotm系统当天的控制规则；
        10、在每天早上8点、中午12点、晚上10点检查农业种植箱的状态是否附合设定的预期执行，
        如果不符合请告知我调整。
        另外：在任何意外的情况需要进一步指导请马联系我。
        """)

def agent_stop():
    global agent_running 
    global scheduled_tasks
    # 取消所有定时任务
    for task in scheduled_tasks.values():
        task.cancel()
    agent_running = False
    scheduled_tasks = {}
    

if __name__ == "__main__":
    agent_start()