panleicim/puppeteerjs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

delete images if OCRResult is identified

Browse Source
This commit is contained in:
Lei PAN
2023-09-01 22:24:49 +02:00
parent 06bf2ead13
commit 2a7c43c2af
2 changed files with 27 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/workers/OCRChecker.js
+24 -2
View File
@@ -2,6 +2,7 @@ const {v4: uuidv4} = require("uuid");
const tesseract = require("node-tesseract-ocr");
const OCRResult = require("../models/OCRResult");
const Jimp = require('jimp');
const fs = require("fs");
function delay(delayInMs) {
return new Promise(resolve => {
@@ -89,6 +90,7 @@ class OCRChecker {
async get_result() {
let fileName = await this.take_screen_shot()
let screenShot = await convertImageToWhiteBlack(fileName);
await delay(1000)
try {
let result = await tesseract
.recognize(screenShot, config)
@@ -100,6 +102,8 @@ class OCRChecker {
} else if (result.includes(CAPTCHA_ERROR_MESSAGE) || result.includes(CAPTCHA_ERROR_MESSAGE_FR)) {
return OCRResult.RECAPTCHA_ERROR
} else if (result.includes(BLOCKED_MSG_EN) || result.includes(BLOCKED_MSG_FR) || result.includes(BLOCKED_MSG_FR_2)) {
await this.deleteFile(fileName)
await this.deleteFile(screenShot)
return OCRResult.BLOCKED
} else if (result.includes(ERR_CACHE_MISS) || result.includes(ERR_CACHE_MISS_2) || result.includes(ERR_CACHE_MISS_3) || result.includes(ERR_CACHE_MISS_4)) {
return OCRResult.TO_REFRESH
@@ -119,6 +123,8 @@ class OCRChecker {
|| result.includes(PAGE_OPTIMIZATION_CHROME_FR_8)
|| result.includes(PAGE_OPTIMIZATION_CHROME_FR_7)
) {
await this.deleteFile(fileName)
await this.deleteFile(screenShot)
return OCRResult.PAGE_OPTIMIZATION
} else if (result.includes(CONFIRM_RESEND_FORM_FR)) {
return OCRResult.CONFIRM_RESEND_FORM
@@ -127,16 +133,24 @@ class OCRChecker {
} else if (result.includes(ABOUT_BLANK)) {
return OCRResult.TERMINAED
} else if (result.includes(DIALOG_TO_SKIP)) {
await this.deleteFile(fileName)
await this.deleteFile(screenShot)
return OCRResult.TO_SKIP
} else if (result.includes(RECAPTCHA_FAILED_FR)) {
return OCRResult.TERMINAED
} else if (result.includes(NO_INTERNET_FR)) {
return OCRResult.NO_INTERNET
} else if (result.includes(BRAVE_SKIP_DEFAULT_PAGE) || result.includes(BRAVE_SKIP_DEFAULT_PAGE_2)) {
await this.deleteFile(fileName)
await this.deleteFile(screenShot)
return OCRResult.BRAVE_SKIP
} else if (result.includes(BRAVE_SKIP_PRIVACY_PAGE) || result.includes(BRAVE_SKIP_PRIVACY_PAGE_2)) {
await this.deleteFile(fileName)
await this.deleteFile(screenShot)
return OCRResult.BRAVE_PRIVACY
} else if (result.includes(BRAVE_NOTIFICATION)) {
await this.deleteFile(fileName)
await this.deleteFile(screenShot)
return OCRResult.BRAVE_NOTIFICATION
} else if (result.includes(BRAVE_SKIP_PUB) || result.includes(BRAVE_SKIP_PUB_2) || result.includes(BRAVE_SKIP_PUB_3) || result.includes(BRAVE_SKIP_PUB_4) || result.includes(BRAVE_SKIP_PUB_5)) {
return OCRResult.BRAVE_PRIVACY_PUB
@@ -149,12 +163,20 @@ class OCRChecker {
}
}
async deleteFile(filename) {
try {
fs.unlinkSync(filename)
} catch (e) {
console.log(e)
return OCRResult.TERMINAED
}
}
async take_screen_shot() {
let name = this.get_file_name()
await this.device.screenshot({path: name});
await delay(1000);
await delay(3000);
return name
// console.log(`stdout: ${stdout1}`);
}
}