From 9a9b4646ca843d0c3aedd0bcf9874b2257094e63 Mon Sep 17 00:00:00 2001
From: wenpeng <wenpeng@meiguanjia.net>
Date: Thu, 1 Aug 2024 16:07:17 +0800
Subject: [PATCH] fix

---
 example2.js               | 15 ++--------
 utils/businessDataUtil.js | 58 +++++++++++++++++++++++++--------------
 2 files changed, 40 insertions(+), 33 deletions(-)

diff --git a/example2.js b/example2.js
index 4c3394d..85a3a30 100644
--- a/example2.js
+++ b/example2.js
@@ -18,24 +18,13 @@ import { getDate, cleanType, changeTimeDimension } from './utils/businessDataUti
     const str = await page.evaluate(node => node.textContent.trim(), childElements[1]);
     console.log(str);
 
-    await changeTimeDimension(page, '周报');
+    // await changeTimeDimension(page, '周报');
 
     const file = './data/businessData.json';
 
     fs.writeFileSync(file, '', 'utf-8');
 
-    await getDate(page, fs, file);
-    console.log('========');
-
-    await cleanType(page, '.arco-picker-start-time', '2024-06');
-    await getDate(page, fs, file);
-    console.log('========');
-
-    await cleanType(page, '.arco-picker-start-time', '2024-05');
-    await getDate(page, fs, file);
-    console.log('========');
-
-    await cleanType(page, '.arco-picker-start-time', '2024-04');
+    await cleanType(page, '.arco-picker-start-time', '2024-07');
     await getDate(page, fs, file);
     console.log('========');
 
diff --git a/utils/businessDataUtil.js b/utils/businessDataUtil.js
index c5b054c..bb4ab59 100644
--- a/utils/businessDataUtil.js
+++ b/utils/businessDataUtil.js
@@ -1,31 +1,41 @@
 export async function getDate(page, fs, file) {
+    const its = new Map();
+
     await new Promise(r => setTimeout(r, 2000));
     await page.waitForSelector('.ai_custome');
-    const aiCustome = await page.$$('.ai_custome .ai_custome_item .amount');
-    let ai = new Set();
-    for (const e of aiCustome) {
-        let str = await page.evaluate(element => element.textContent, e);
-        ai.add(str);
-    }
+    const aiCustome = await page.$$('.ai_custome .ai_custome_item');
 
-    let importStorage = '\n' + JSON.stringify(Array.from(ai), null, 2) + '\n';
-    fs.appendFileSync(file, importStorage, 'utf-8');
+    if (aiCustome.length > 0) {
+        for (const iterator of aiCustome) {
+            const key = await iterator.$('span');
+            const value = await iterator.$('div');
+            const keyStr = await page.evaluate(element => element.textContent.trim(), key);
+            const valueStr = await page.evaluate(element => element.textContent.trim(), value);
+            its.set(keyStr, valueStr);
+        }
+    } else {
+        console.log('.ai_custome_item not found');
+    }
 
     const elementHandle = await page.waitForSelector('.bi_warp iframe');
     const frame = await elementHandle.contentFrame();
     await frame.waitForSelector('#label-content');
-    const items = await frame.$$('#label-content'); // Use $$ to get all matching elements
-    let its = new Set();
-    if (items.length > 0) { // Check if items array is not empty
+    const items = await frame.$$('#label-content');
+
+    if (items.length > 0) {
         for (const e of items) {
-            let str = await frame.evaluate(element => element.textContent, e); // Use frame.evaluate() here
-            its.add(str.replace(/\s+/g, ' '));
+            const key = await e.$('span p');
+            const value = await e.$('span span');
+            const keyStr = await frame.evaluate(element => element.textContent.trim(), key);
+            const valueStr = await frame.evaluate(element => element.textContent.trim(), value);
+
+            its.set(keyStr, valueStr);
         }
     } else {
         console.log('#label-content not found in the iframe');
     }
 
-    importStorage = JSON.stringify(Array.from(its), null, 2) + '\n';
+    const importStorage = JSON.stringify(Array.from(its), null, 2);
     fs.appendFileSync(file, importStorage, 'utf-8');
 }
 
@@ -35,21 +45,29 @@ export async function cleanType(page, element, timeContent) {
     await time.type(timeContent, { delay: 100 });
 }
 
-export async function changeTimeDimension(page, timeDimension) {
+/**
+ * 
+ * @param {Page} page 
+ * @param {Number} dismensionNumber 
+ */
+export async function changeTimeDimension(page, dismensionNumber) {
+    const timeDimension = [
+        '日报',
+        '周报',
+        '月报',
+        '自定义'
+    ];
+
     let rb = await page.$('.arco-space-item .arco-select-view-value');
     let s = await page.evaluate(node => node.textContent.trim(), rb);
-    console.log(s);
 
     await rb.click();
     const timeMap = new Map();
     rb = await page.$$('.arco-select-dropdown-list .arco-select-option');
     for (const iterator of rb) {
         s = await page.evaluate(node => node.textContent.trim(), iterator)
-        console.log(s);
         timeMap.set(s, iterator);
     }
-    console.log('===');
-    console.log(timeMap);
 
-    await timeMap.get(timeDimension).click();
+    await timeMap.get(timeDimension[dismensionNumber]).click();
 }
\ No newline at end of file