yujunjun/strapi
1
0
Fork 0
mirror of https://github.com/strapi/strapi.git synced 2025-09-22 14:59:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make deleteAllRecords accept contentTypes to delete rather than exceptions

Browse Source
This commit is contained in:
Bassel 2022-11-21 14:52:57 +02:00
parent 8a0db019bc
commit bba2c8a994
2 changed files with 35 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
packages/core/data-transfer/lib/providers/local-file-source-provider.ts
View File

@ -87,9 +87,10 @@ class LocalFileSourceProvider implements ISourceProvider {
}
async getSchemas() {
const schemas = await collect(this.streamSchemas() as Readable);
// const schemas = await collect(this.streamSchemas() as Readable);
return keyBy('uid', schemas);
// return keyBy('uid', schemas);
return null;
}
streamEntities(): NodeJS.ReadableStream {

57
packages/core/data-transfer/lib/providers/local-strapi-destination-provider/restore.ts
View File

@ -1,31 +1,38 @@
import type { ContentTypeSchema } from '@strapi/strapi';
const defaultExceptions = ['admin::permission', 'admin::user', 'admin::role', 'admin::api-token', 'admin::api-token-permission', 'plugin::i18n.locale']
export type DeleteOptions = {
exceptions?: string[],
conditions?: {
[contentTypeUid: string]: {
where: any
}
}
}
contentTypes?: ContentTypeSchema[];
conditions?: {
[contentTypeUid: string]: {
where: any;
};
};
};
export const deleteAllRecords = async (strapi: Strapi.Strapi, deleteOptions?: DeleteOptions) => {
const conditions = deleteOptions?.conditions ?? {};
const exceptions = deleteOptions?.exceptions ?? defaultExceptions;
const contentTypes: ContentTypeSchema[] = Object.values(strapi.contentTypes);
let count = 0;
await Promise.all(contentTypes.map(async (contentType) => {
const filters = conditions[contentType.uid] ?? {}
if (!exceptions.includes(contentType.uid)) {
const result = await strapi?.db.query(contentType.uid).deleteMany({
filters
});
count += result.count;
}
}))
const conditions = deleteOptions?.conditions ?? {};
const exceptions = [
'admin::permission',
'admin::user',
'admin::role',
'admin::api-token',
'admin::api-token-permission',
];
const allContentTypes: ContentTypeSchema[] = Object.values(strapi.contentTypes);
const defaultContentTypes: ContentTypeSchema[] = allContentTypes.filter(
(contentType) => !exceptions.includes(contentType.uid)
);
const contentTypes: ContentTypeSchema[] = deleteOptions?.contentTypes ?? defaultContentTypes;
let count = 0;
await Promise.all(
contentTypes.map(async (contentType) => {
const filters = conditions[contentType.uid] ?? {};
const result = await strapi?.db.query(contentType.uid).deleteMany({
filters,
});
count += result.count;
})
);
return { count }
}
return { count };
};