100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
||
import sys
|
||
|
||
path = r'C:\Users\jsmbz\Documents\flutter-app-02\lib\features\home\presentation\widgets\map\testmap_pages.dart'
|
||
|
||
with open(path, 'r', encoding='utf-8') as f:
|
||
content = f.read()
|
||
|
||
# Find the "存在任务" branch that just returns without saving taskId
|
||
# We'll replace the "return;" in that branch with code that does async requery
|
||
# But fold callbacks can't be async...
|
||
|
||
# Instead, let's find the fold end and add async handler after it
|
||
# Find: "\n );\n } catch (e) {"
|
||
old_pat = '\n );\n } catch (e) {'
|
||
idx = content.find(old_pat)
|
||
if idx < 0:
|
||
print('Pattern not found, trying variants...')
|
||
# Try with different whitespace
|
||
old_pat = ' );\n } catch (e) {'
|
||
idx = content.find(old_pat)
|
||
if idx < 0:
|
||
print('Still not found')
|
||
sys.exit(1)
|
||
|
||
print(f'Found at index {idx}')
|
||
|
||
# Insert code between the fold close and the catch
|
||
# We'll restructure to handle needRequery
|
||
insert_code = '''
|
||
if (needRequery) {
|
||
debugPrint('设备已有任务，重新查询任务池...');
|
||
await taskCubit.fetchAndFilterTask(deviceId);
|
||
await Future.delayed(const Duration(milliseconds: 300));
|
||
final existingTasks = taskCubit.state.activeTasks;
|
||
if (existingTasks.isNotEmpty) {
|
||
final existingTask = existingTasks.first;
|
||
taskCubit.selectTask(existingTask);
|
||
debugPrint('复用已有任务 #${existingTask.id}，taskId已保存');
|
||
_showPageToast(message: "设备正在作业中", type: ToastType.success);
|
||
_saveDataToLocal();
|
||
return;
|
||
}
|
||
_showPageToast(message: "设备正在作业中", type: ToastType.warn);
|
||
return;
|
||
}
|
||
} catch (e) {'''
|
||
|
||
# Now find and fix the fold callback to set needRequery flag
|
||
# Replace the "存在任务" branch
|
||
old_branch = '''if (failMsg.contains('存在任务') || failMsg.contains('already exists')) {
|
||
debugPrint('⚠️ [开始作业] 设备已有任务，保留作业状态');
|
||
_showPageToast(message: "设备正在作业中", type: ToastType.warn);
|
||
// 🔥 不重置 _workStatus，按钮保持可见
|
||
return;'''
|
||
|
||
new_branch = '''if (failMsg.contains('存在任务') || failMsg.contains('already exists')) {
|
||
needRequery = true;
|
||
return;'''
|
||
|
||
branch_idx = content.find(old_branch)
|
||
if branch_idx < 0:
|
||
# Try without the debug print variation
|
||
old_branch = '''if (failMsg.contains('存在任务') || failMsg.contains('already exists')) {
|
||
debugPrint'''
|
||
branch_idx = content.find(old_branch)
|
||
if branch_idx < 0:
|
||
print('Branch pattern not found')
|
||
else:
|
||
print(f'Branch found at {branch_idx} (partial match)')
|
||
else:
|
||
print(f'Branch found at {branch_idx}')
|
||
# Do the branch replacement
|
||
content = content[:branch_idx] + new_branch + content[branch_idx + len(old_branch):]
|
||
|
||
# Also add needRequery var declaration before fold
|
||
# Find " result.fold(" and add var before it
|
||
fold_marker = ' result.fold('
|
||
fold_idx = content.find(fold_marker)
|
||
if fold_idx >= 0:
|
||
# Check if needRequery is already declared
|
||
check = content.find('needRequery', fold_idx - 50, fold_idx)
|
||
if check < 0:
|
||
content = content[:fold_idx] + ' var needRequery = false;\n' + content[fold_idx:]
|
||
print('Added needRequery declaration')
|
||
|
||
# Now find and replace the fold end pattern again (after content modifications)
|
||
old_pat = '\n );\n } catch (e) {'
|
||
idx = content.find(old_pat)
|
||
if idx < 0:
|
||
print('Fold end pattern not found after modifications')
|
||
sys.exit(1)
|
||
|
||
content = content[:idx + len('\n );')] + insert_code + content[idx + len(old_pat):]
|
||
|
||
with open(path, 'w', encoding='utf-8', newline='') as f:
|
||
f.write(content)
|
||
|
||
print('DONE')
|