fix message file_preview url:if as_attachment is in the url, add it to the sign_url

1 year ago · 0ab99c485c
parent eb92dd59f9
commit 0ab99c485c
1 changed files with 3 additions and 1 deletions
--- a/api/models/model.py
+++ b/api/models/model.py
@ -1012,7 +1012,9 @@ class Message(db.Model):  # type: ignore[name-defined]
                sign_url = file_helpers.get_signed_file_url(upload_file_id)
            else:
                continue
-
+            # if as_attachment is in the url, add it to the sign_url
+            if "as_attachment" in url:
+                sign_url += "&as_attachment=true"
            re_sign_file_url_answer = re_sign_file_url_answer.replace(url, sign_url)

        return re_sign_file_url_answer