浏览代码

fix: 修复完整URL显示错误的问题

SongZihuan 2 年之前
父节点
当前提交
44290a0cba
共有 2 个文件被更改,包括 4 次插入2 次删除
  1. 2 1
      app/auth.py
  2. 2 1
      app/oss.py

+ 2 - 1
app/auth.py

@@ -14,6 +14,7 @@ from wtforms.validators import DataRequired, Length, Regexp, EqualTo
 import app
 from object.user import User
 from send_email import send_msg
+from configure import conf
 
 auth = Blueprint("auth", __name__)
 
@@ -171,7 +172,7 @@ def register_page():
     form = RegisterForm()
     if form.validate_on_submit():
         token = User.creat_token(form.email.data, form.passwd.data)
-        register_url = url_for("auth.confirm_page", token=token, _external=True)
+        register_url = conf["URL_NAME"] + url_for("auth.confirm_page", token=token)
         hblog: app.Hblog = current_app
         send_msg("注册确认", hblog.mail, form.email.data, "register", register_url=register_url)
         flash("注册提交成功, 请进入邮箱点击确认注册链接")

+ 2 - 1
app/oss.py

@@ -4,6 +4,7 @@ from flask_wtf import FlaskForm
 from wtforms import FileField, StringField, SubmitField
 from wtforms.validators import DataRequired, Length
 
+import configure
 from aliyun import aliyun
 import app
 
@@ -52,7 +53,7 @@ def upload_page():
         path += file.filename
         aliyun.upload_file(path, file)
         app.HBlogFlask.print_sys_opt_success_log(f"Upload file {path}")
-        flash(f"文件 {file.filename} 已上传: {url_for('oss.get_page', name=path, _external=True)}")
+        flash(f"文件 {file.filename} 已上传: {configure.conf['URL_NAME'] + url_for('oss.get_page', name=path)}")
         return redirect(url_for("oss.upload_page"))
     app.HBlogFlask.print_load_page_log(f"OSS upload")
     return render_template("oss/upload.html", UploadForm=form)