# _*_ coding: utf-8 _*_

from flask_wtf import FlaskForm
from flask_wtf.file import FileAllowed
from wtforms import StringField, PasswordField, SubmitField, FileField, TextAreaField, RadioField, SelectField, BooleanField, EmailField
from wtforms.validators import DataRequired, ValidationError, EqualTo, Email, InputRequired, Regexp, Length, Optional
from app.models import Admin


class LoginForm(FlaskForm):
    """
    管理员登录表单
    """
    account = StringField(
        label="账号",
        validators=[
            DataRequired("账号不能为空")
        ],
        description="账号",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入账号！",
        }
    )
    pwd = PasswordField(
        label="密码",
        validators=[
            DataRequired("密码不能为空")
        ],
        description="密码",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入密码！",
        }
    )
    submit = SubmitField(
        '登录',
        render_kw={
            "class": "btn btn-primary btn-block btn-flat",
        }
    )

    # 验证账号，命名规则：validate_ + 字段名。如果要验证密码，则可以创建函数validate_pwd
    def validate_account(self, field):
        account = field.data
        admin = Admin.query.filter_by(name=account).count()
        if admin == 0:
            raise ValidationError("账号不存在! ")


class PwdForm(FlaskForm):
    old_pwd = PasswordField(
        label="旧密码",
        validators=[
            DataRequired("旧密码不能为空！")
        ],
        description="旧密码",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入旧密码！",
        }
    )
    new_pwd = PasswordField(
        label="新密码",
        validators=[
            DataRequired("新密码不能为空！")
        ],
        description="新密码",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入新密码！",
        }
    )
    submit = SubmitField(
        '保存',
        render_kw={
            "class": "btn btn-primary",
        }
    )

    def validate_old_pwd(self, field):
        from flask import session
        pwd = field.data
        name = session["admin"]
        admin = Admin.query.filter_by(
            name=name
        ).first()
        if not admin.check_pwd(pwd):
            raise ValidationError("旧密码错误！")


