Add base tag to reduce the output size (#103)

This commit is contained in:
KnugiHK
2024-06-30 19:16:58 +08:00
parent 5449646a1b
commit b0942d695b
4 changed files with 9 additions and 3 deletions

View File

@@ -34,6 +34,7 @@ class ChatStore():
self.their_avatar = None self.their_avatar = None
self.their_avatar_thumb = None self.their_avatar_thumb = None
self.status = None self.status = None
self.media_base = ""
def add_message(self, id, message): def add_message(self, id, message):
if not isinstance(message, Message): if not isinstance(message, Message):

View File

@@ -246,10 +246,13 @@ def media(db, data, media_folder, filter_date, filter_chat, separate_media=False
while content is not None: while content is not None:
file_path = f"{media_folder}/Message/{content['ZMEDIALOCALPATH']}" file_path = f"{media_folder}/Message/{content['ZMEDIALOCALPATH']}"
ZMESSAGE = content["ZMESSAGE"] ZMESSAGE = content["ZMESSAGE"]
message = data[content["ZCONTACTJID"]].messages[ZMESSAGE] contact = data[content["ZCONTACTJID"]]
message = contact.messages[ZMESSAGE]
message.media = True message.media = True
if contact.media_base == "":
contact.media_base = media_folder + "/"
if os.path.isfile(file_path): if os.path.isfile(file_path):
message.data = file_path message.data = '/'.join(file_path.split("/")[1:])
if content["ZVCARDSTRING"] is None: if content["ZVCARDSTRING"] is None:
guess = mime.guess_type(file_path)[0] guess = mime.guess_type(file_path)[0]
if guess is not None: if guess is not None:
@@ -259,7 +262,7 @@ def media(db, data, media_folder, filter_date, filter_chat, separate_media=False
else: else:
message.mime = content["ZVCARDSTRING"] message.mime = content["ZVCARDSTRING"]
if separate_media: if separate_media:
chat_display_name = slugify(data[content["ZCONTACTJID"]].name or message.sender \ chat_display_name = slugify(contact.name or message.sender \
or content["ZCONTACTJID"].split('@')[0], True) or content["ZCONTACTJID"].split('@')[0], True)
current_filename = file_path.split("/")[-1] current_filename = file_path.split("/")[-1]
new_folder = os.path.join(media_folder, "separated", chat_display_name) new_folder = os.path.join(media_folder, "separated", chat_display_name)

View File

@@ -93,6 +93,7 @@ def rendering(
w3css=w3css, w3css=w3css,
next=next, next=next,
status=chat.status, status=chat.status,
media_base=chat.media_base
) )
) )

View File

@@ -87,6 +87,7 @@
max-height: 100px !important; max-height: 100px !important;
} }
</style> </style>
<base href="{{ media_base }}" target="_blank">
</head> </head>
<body> <body>
<header class="w3-center w3-top"> <header class="w3-center w3-top">