豆包生成文章ADB 方式采集问答内容对接 MySQL 入库

信息发布软件

豆包生成文章ADB 方式采集问答内容对接 MySQL 入库

1. /**
   
2. * 案例 12:豆包生成文章 + ADB 方式采集问答内容 + 对接 MySQL 入库
   
3. *
   
4. * 只教一件事:怎么用 ADB(uiautomator dump)把豆包对话页的问答文本整页抓下来,
   
5. * 攒成列表,再通过 SDK 的 mysql 模块直连数据库逐条入库。
   
6. *
   
7. * 流程(每个主题一轮):
   
8. *   打开豆包 → 输入框输入"写一篇 XX 的文章"并发送 → 轮询 dump 等生成稳定
   
9. *   → 对比发送前后的页面文本,新出现的长文本就是 AI 回答
   
10. *   → push 进内存列表 + INSERT 进 MySQL
    
11. *
    
12. * 采集原理:
    
13. *   adb.exec("uiautomator dump /sdcard/qa_dump.xml") 把当前界面控件树导出成 XML,
    
14. *   再 adb.exec("cat ...") 读回来,正则抠所有 text="..." —— 一次拿到整页文字,
    
15. *   不用逐个控件找。若无障碍服务和 uiautomator 冲突导致 dump 失败,
    
16. *   自动降级用 auto.getXml()(AIWROK 自己的控件树,同样能抠 text)。
    
17. *
    
18. * 配置说明:
    
19. *   豆包安卓包名:com.larus.nova(应用宝确认)
    
20. *   DB_MODE="sqlite"(默认):设备本地生成 /sdcard/qa_articles.db,零配置直接能跑;
    
21. *   DB_MODE="mysql":DB_URL 要填"模拟器能访问到"的 MySQL 地址(127.0.0.1 是模拟器自己,
    
22. *   连不上宿主机),且 URL 必须带 connectTimeout/socketTimeout,否则连不上会无限卡死。
    
23. *   数据库不可用时不中止,降级为只采集进内存列表。
    
24. *
    
25. * 运行:雷电模拟器 + 豆包已登录 + 无障碍开启;不用 HID、不用截图
    
26. * (本脚本无截图对象,若日后加了 screenShot,记得最后 .recycle() 释放)。
    
27. */
    
28. 
    
29. // ==================== 配置(改这里)====================
    
30. var DB_MODE = "sqlite";              // "sqlite"=设备本地库文件(零配置,默认) | "mysql"=远程 MySQL
    
31. var PKG = "com.larus.nova";          // 豆包
    
32. var DB_URL = "jdbc:mysql://192.168.1.100:3306/aiwork?useSSL=false&characterEncoding=utf8&connectTimeout=5000&socketTimeout=10000";  // 改成你的 MySQL(必须带超时,否则连不上会无限卡死)
    
33. var DB_USER = "root";
    
34. var DB_PASS = "请填写数据库密码";
    
35. var SQLITE_PATH = "/sdcard/qa_articles.db";  // SQLite 模式:本地库文件路径
    
36. var DB_TABLE = "qa_articles";
    
37. var TOPICS = ["春天的景色", "人工智能改变生活", "健康饮食习惯"];  // 要生成文章的题目列表
    
38. var POLL_MS = 3000;          // 生成期间 dump 轮询间隔
    
39. var MIN_GEN_MS = 35000;      // 最短等待:豆包生成一篇 800 字文章实测要 35 秒左右,之前 12 秒就误判"稳定"
    
40. var MAX_WAIT_MS = 90000;     // 单篇最长等待生成时间
    
41. var STEP_WAIT = 3000;        // 每篇入库后的缓冲
    
42. 
    
43. // ==================== 全局变量放顶部 ====================
    
44. importClass(android.database.sqlite.SQLiteDatabase);
    
45. importClass(java.io.File);
    
46. var gStop = false;
    
47. var gConn = null;
    
48. var gDumpWarned = false;       // adb dump 不可用只提示一次
    
49. var gList = [];              // 采集到的问答列表 [{topic, question, answer, time}]
    
50. 
    
51. // ==================== 工具 ====================
    
52. function doSleep(ms) { sleep.millisecond(ms); }
    
53. function isStop() { return gStop; }
    
54. 
    
55. event.onStopBtnEvent(function () {
    
56.     gStop = true;
    
57.     printl("[停止] 收到停止请求,收尾后退出");
    
58. });
    
59. 
    
60. // ==================== 数据库:sqlite(本地文件) / mysql(JDBC) 双模式 ====================
    
