Samples work in ui now
This commit is contained in:
50
ui/src/server/settings.ts
Normal file
50
ui/src/server/settings.ts
Normal file
@@ -0,0 +1,50 @@
|
||||
import { PrismaClient } from '@prisma/client';
|
||||
import { defaultDatasetsFolder } from '@/paths';
|
||||
import { defaultTrainFolder } from '@/paths';
|
||||
import NodeCache from 'node-cache';
|
||||
|
||||
const myCache = new NodeCache();
|
||||
const prisma = new PrismaClient();
|
||||
|
||||
export const flushCache = () => {
|
||||
myCache.flushAll();
|
||||
};
|
||||
|
||||
export const getDatasetsRoot = async () => {
|
||||
const key = 'DATASETS_FOLDER';
|
||||
let datasetsPath = myCache.get(key) as string;
|
||||
if (datasetsPath) {
|
||||
return datasetsPath;
|
||||
}
|
||||
let row = await prisma.settings.findFirst({
|
||||
where: {
|
||||
key: 'DATASETS_FOLDER',
|
||||
},
|
||||
});
|
||||
datasetsPath = defaultDatasetsFolder;
|
||||
if (row?.value && row.value !== '') {
|
||||
datasetsPath = row.value;
|
||||
}
|
||||
myCache.set(key, datasetsPath);
|
||||
return datasetsPath as string;
|
||||
};
|
||||
|
||||
|
||||
export const getTrainingFolder = async () => {
|
||||
const key = 'TRAINING_FOLDER';
|
||||
let trainingRoot = myCache.get(key) as string;
|
||||
if (trainingRoot) {
|
||||
return trainingRoot;
|
||||
}
|
||||
let row = await prisma.settings.findFirst({
|
||||
where: {
|
||||
key: key,
|
||||
},
|
||||
});
|
||||
trainingRoot = defaultTrainFolder;
|
||||
if (row?.value && row.value !== '') {
|
||||
trainingRoot = row.value;
|
||||
}
|
||||
myCache.set(key, trainingRoot);
|
||||
return trainingRoot as string;
|
||||
};
|
||||
Reference in New Issue
Block a user