diff --git a/app/src/protyle/export/index.ts b/app/src/protyle/export/index.ts
index db298c915..2b76a1ff9 100644
--- a/app/src/protyle/export/index.ts
+++ b/app/src/protyle/export/index.ts
@@ -401,18 +401,9 @@ const renderPDF = async (id: string) => {
previewElement.innerHTML = '
' + data.content + '
';
const wysElement = previewElement.querySelector(".protyle-wysiwyg");
wysElement.setAttribute("data-doc-type", data.type || "NodeDocument");
- if (data.attrs.memo) {
- wysElement.setAttribute("memo", data.attrs.memo);
- }
- if (data.attrs.name) {
- wysElement.setAttribute("name", data.attrs.name);
- }
- if (data.attrs.bookmark) {
- wysElement.setAttribute("bookmark", data.attrs.bookmark);
- }
- if (data.attrs.alias) {
- wysElement.setAttribute("alias", data.attrs.alias);
- }
+ Object.keys(data.attrs).forEach(key => {
+ wysElement.setAttribute(key, data.attrs[key]);
+ })
// https://github.com/siyuan-note/siyuan/issues/13669
wysElement.querySelectorAll('[data-node-id]').forEach((item) => {
if (item.querySelector(".img")) {
diff --git a/app/src/protyle/export/util.ts b/app/src/protyle/export/util.ts
index 70e00f496..8c06faf82 100644
--- a/app/src/protyle/export/util.ts
+++ b/app/src/protyle/export/util.ts
@@ -155,18 +155,9 @@ export const exportImage = (id: string) => {
const refreshPreview = (response: IWebSocketData) => {
previewElement.innerHTML = response.data.content;
previewElement.setAttribute("data-doc-type", response.data.type || "NodeDocument");
- if (response.data.attrs.memo) {
- previewElement.setAttribute("memo", response.data.attrs.memo);
- }
- if (response.data.attrs.name) {
- previewElement.setAttribute("name", response.data.attrs.name);
- }
- if (response.data.attrs.bookmark) {
- previewElement.setAttribute("bookmark", response.data.attrs.bookmark);
- }
- if (response.data.attrs.alias) {
- previewElement.setAttribute("alias", response.data.attrs.alias);
- }
+ Object.keys(response.data.attrs).forEach(key => {
+ previewElement.setAttribute(key, response.data.attrs[key]);
+ });
previewElement.querySelectorAll(".code-block").forEach(item => {
item.setAttribute("linewrap", "true");
});