61. function dbConnect() {
    
62.     if (DB_MODE == "mysql") {
    
63.         printl("[DB] 正在连接 MySQL(5 秒超时)...");
    
64.         try {
    
65.             gConn = mysql.getConnection(DB_URL, DB_USER, DB_PASS);
    
66.             printl("[DB] ✅ MySQL 连接成功");
    
67.             return true;
    
68.         } catch (e) {
    
69.             printl("[DB] ❌ MySQL 连接失败: " + e);
    
70.             return false;
    
71.         }
    
72.     }
    
73.     // sqlite:设备本地库文件,零配置
    
74.     try {
    
75.         gConn = SQLiteDatabase.openOrCreateDatabase(new File(SQLITE_PATH), null);
    
76.         printl("[DB] ✅ SQLite 已打开: " + SQLITE_PATH);
    
77.         return true;
    
78.     } catch (e2) {
    
79.         printl("[DB] ❌ SQLite 打开失败: " + e2);
    
80.         return false;
    
81.     }
    
82. }
    
83. 
    
84. function dbEnsureTable() {
    
85.     var sql;
    
86.     if (DB_MODE == "mysql") {
    
87.         sql = "CREATE TABLE IF NOT EXISTS " + DB_TABLE + " (" +
    
88.             "id INT AUTO_INCREMENT PRIMARY KEY," +
    
89.             "topic VARCHAR(255)," +
    
90.             "question TEXT," +
    
91.             "answer LONGTEXT," +
    
92.             "collected_at VARCHAR(32)" +
    
93.             ") DEFAULT CHARSET=utf8mb4";
    
94.     } else {
    
95.         sql = "CREATE TABLE IF NOT EXISTS " + DB_TABLE + " (" +
    
96.             "id INTEGER PRIMARY KEY AUTOINCREMENT," +
    
97.             "topic TEXT, question TEXT, answer TEXT, collected_at TEXT)";
    
98.     }
    
99.     try {
    
100.         gConn.execSQL(sql);
     
101.         printl("[DB] 表 " + DB_TABLE + " 就绪");
     
102.         return true;
     
103.     } catch (e) {
     
104.         printl("[DB] ❌ 建表失败: " + e);
     
105.         return false;
     
106.     }
     
107. }
     
108. 
     
109. function dbInsert(item) {
     
110.     try {
     
111.         var sql = "INSERT INTO " + DB_TABLE +
     
112.             " (topic, question, answer, collected_at) VALUES (?,?,?,?)";
     
113.         if (DB_MODE == "mysql") {
     
114.             var stmt = gConn.prepareStatement(sql);
     
115.             stmt.setString(1, item.topic);
     
116.             stmt.setString(2, item.question);
     
117.             stmt.setString(3, item.answer);
     
118.             stmt.setString(4, item.time);
     
119.             var rows = stmt.executeUpdate();
     
120.             stmt.close();
     
121.             printl("[DB] ✅ 入库 " + rows + " 条:《" + item.topic + "》");
     
122.         } else {
     
123.             var stmt2 = gConn.compileStatement(sql);
     
124.             stmt2.bindString(1, item.topic);
     
125.             stmt2.bindString(2, item.question);
     
126.             stmt2.bindString(3, item.answer);
     
127.             stmt2.bindString(4, item.time);
     
128.             stmt2.executeInsert();
     
129.             stmt2.close();
     
130.             printl("[DB] ✅ 入库成功:《" + item.topic + "》");
     
131.         }
     
132.         return true;
     
133.     } catch (e) {
     
134.         printl("[DB] ❌ 入库失败: " + e);
     
135.         return false;
     
136.     }
     
137. }
     
138. 
     
139. function dbClose() {
     
140.     try {
     
141.         if (gConn != null) { gConn.close(); gConn = null; printl("[DB] 已断开连接"); }
     
142.     } catch (e) {}
     
143. }
     
144. 
     
145. // ==================== ADB 采集:整页 dump → 抠文本 ====================
     
146. // 优先 adb uiautomator dump;失败降级 auto.getXml()
     
147. function dumpPageXml() {
     
148.     var xml = "";
     
149.     try {
     
150.         adb.exec("uiautomator dump /sdcard/qa_dump.xml", false);
     
151.         xml = adb.exec("cat /sdcard/qa_dump.xml", false);
     
152.     } catch (e) {}
     
153.     if (xml == null || xml.indexOf("<hierarchy") < 0) {
     
154.         if (!gDumpWarned) { gDumpWarned = true; printl("[采集] adb dump 不可用(与无障碍服务冲突),后续统一用 auto.getXml()"); }
     
155.         try { xml = auto.getXml(); } catch (e2) { xml = ""; }
     
156.     }
     
157.     return xml == null ? "" : xml;
     
158. }
     
