要让 main 函数在报错后继续运行而不是停止整个程序，你需要在调用 main 的地方添加异常处理。以下是修改后的代码方案：

修改方案

1. 首先，创建一个安全的 main 包装函数：

python

复制

下载

def safe_main(sql: str, role: str, levelname: str, sendplan: str, plantime: datetime, 
             resultlist='{}', savepath='', savepath2=''):
    try:
        main(sql, role, levelname, sendplan, plantime, resultlist, savepath, savepath2)
    except Exception as e:
        current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        error_msg = f"[{current_time}] Error processing SQL: {sql}\nError details: {str(e)}"
        print(error_msg)
        
        # 可以选择记录错误到数据库
        try:
            connection = get_db_connection()
            with connection.cursor() as cursor:
                insert_query = """INSERT INTO ai_error_log 
                                (sqlcmd, error_message, error_time, role) 
                                VALUES (%s, %s, %s, %s)"""
                cursor.execute(insert_query, (sql, str(e), datetime.now(), role))
                connection.commit()
        except Exception as db_error:
            print(f"Failed to log error to database: {db_error}")
        finally:
            if 'connection' in locals():
                connection.close()

2. 然后在所有调用 main 的地方替换为调用 safe_main：

python

复制

下载

# 原代码中的这些部分
# main(sql, companypartition, '一般', '2', morning_datetime)
# 替换为
safe_main(sql, companypartition, '一般', '2', morning_datetime)