浏览代码

Strip off http:// or https:// prefix in domain

Son NK 5 年之前
父节点
当前提交
103418dff7
共有 1 个文件被更改,包括 7 次插入0 次删除
  1. 7 0
      app/dashboard/views/custom_domain.py

+ 7 - 0
app/dashboard/views/custom_domain.py

@@ -33,6 +33,13 @@ def custom_domain():
 
             if new_custom_domain_form.validate():
                 new_domain = new_custom_domain_form.domain.data.lower().strip()
+
+                if new_domain.startswith("http://"):
+                    new_domain = new_domain[len("http://") :]
+
+                if new_domain.startswith("https://"):
+                    new_domain = new_domain[len("https://") :]
+
                 if CustomDomain.get_by(domain=new_domain):
                     flash(f"{new_domain} already added", "warning")
                 elif get_email_domain_part(current_user.email) == new_domain: