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() { 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 { 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'; 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 = { export type DeleteOptions = {
exceptions?: string[], contentTypes?: ContentTypeSchema[];
conditions?: { conditions?: {
[contentTypeUid: string]: { [contentTypeUid: string]: {
where: any where: any;
} };
} };
} };
export const deleteAllRecords = async (strapi: Strapi.Strapi, deleteOptions?: DeleteOptions) => { export const deleteAllRecords = async (strapi: Strapi.Strapi, deleteOptions?: DeleteOptions) => {
const conditions = deleteOptions?.conditions ?? {}; const conditions = deleteOptions?.conditions ?? {};
const exceptions = deleteOptions?.exceptions ?? defaultExceptions; const exceptions = [
const contentTypes: ContentTypeSchema[] = Object.values(strapi.contentTypes); 'admin::permission',
let count = 0; 'admin::user',
await Promise.all(contentTypes.map(async (contentType) => { 'admin::role',
const filters = conditions[contentType.uid] ?? {} 'admin::api-token',
if (!exceptions.includes(contentType.uid)) { 'admin::api-token-permission',
const result = await strapi?.db.query(contentType.uid).deleteMany({ ];
filters const allContentTypes: ContentTypeSchema[] = Object.values(strapi.contentTypes);
}); const defaultContentTypes: ContentTypeSchema[] = allContentTypes.filter(
count += result.count; (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 };
} };