mirror of
http://112.124.100.131/huang.ze/ebiz-dify-ai.git
synced 2025-12-09 02:46:52 +08:00
Fix: multi image preview sign (#5376)
Co-authored-by: huangyusong <huangyusong@yingzi.com>
This commit is contained in:
@@ -425,9 +425,9 @@ class DocumentSegment(db.Model):
|
||||
def get_sign_content(self):
|
||||
pattern = r"/files/([a-f0-9\-]+)/image-preview"
|
||||
text = self.content
|
||||
match = re.search(pattern, text)
|
||||
|
||||
if match:
|
||||
matches = re.finditer(pattern, text)
|
||||
signed_urls = []
|
||||
for match in matches:
|
||||
upload_file_id = match.group(1)
|
||||
nonce = os.urandom(16).hex()
|
||||
timestamp = str(int(time.time()))
|
||||
@@ -437,8 +437,15 @@ class DocumentSegment(db.Model):
|
||||
encoded_sign = base64.urlsafe_b64encode(sign).decode()
|
||||
|
||||
params = f"timestamp={timestamp}&nonce={nonce}&sign={encoded_sign}"
|
||||
replacement = r"\g<0>?{params}".format(params=params)
|
||||
text = re.sub(pattern, replacement, text)
|
||||
signed_url = f"{match.group(0)}?{params}"
|
||||
signed_urls.append((match.start(), match.end(), signed_url))
|
||||
|
||||
# Reconstruct the text with signed URLs
|
||||
offset = 0
|
||||
for start, end, signed_url in signed_urls:
|
||||
text = text[:start + offset] + signed_url + text[end + offset:]
|
||||
offset += len(signed_url) - (end - start)
|
||||
|
||||
return text
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user