159. 
     
160. // 抠 XML 里所有 text="..." 的非空文本(顺带还原常见转义符)
     
161. function extractTexts(xml) {
     
162.     var out = [];
     
163.     var re = /text="([^"]*)"/g;
     
164.     var m;
     
165.     while ((m = re.exec(xml)) != null) {
     
166.         var t = m[1];
     
167.         if (t != null && t.length > 1) {
     
168.             t = t.replace(/&/g, "&").replace(/</g, "<").replace(/>/g, ">")
     
169.                  .replace(/"/g, '"').replace(/
     
170. /g, "\n");
     
171.             out.push(t);
     
172.         }
     
173.     }
     
174.     return out;
     
175. }
     
176. 
     
177. // 页面文本指纹:用来判断生成有没有"写完"(文本不再变 = 完成)
     
178. function pageSignature(texts) {
     
179.     return texts.join("#");
     
180. }
     
181. 
     
182. // 发送前后的文本做差:新出现的长文本就是 AI 的回答
     
183. function pickAnswer(baseTexts, afterTexts) {
     
184.     var best = "";
     
185.     for (var i = 0; i < afterTexts.length; i++) {
     
186.         var t = afterTexts[i];
     
187.         if (t.length < 30) continue;          // 回答正文至少 30 字,短的是按钮/标题
     
188.         var isNew = true;
     
189.         for (var j = 0; j < baseTexts.length; j++) {
     
190.             if (baseTexts[j] == t) { isNew = false; break; }
     
191.         }
     
192.         if (isNew && t.length > best.length) best = t;
     
193.     }
     
194.     return best;
     
195. }
     
196. 
     
197. // ==================== 豆包操作:输入 → 发送 → 等生成稳定 ====================
     
198. function askDoubao(promptText) {
     
199.     // 1) 找输入框:先标准 EditText,再 editable 兜底
     
200.     var nodeObj = null;
     
201.     try {
     
202.         var ed = auto.nodeSearch(3000).className("android.widget.EditText");
     
203.         if (ed != '' && ed.count() > 0) nodeObj = ed.getNode(0);
     
204.     } catch (e) {}
     
205.     if (nodeObj == null) {
     
206.         try {
     
207.             var ed2 = auto.nodeSearch(2000).editable(true);
     
208.             if (ed2 != '' && ed2.count() > 0) nodeObj = ed2.getNode(0);
     
209.         } catch (e2) {}
     
210.     }
     
211.     if (nodeObj == null) { printl("[输入] &#10060; 没找到输入框"); return false; }
     
212.     nodeObj.click();
     
213.     sleep.second(1);
     
214.     nodeObj.input(promptText);   // 中文只能走 node.input,adb input text 不支持中文
     
215.     doSleep(800);
     
216. 
     
217.     // 2) 发送:文本"发送" → desc"发送" → 回车,三路兜底
     
218.     var sent = false;
     
219.     try {
     
220.         var sb = auto.nodeSearch(1500).text("发送");
     
221.         if (sb != '' && sb.count() > 0) { sb.getNode(0).click(); sent = true; }
     
222.     } catch (e) {}
     
223.     if (!sent) {
     
224.         try {
     
225.             var sb2 = auto.nodeSearch(1500).desc("发送");
     
226.             if (sb2 != '' && sb2.count() > 0) { sb2.getNode(0).click(); sent = true; }
     
227.         } catch (e2) {}
     
228.     }
     
229.     if (!sent) {
     
230.         try { input.enter(); sent = true; printl("[发送] 没找到发送按钮,用回车发送"); } catch (e3) {}
     
231.     }
     
232.     printl("[发送] " + (sent ? "&#9989; 已发送" : "&#10060; 发送失败"));
     
233.     return sent;
     
234. }
     
235. 
     
236. // 轮询 dump:先等满 MIN_GEN_MS,之后页面文本连续两次不变 → 判定生成完成
     
