import sqlite3
import os

DATABASE_PATH = os.path.join(os.path.dirname(__file__), '../database/app.db')

def clean_test_tasks():
    conn = sqlite3.connect(DATABASE_PATH)
    cursor = conn.cursor()
    
    print('=== 清理测试任务 ===')
    cursor.execute('''
        DELETE FROM tasks 
        WHERE id IN (49, 50, 51, 52)
    ''')
    deleted_count = cursor.rowcount
    print(f'删除了 {deleted_count} 个测试任务')
    
    conn.commit()
    conn.close()
    
    print('清理完成！')

if __name__ == '__main__':
    clean_test_tasks()