245 lines
8.1 KiB
JavaScript
245 lines
8.1 KiB
JavaScript
|
import { Tools } from "../../main/tools";
|
|||
|
|
import { define } from "../define";
|
|||
|
|
import path from "path";
|
|||
|
|
import { DEFINE_STRING } from "../define_string";
|
|||
|
|
import { get, has } from "lodash";
|
|||
|
|
let tools = new Tools();
|
|||
|
|
const { v4: uuidv4 } = require('uuid'); // 引入UUID库来生成唯一标识符
|
|||
|
|
|
|||
|
|
export const ImageSetting = {
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取自动保存图片的方式
|
|||
|
|
* @returns
|
|||
|
|
*/
|
|||
|
|
async GetAutoSaveImageClassifyOptions() {
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
data: [{
|
|||
|
|
label: "不分类",
|
|||
|
|
value: "one"
|
|||
|
|
},
|
|||
|
|
// {
|
|||
|
|
// label: "按模型",
|
|||
|
|
// value: "model"
|
|||
|
|
// },
|
|||
|
|
{
|
|||
|
|
label: "按风格",
|
|||
|
|
value: "style"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
label: "风格+模型",
|
|||
|
|
value: "theme"
|
|||
|
|
}]
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取自动保存图片的方式(sd，mj，d3 等)
|
|||
|
|
*/
|
|||
|
|
async GetImageGenerateCategory() {
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
data: [{
|
|||
|
|
label: "SD",
|
|||
|
|
value: "sd"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
label: "MJ",
|
|||
|
|
value: "mj"
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
label: "D3",
|
|||
|
|
value: "d3"
|
|||
|
|
}]
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 保存自动保存图片的设置
|
|||
|
|
*/
|
|||
|
|
async SaveImageAutoSaveSetting(value) {
|
|||
|
|
try {
|
|||
|
|
value = JSON.parse(value);
|
|||
|
|
await tools.writeJsonFilePropertyValue(define.img_base, "auto_save_image", value);
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
return {
|
|||
|
|
code: 0,
|
|||
|
|
message: error.toString()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 保存自动保存图片的设置
|
|||
|
|
*/
|
|||
|
|
async GetImageAutoSaveSetting() {
|
|||
|
|
try {
|
|||
|
|
let res = await tools.getJsonFilePropertyValue(define.img_base, "auto_save_image", {}, false);
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
data: res
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
return {
|
|||
|
|
code: 0,
|
|||
|
|
message: error.toString()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 保存图片到指定文件夹，只是将任务添加到队列中
|
|||
|
|
*/
|
|||
|
|
async SaveImageToOtherFolder(output = [], value) {
|
|||
|
|
try {
|
|||
|
|
let show = false;
|
|||
|
|
if (value) {
|
|||
|
|
value = JSON.parse(value);
|
|||
|
|
show = false;
|
|||
|
|
} else {
|
|||
|
|
let auto_save_image = await tools.getJsonFilePropertyValue(define.img_base, "auto_save_image", {}, false);
|
|||
|
|
value = {
|
|||
|
|
save_match_count: auto_save_image.save_match_count,
|
|||
|
|
save_folder: auto_save_image.main_save_folder
|
|||
|
|
}
|
|||
|
|
if (!auto_save_image.auto_save) {
|
|||
|
|
return {
|
|||
|
|
code: 1
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let batch = DEFINE_STRING.QUEUE_BATCH.IMAGE_SAVE_TO_OTHER_FOLDER;
|
|||
|
|
// 获取当前的所有的output文件夹
|
|||
|
|
if (output.length == 0) {
|
|||
|
|
output = await tools.getSubFolderList(path.join(global.config.project_path, "tmp/bak/"), "start", "output_crop_");
|
|||
|
|
}
|
|||
|
|
for (let i = 0; i < output.length; i++) {
|
|||
|
|
const element = path.join(global.config.project_path, 'tmp/bak/' + output[i]);
|
|||
|
|
// 获取指定的文件夹里面的文件
|
|||
|
|
let png_files = await tools.getFilesWithExtensions(element, ".png");
|
|||
|
|
console.log(png_files);
|
|||
|
|
// 判断当前的png_files中的数据是不是大于value.save_match_count，大于的话，删除数组里面的数据
|
|||
|
|
if (value.save_match_count && png_files.length > value.save_match_count) {
|
|||
|
|
// 删除数组
|
|||
|
|
png_files.splice(value.save_match_count);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for (let j = 0; j < png_files.length; j++) {
|
|||
|
|
const item = png_files[j];
|
|||
|
|
global.fileQueue.enqueue(async () => {
|
|||
|
|
// 复制文件到指定的文件夹
|
|||
|
|
let dst = path.join(value.save_folder, `${Date.now().toString()}_${uuidv4().split('-')[0]}.png`);
|
|||
|
|
|
|||
|
|
await tools.copyFileOrDirectory(item, dst);
|
|||
|
|
}, `${batch}_${item}`, batch)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
global.fileQueue.setBatchCompletionCallback(batch, (failedTasks) => {
|
|||
|
|
if (failedTasks.length > 0) {
|
|||
|
|
let message = `
|
|||
|
|
转存任务都已完成。
|
|||
|
|
但是以下任务执行失败：
|
|||
|
|
`
|
|||
|
|
failedTasks.forEach(({ taskId, error }) => {
|
|||
|
|
message += `${taskId}-, \n 错误信息: ${error}` + '\n';
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
|||
|
|
code: 0,
|
|||
|
|
message: message
|
|||
|
|
})
|
|||
|
|
} else {
|
|||
|
|
if (show) {
|
|||
|
|
global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
|
|||
|
|
code: 1,
|
|||
|
|
message: "所有图片转存完成"
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
code: 1
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} catch (error) {
|
|||
|
|
return {
|
|||
|
|
code: 0,
|
|||
|
|
message: error.toString()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取指定的配置文件里面指定的属性的数据
|
|||
|
|
* @param {*} value 执行方法必要的信息
|
|||
|
|
* 0 define中的指定属性（指定的配置文件）
|
|||
|
|
* 1 要获取的什么属性信息 property，property 为null，赶回当前配置文件的所有数据
|
|||
|
|
* 2 是不是要校验属性不存在
|
|||
|
|
* 3 属性没有找到的默认值
|
|||
|
|
* @returns
|
|||
|
|
*/
|
|||
|
|
async GetDefineConfigJsonByProperty(value) {
|
|||
|
|
try {
|
|||
|
|
value = JSON.parse(value);
|
|||
|
|
let defin_property = value[0];
|
|||
|
|
let property = value[1];
|
|||
|
|
let need_check = value[2];
|
|||
|
|
let default_data = value[3];
|
|||
|
|
// 判断define中的属性是不是存在
|
|||
|
|
let hasProperty = has(define, defin_property);
|
|||
|
|
if (!hasProperty) {
|
|||
|
|
throw new Error(`${defin_property} 属性不存在。`);
|
|||
|
|
}
|
|||
|
|
let data = await tools.getJsonFilePropertyValue(define[defin_property], property, default_data, need_check);
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
data: data
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
return {
|
|||
|
|
code: 0,
|
|||
|
|
message: "获取配置信息失败，错误信息入下 ： " + error.message.toString()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 保存指定的配置文件里面指定的属性的数据
|
|||
|
|
* @param {*} value
|
|||
|
|
* 0 define中的指定属性（指定的配置文件）
|
|||
|
|
* 1 要获取的什么属性信息 property，property 为null，赶回当前配置文件的所有数据
|
|||
|
|
* 2 是不是要校验属性不存在
|
|||
|
|
* 3 属性没有找到的默认值
|
|||
|
|
*/
|
|||
|
|
async SaveDefineConfigJsonByProperty(value) {
|
|||
|
|
try {
|
|||
|
|
value = JSON.parse(value);
|
|||
|
|
let defin_property = value[0];
|
|||
|
|
let property = value[1];
|
|||
|
|
let data = value[2];
|
|||
|
|
let need_check = value[3] ? value[3] : false;
|
|||
|
|
// 判断define中的属性是不是存在
|
|||
|
|
let hasProperty = has(define, defin_property);
|
|||
|
|
if (!hasProperty) {
|
|||
|
|
throw new Error(`${defin_property} 属性不存在。`);
|
|||
|
|
}
|
|||
|
|
await tools.writeJsonFilePropertyValue(define[defin_property], property, data, need_check);
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
message: "保存指定的配置成功"
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
return {
|
|||
|
|
code: 0,
|
|||
|
|
message: "保存配置信息失败，错误信息入下 ： " + error.message.toString()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|