dekun
3149770887
Remove Gate scout auto-seeding and the manual hub login button while keeping Gate login and instance SSO. Add password change and database backup/restore, plus deploy scripts with env auto-generation. Co-authored-by: Cursor <cursoragent@cursor.com>
114 lines
3.9 KiB
Python
114 lines
3.9 KiB
Python
from flask_wtf import FlaskForm
|
||
from flask_wtf.file import FileAllowed, FileField
|
||
from wtforms import IntegerField, PasswordField, SelectField, StringField, SubmitField
|
||
from wtforms.validators import DataRequired, EqualTo, Length, NumberRange, Optional, ValidationError
|
||
|
||
|
||
class LoginForm(FlaskForm):
|
||
username = StringField("用户名", validators=[DataRequired(message="请输入用户名")])
|
||
password = PasswordField("密码", validators=[DataRequired(message="请输入密码")])
|
||
submit = SubmitField("登录")
|
||
|
||
|
||
class GroupForm(FlaskForm):
|
||
name = StringField("分组名称", validators=[DataRequired(message="请输入分组名称")])
|
||
icon = SelectField(
|
||
"图标",
|
||
choices=[
|
||
("", "自动（按分组名匹配）"),
|
||
("order", "下单 / 订单"),
|
||
("trade", "交易 / 行情"),
|
||
("review", "复盘 / 历史"),
|
||
("api", "API / 接口"),
|
||
("gate", "Gate / 扫单"),
|
||
("chart", "图表 / K线"),
|
||
("server", "服务器 / 面板"),
|
||
("folder", "文件夹（通用）"),
|
||
],
|
||
default="",
|
||
validators=[Optional()],
|
||
)
|
||
sort_order = IntegerField(
|
||
"排序（越小越靠前）",
|
||
validators=[Optional(), NumberRange(min=-10**6, max=10**6)],
|
||
default=0,
|
||
)
|
||
submit = SubmitField("保存")
|
||
|
||
|
||
class ServiceForm(FlaskForm):
|
||
name = StringField("服务名称", validators=[DataRequired(message="请输入服务名称")])
|
||
scheme = SelectField(
|
||
"协议",
|
||
choices=[("http", "HTTP"), ("https", "HTTPS")],
|
||
default="http",
|
||
validators=[DataRequired(message="请选择协议")],
|
||
)
|
||
host = StringField(
|
||
"主机或域名",
|
||
validators=[DataRequired(message="请输入主机或域名")],
|
||
)
|
||
port = IntegerField(
|
||
"端口",
|
||
validators=[
|
||
DataRequired(message="请输入端口"),
|
||
NumberRange(min=1, max=65535, message="端口范围 1–65535"),
|
||
],
|
||
)
|
||
path = StringField("路径（可选，默认 /）", validators=[Optional()])
|
||
group_id = SelectField("分组", coerce=int, validators=[DataRequired(message="请选择分组")])
|
||
sort_order = IntegerField(
|
||
"排序（越小越靠前）",
|
||
validators=[Optional(), NumberRange(min=-10**6, max=10**6)],
|
||
default=0,
|
||
)
|
||
embed_kind = SelectField(
|
||
"嵌入类型",
|
||
choices=[
|
||
("", "普通（直接打开路径）"),
|
||
("hub", "复盘中控（云端 hub，需 embed-auth 登录）"),
|
||
("gate_scout", "Gate 扫描端（iframe 代登录）"),
|
||
("gate_exec", "Gate 执行器（iframe 代登录）"),
|
||
],
|
||
default="",
|
||
validators=[Optional()],
|
||
)
|
||
submit = SubmitField("保存")
|
||
|
||
def validate_path(self, field):
|
||
v = (field.data or "").strip()
|
||
if v and not v.startswith("/"):
|
||
raise ValidationError("路径需以 / 开头，例如 /admin")
|
||
|
||
|
||
class ChangePasswordForm(FlaskForm):
|
||
current_password = PasswordField(
|
||
"当前密码", validators=[DataRequired(message="请输入当前密码")]
|
||
)
|
||
new_password = PasswordField(
|
||
"新密码",
|
||
validators=[
|
||
DataRequired(message="请输入新密码"),
|
||
Length(min=6, message="新密码至少 6 位"),
|
||
],
|
||
)
|
||
confirm_password = PasswordField(
|
||
"确认新密码",
|
||
validators=[
|
||
DataRequired(message="请再次输入新密码"),
|
||
EqualTo("new_password", message="两次输入的新密码不一致"),
|
||
],
|
||
)
|
||
submit = SubmitField("保存密码")
|
||
|
||
|
||
class RestoreBackupForm(FlaskForm):
|
||
backup_file = FileField(
|
||
"备份文件",
|
||
validators=[
|
||
DataRequired(message="请选择备份文件"),
|
||
FileAllowed(["db"], message="仅支持 .db 文件"),
|
||
],
|
||
)
|
||
submit = SubmitField("一键恢复")
|