237. function waitForAnswer(baseSig) {
     
238.     var waited = 0;
     
239.     var lastSig = "";
     
240.     var stable = 0;
     
241.     while (waited < MAX_WAIT_MS && !isStop()) {
     
242.         sleep.second(POLL_MS / 1000);
     
243.         waited += POLL_MS;
     
244.         var sig = pageSignature(extractTexts(dumpPageXml()));
     
245.         if (waited < MIN_GEN_MS) {
     
246.             // 35 秒内豆包大概率还在生成,不做稳定判断,避免刚发出去就误判完成
     
247.             if (waited % 15000 == 0) printl("[生成] 最短等待中...已 " + (waited / 1000) + " 秒");
     
248.             lastSig = sig;
     
249.             continue;
     
250.         }
     
251.         if (sig != baseSig && sig.length > baseSig.length && sig == lastSig) {
     
252.             stable++;
     
253.             if (stable >= 2) {
     
254.                 printl("[生成] 文本已稳定,判定生成完成(用时 " + (waited / 1000) + " 秒)");
     
255.                 return true;
     
256.             }
     
257.         } else {
     
258.             stable = 0;
     
259.         }
     
260.         lastSig = sig;
     
261.         if (waited % 15000 == 0) printl("[生成] 已等待 " + (waited / 1000) + " 秒...");
     
262.     }
     
263.     printl("[生成] &#9200; 超时,按当前页面内容采集");
     
264.     return true;   // 超时也照采,有多少算多少
     
265. }
     
266. 
     
267. // ==================== 主流程 ====================
     
268. function main() {
     
269.     printl("╔════ 案例12:豆包文章生成 + ADB采集入库 ════╗");
     
270.     printl("[配置] 主题 " + TOPICS.length + " 个 | 表 " + DB_TABLE);
     
271. 
     
272.     // 数据库连不上不中止:降级为只采集进内存列表
     
273.     var dbOk = dbConnect();
     
274.     if (dbOk) dbOk = dbEnsureTable();
     
275.     if (!dbOk) printl("[DB] &#9888;&#65039; 数据库不可用,本次只采集进列表,不入库");
     
276.     printl("[DB] 数据存放位置: " + (DB_MODE == "mysql" ? DB_URL : SQLITE_PATH));
     
277. 
     
278.     // 重启豆包,保证从干净对话页开始
     
279.     app.closeApp(PKG);
     
280.     doSleep(1500);
     
281.     app.openApp(PKG);
     
282.     printl("[启动] 等 15 秒让豆包加载完");
     
283.     sleep.second(15);
     
284. 
     
285.     for (var i = 0; i < TOPICS.length; i++) {
     
286.         if (isStop()) break;
     
287.         var topic = TOPICS[i];
     
288.         var promptText = "请写一篇关于《" + topic + "》的文章,800字左右";
     
289.         printl("──── 第 " + (i + 1) + "/" + TOPICS.length + " 篇:" + topic + " ────");
     
290. 
     
291.         // 发送前先记基线,回答 = 之后新出现的长文本
     
292.         var baseTexts = extractTexts(dumpPageXml());
     
293.         var baseSig = pageSignature(baseTexts);
     
294. 
     
295.         if (!askDoubao(promptText)) continue;
     
296.         waitForAnswer(baseSig);
     
297.         if (isStop()) break;
     
298. 
     
299.         var afterTexts = extractTexts(dumpPageXml());
     
300.         var answer = pickAnswer(baseTexts, afterTexts);
     
301.         if (answer == "") {
     
302.             printl("[采集] &#10060; 没抓到正文(可能生成失败/被折叠),跳过");
     
303.             continue;
     
304.         }
     
305. 
     
306.         var item = { topic: topic, question: promptText, answer: answer, time: time.now() };
     
307.         gList.push(item);                       // 内存列表
     
308.         if (dbOk) dbInsert(item);               // 数据库
     
309.         printl("[采集] 正文 " + answer.length + " 字 | 已存: " + (DB_MODE == "mysql" ? DB_URL : SQLITE_PATH));
     
310.         printl("──── 《" + topic + "》全文开始 ────");
     
311.         printl(answer);
     
312.         printl("──── 《" + topic + "》全文结束 ────");
     
313.         sleep.second(STEP_WAIT / 1000);
     
314.     }
     
315. 
     
316.     // 列表汇总
     
317.     printl("╔════ 采集列表(共 " + gList.length + " 条)════╗");
     
318.     for (var k = 0; k < gList.length; k++) {
     
319.         printl("[" + (k + 1) + "] 《" + gList[k].topic + "》 | " + gList[k].time +
     
320.             " | " + gList[k].answer.length + " 字");
     
321.     }
     
322.     printl("[DB] 全部数据存放位置: " + (DB_MODE == "mysql" ? DB_URL : SQLITE_PATH));
     
323.     if (dbOk) dbClose();
     
324.     printl("[结束] 豆包采集流程执行完毕");
     
325. }
     
326. 
     
327. main();
     

复制代码