class AreaForm(FlaskForm):
    """添加/编辑地区的表单"""
    name = StringField(
        label="名称",
        validators=[
            DataRequired("地区名不能为空")
        ],
        description="地区",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入地区名称！"
        }
    )
    is_recommended = RadioField(
        label='是否推荐',
        description="是否推荐",
        coerce=int,
        choices=[(0, '否'), (1, '是')], default=0,
    )
    introduction = TextAreaField(
        label="简介",
        validators=[
            DataRequired("简介不能为空！")
        ],
        description="简介",
        render_kw={
            "class": "form-control",
            "rows": 5
        }
    )

    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class DeviceTypeForm(FlaskForm):
    """添加/编辑地区的表单"""
    name = StringField(
        label="设备名称",
        validators=[
            DataRequired("设备名称不能为空！")
        ],
        description="设备名称",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入设备名称！"
        }
    )

    type = StringField(
        label="设备编号",
        validators=[
            DataRequired("设备编号不能为空！")
        ],
        description="设备编号",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入整数类型的设备编号！"
        }
    )

    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class DetectingPointForm(FlaskForm):
    """添加/编辑检测点的表单"""
    name = StringField(
        label="检测点名称",
        validators=[
            DataRequired("检测点名称不能为空！")
        ],
        description="检测点名称",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入检测点名称！"
        }
    )

    no = StringField(
        label="检测点编号",
        validators=[
            DataRequired("检测点编号不能为空！")
        ],
        description="检测点编号",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入整数类型的设备编号！"
        }
    )

    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class PowerForm(FlaskForm):
    """添加/编辑权限的表单"""
    name = StringField(
        label="权限名称",
        validators=[
            DataRequired("权限名称不能为空！")
        ],
        description="权限名称",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入权限名称！"
        }
    )

    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class RoleForm(FlaskForm):
    """添加/编辑权限的表单"""
    name = StringField(
        label="角色名称",
        validators=[
            DataRequired("角色名称不能为空！")
        ],
        description="角色名称",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入角色名称！"
        }
    )

    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class ScenicForm(FlaskForm):
    title = StringField(
        label="景区名称",
        validators=[
            DataRequired("景区名称不能为空！")
        ],
        description="景区名称",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入景区名称！"
        }
    )
    address = StringField(
        label="景区地址",
        validators=[
            DataRequired("景区地址不能为空！")
        ],
        description="景区地址",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入景区地址！"
        }
    )
    star = SelectField(
        label="星级",
        validators=[
            DataRequired("请选择星级！")
        ],
        coerce=int,
        choices=[(1, "1星"), (2, "2星"), (3, "3星"), (4, "4星"), (5, "5星")], default=5,
        description="星级",
        render_kw={
            "class": "form-control",
        }
    )

    logo = FileField(
        label="封面",
        validators=[
            DataRequired("请上传封面！"),
            FileAllowed(['jpg', 'png'], '请上传jpg或png格式图片!')
        ],
        description="封面",
    )

    is_hot = RadioField(
        label='是否热门',
        description="是否热门",
        coerce=int,
        choices=[(0, '否'), (1, '是')], default=0,
    )
    is_recommended = RadioField(
        label='是否推荐',
        description="是否推荐",
        coerce=int,
        choices=[(0, '否'), (1, '是')], default=0,
    )
    introduction = TextAreaField(
        label="景区简介",
        validators=[
            DataRequired("简介不能为空！")
        ],
        description="简介",
        render_kw={
            "class": "form-control",
            "rows": 5
        }
    )
    content = TextAreaField(
        label="景区内容",
        validators=[
            DataRequired("景区内容不能为空！")
        ],
        description="景区内容",
        render_kw={
            "class": "form-control ckeditor",
            "rows": 10
        }
    )
    area_id = SelectField(
        label="所属地区",
        validators=[
            DataRequired("请选择标签！")
        ],
        coerce=int,
        description="所属地区",
        render_kw={
            "class": "form-control",
        }
    )
    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class AdminForm(FlaskForm):
    def __init__(self, edit, **kwargs):
        super().__init__(**kwargs)

        self.name.render_kw = {
            "class": "form-control",
            "placeholder": "请输入管理员名称！",
            "readonly": edit,
        }

    name = StringField(
        label="管理员名称",
        validators=[
            DataRequired("管理员名称不能为空！")
        ],

        description="管理员名称",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入管理员名称！",
        }
    )

    pwd = PasswordField(
        label="密码",
        validators=[
            DataRequired("密码不能为空")
        ],
        description="密码",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入密码！",
        }
    )
    repwd = PasswordField(
        label="确认密码",
        validators=[
            DataRequired("确认密码不能为空"),
            EqualTo('pwd', message="两次输入密码不一致"),
        ],
        description="确认密码",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入确认密码！",
        }
    )
    role_id = SelectField(
        label="角色",
        validators=[
            DataRequired("请选择角色！")
        ],
        coerce=int,
        description="角色",
        render_kw={
            "class": "form-control",
        }
    )
    power_id = SelectField(
        label="权限",
        validators=[
            DataRequired("请选择权限！")
        ],
        coerce=int,
        description="权限",
        render_kw={
            "class": "form-control",
        }
    )
    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class UserForm(FlaskForm):
    def __init__(self, edit, **kwargs):
        super().__init__(**kwargs)

        self.username.render_kw = {
            "class": "form-control",
            "placeholder": "请输入用户名！",
            "readonly": edit,
        }

    username = StringField(
        label="用户名",
        validators=[
            DataRequired("用户名不能为空！")
        ],

        description="用户名",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入用户名！",
        }
    )
    email = EmailField(
        label="邮箱",
        validators=[
            Optional(),
            Email()
        ],

        description="邮箱",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入邮箱地址！",
        }
    )

    phone = StringField(
        label="手机号码",
        validators=[
            Optional(),
            Length(11, 11),
            Regexp('^1[3-9]\\d{9}$', 0, '手机号码格式错误')
        ],

        description="手机号码",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入手机号码！",
        }
    )

    pwd = PasswordField(
        label="密码",
        validators=[
            DataRequired("密码不能为空")
        ],
        description="密码",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入密码！",
        }
    )
    repwd = PasswordField(
        label="确认密码",
        validators=[
            DataRequired("确认密码不能为空"),
            EqualTo('pwd', message="两次输入密码不一致"),
        ],
        description="确认密码",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入确认密码！",
        }
    )
    role_id = SelectField(
        label="角色",
        validators=[
            DataRequired("请选择角色！")
        ],
        coerce=int,
        description="角色",
        render_kw={
            "class": "form-control",
        }
    )
    power_id = SelectField(
        label="权限",
        validators=[
            DataRequired("请选择权限！")
        ],
        coerce=int,
        description="权限",
        render_kw={
            "class": "form-control",
        }
    )
    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class DeviceForm(FlaskForm):

    def __init__(self, edit, **kwargs):
        super().__init__(**kwargs)

        self.title.render_kw = {
            "class": "form-control",
            "placeholder": "请输入设备名称！",
            "readonly": edit,
        }

        self.devicetype_id.render_kw = {
                "class": "form-control",
                "readonly": edit,
        }

    title = StringField(
        label="设备名称",
        validators=[
            DataRequired("设备名称不能为空！")
        ],

        description="设备名称",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入设备名称！",
        }
    )
    address = StringField(
        label="设备地址",
        validators=[
            DataRequired("设备地址不能为空！")
        ],
        description="设备地址",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入设备地址！",

        }
    )
    threshold = StringField(
        label="报警阈值",

        description="报警阈值",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入报警阈值！",

        }
    )
    logo = FileField(
        label="封面",
        validators=[
            # DataRequired("请上传封面！"),
            FileAllowed(['jpg', 'png'], '请上传jpg或png格式图片!')
        ],
        description="封面",
    )

    introduction = TextAreaField(
        label="备注信息",

        description="备注",
        render_kw={
            "class": "form-control",
            "rows": 5
        }
    )

    devicetype_id = SelectField(
        label="设备类型",
        validators=[
            DataRequired("请选择设备类型！")
        ],
        coerce=int,
        description="设备类型",
        render_kw={
            "class": "form-control",
        }
    )
    detectingpoint_id = SelectField(
        label="检测点",
        validators=[
            DataRequired("请选择检测点！")
        ],
        coerce=int,
        description="检测点",
        render_kw={
            "class": "form-control",
        }
    )
    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )


class TravelsForm(FlaskForm):
    title = StringField(
        label="标题",
        validators=[
            DataRequired("标题不能为空！")
        ],
        description="标题",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入标题！"
        }
    )
    author = StringField(
        label="作者",
        validators=[
            DataRequired("作者不能为空！")
        ],
        description="作者",
        render_kw={
            "class": "form-control",
            "placeholder": "请输入作者！"
        }
    )
    content = TextAreaField(
        label="游记内容",
        validators=[
            DataRequired("游记内容不能为空！")
        ],
        description="游记内容",
        render_kw={
            "class": "form-control ckeditor",
        }
    )
    scenic_id = SelectField(
        label="所属景区",
        validators=[
            DataRequired("请选择景区！")
        ],
        coerce=int,
        description="所属景区",
        render_kw={
            "class": "form-control",
        }
    )
    submit = SubmitField(
        '添加',
        render_kw={
            "class": "btn btn-primary",
        }
    )