preLoadFile.js 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195
  1. function getDaysAgo(n) {
  2. const timestamp = Date.now() - (n * 24 * 60 * 60 * 1000);
  3. const date = new Date(timestamp);
  4. const year = date.getFullYear();
  5. const month = String(date.getMonth() + 1).padStart(2, '0');
  6. const day = String(date.getDate()).padStart(2, '0');
  7. return `${year}-${month}-${day}`;
  8. }
  9. function csvToArray(csvString) {
  10. const lines = csvString.split('\n');
  11. const result = [];
  12. for (let line of lines) {
  13. // 跳过空行
  14. if (line.trim() === '') continue;
  15. // 按逗号分割,并去除每个字段的空白
  16. const row = line.split(',').map(field => field.trim());
  17. result.push(row);
  18. }
  19. return result;
  20. }
  21. function arrayToCSV(array) {
  22. return array.map(row =>
  23. row.map(field => String(field)).join(',')
  24. ).join('\n');
  25. }
  26. function creatToday(fileList) {
  27. for (let index = 1; index < fileList.length - 1; index++) {
  28. fileList[index][2] = fileList[index][5]
  29. fileList[index][3] = fileList[index][6]
  30. fileList[index][4] = fileList[index][7]
  31. fileList[index][5] = ""
  32. fileList[index][6] = ""
  33. fileList[index][7] = ""
  34. fileList[index][8] = ""
  35. fileList[index][9] = ""
  36. fileList[index][10] = ""
  37. }
  38. const fileData = {
  39. filename: getDaysAgo(0) + '.csv',
  40. content: arrayToCSV(fileList)
  41. };
  42. // 调用保存方法
  43. window.electronAPI.saveFile(fileData).then(result => {
  44. if (result.success) {
  45. console.log('文件保存成功:', result.path);
  46. } else {
  47. console.error('文件保存失败:', result.error);
  48. }
  49. });
  50. }
  51. function saveFile(adtaArr, next) {
  52. // 保存数据
  53. const fileData = {
  54. filename: getDaysAgo(0) + '.csv',
  55. content: arrayToCSV(adtaArr)
  56. };
  57. // 调用保存方法
  58. window.electronAPI.saveFile(fileData).then(result => {
  59. if (result.success) {
  60. console.log('文件保存成功:', result.path);
  61. setTimeout(() => {
  62. if (next) checkData(adtaArr, next)
  63. }, 1000);
  64. } else {
  65. console.error('文件保存失败:', result.error);
  66. }
  67. });
  68. }
  69. function cutString (original, before, after, index) {
  70. index = index || 0;
  71. if (typeof index === "number") {
  72. const P = original.indexOf(before, index);
  73. if (P > -1) {
  74. if (after) {
  75. const f = original.indexOf(after, P + before.length);
  76. return (f>-1)? original.slice(P + before.toString().length, f):console.error("Tool [在文本中找不到 参数三 "+after+"]");
  77. } else {
  78. return original.slice(P + before.toString().length);
  79. }
  80. } else {
  81. console.error("Tool [在文本中找不到 参数一 " + before + "]");
  82. return
  83. }
  84. } else {
  85. console.error("Tool [sizeTransition:" + index + "不是一个整数!]");
  86. }
  87. }
  88. function checkData (adtaArr) {
  89. for (let index = 0; index < adtaArr.length; index++) {
  90. const element = adtaArr[index];
  91. console.log(element)
  92. if (element[1].includes('https://') && element[5] === '') {
  93. console.log(element[1])
  94. if (location.href.slice(0, 40) == element[1].slice(0, 40)) {
  95. setTimeout(() => {
  96. document.querySelector('.yt-truncated-text__absolute-button').click()
  97. setTimeout(() => {
  98. let pdInfo = document.querySelector('.style-scope.ytd-about-channel-renderer').innerText
  99. let bfsl = parseInt(cutString(pdInfo,'视频\n\n\t', '次观看'))
  100. let spsl = parseInt(cutString(pdInfo,'订阅者\n\n\t', '个视频'))
  101. let dyrs = parseInt(cutString(pdInfo,'注册\n\n\t', '位订阅者'))
  102. bfsl = bfsl ? bfsl : 0
  103. spsl = spsl ? spsl : 0
  104. dyrs = dyrs ? dyrs : 0
  105. console.log(bfsl, spsl, dyrs)
  106. adtaArr[index][5] = bfsl
  107. adtaArr[index][6] = spsl
  108. adtaArr[index][7] = dyrs
  109. adtaArr[index][8] = bfsl - parseInt(adtaArr[index][2])
  110. adtaArr[index][9] = spsl - parseInt(adtaArr[index][3])
  111. adtaArr[index][10] = dyrs - parseInt(adtaArr[index][4])
  112. saveFile(adtaArr, true)
  113. }, 3000);
  114. }, 2000);
  115. } else {
  116. location.href = element[1]
  117. }
  118. return
  119. }
  120. }
  121. // 汇总计算
  122. adtaArr[adtaArr.length - 1][2] = 0
  123. adtaArr[adtaArr.length - 1][3] = 0
  124. adtaArr[adtaArr.length - 1][4] = 0
  125. adtaArr[adtaArr.length - 1][5] = 0
  126. adtaArr[adtaArr.length - 1][6] = 0
  127. adtaArr[adtaArr.length - 1][7] = 0
  128. adtaArr[adtaArr.length - 1][8] = 0
  129. adtaArr[adtaArr.length - 1][9] = 0
  130. adtaArr[adtaArr.length - 1][10] = 0
  131. for (let index = 1; index < adtaArr.length - 1; index++) {
  132. const element = adtaArr[index];
  133. adtaArr[adtaArr.length - 1][2] += parseInt(element[2])
  134. adtaArr[adtaArr.length - 1][3] += parseInt(element[3])
  135. adtaArr[adtaArr.length - 1][4] += parseInt(element[4])
  136. adtaArr[adtaArr.length - 1][5] += parseInt(element[5])
  137. adtaArr[adtaArr.length - 1][6] += parseInt(element[6])
  138. adtaArr[adtaArr.length - 1][7] += parseInt(element[7])
  139. adtaArr[adtaArr.length - 1][8] += parseInt(element[8])
  140. adtaArr[adtaArr.length - 1][9] += parseInt(element[9])
  141. adtaArr[adtaArr.length - 1][10] += parseInt(element[10])
  142. }
  143. saveFile(adtaArr, false)
  144. alert('所有项目已经检查完!')
  145. }
  146. // 调用保存方法
  147. window.electronAPI.readFile({filename: getDaysAgo(0) + '.csv', encoding: "utf8"}).then(result => {
  148. if (result.success) {
  149. console.log(result.content);
  150. // 如果内容是空的,则读取昨天的
  151. if (result.content == '') {
  152. window.electronAPI.readFile({filename: getDaysAgo(1) + '.csv', encoding: "utf8"}).then(result2 => {
  153. if (result2.success) {
  154. console.log(result2.content);
  155. // 如果内容是空的,则读取昨天的
  156. if (result2.content == '') {
  157. alert('缺少昨天的数据!')
  158. } else {
  159. let fileData = csvToArray(result2.content)
  160. creatToday(fileData)
  161. console.log(fileData)
  162. checkData(fileData)
  163. }
  164. } else {
  165. console.error('文件读取失败:', result2.error);
  166. }
  167. });
  168. } else {
  169. // 读取加载文件
  170. let fileData = csvToArray(result.content)
  171. console.log(fileData)
  172. checkData(fileData)
  173. }
  174. } else {
  175. console.error('文件读取失败:', result.error);
  176. }
  177. });