from . import db
from datetime import datetime


# 角色
class Role(db.Model):
    __tablename__ = "role"
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100))  # 名称
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 新增时间

    admin = db.relationship("Admin", backref='role')  # 外键关系关联


# 权限
class Power(db.Model):
    __tablename__ = "power"
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100))  # 名称
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 新增时间

    admin = db.relationship("Admin", backref='power')  # 外键关系关联


# 会员数据模型
class User(db.Model):
    __tablename__ = "user"
    #__table_args__ = {"useexisting": True}
    id = db.Column(db.Integer, primary_key=True)  # 编号
    username = db.Column(db.String(100))  # 用户名
    pwd = db.Column(db.String(1024))  # 密码
    email = db.Column(db.String(100))  # 邮箱
    phone = db.Column(db.String(11))  # 手机号
    info = db.Column(db.Text)  # 个性简介
    face = db.Column(db.String(255), unique=True)  # 头像
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 注册时间

    userlogs = db.relationship('Userlog', backref='user')  # 会员日志外键关系关联
    collect = db.relationship('Collect', backref='user')  # 收藏外键关系关联

    role_id = db.Column(db.Integer, db.ForeignKey('role.id'))  # 所属标签
    power_id = db.Column(db.Integer, db.ForeignKey('power.id'))  # 所属标签

    def __repr__(self):
        return '<User %r>' % self.name

    def check_pwd(self, pwd):
        """
        检测密码是否正确
        :param pwd: 密码
        :return: 返回布尔值
        """
        from werkzeug.security import check_password_hash
        return check_password_hash(self.pwd, pwd)


# 管理员
class Admin(db.Model):
    __tablename__ = "admin"
    #__table_args__ = {"useexisting": True}
    id = db.Column(db.Integer, primary_key=True)  # 编号
    name = db.Column(db.String(100), unique=True)  # 管理员账号
    pwd = db.Column(db.String(100))  # 管理员密码
    is_super = db.Column(db.Integer)
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 时间

    role_id = db.Column(db.Integer, db.ForeignKey('role.id'))  # 所属标签
    power_id = db.Column(db.Integer, db.ForeignKey('power.id'))  # 所属标签
    adminlogs = db.relationship("Adminlog", backref='admin')  # 管理员登录日志外键关系关联
    oplogs = db.relationship("Oplog", backref='admin')  # 管理员操作日志外键关系关联

    # handle_user = db.Column(db.Integer, db.ForeignKey('alarm.handle_user'))  # 所属标签
    alarm = db.relationship("Alarm", backref='admin')  # 外键关系关联

    def __repr__(self):
        return "<Admin %r>" % self.name

    def check_pwd(self, pwd):
        """
        检测密码是否正确
        :param pwd: 密码
        :return: 返回布尔值
        """
        from werkzeug.security import check_password_hash
        return check_password_hash(self.pwd, pwd)


# 管理员登录日志
class Adminlog(db.Model):
    __tablename__ = "adminlog"
    #__table_args__ = {"useexisting": True}
    id = db.Column(db.Integer, primary_key=True)  # 编号
    admin_id = db.Column(db.Integer, db.ForeignKey('admin.id'))  # 所属管理员
    ip = db.Column(db.String(100))  # 登录IP
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 登录时间

    def __repr__(self):
        return "<Adminlog %r>" % self.id


# 操作日志
class Oplog(db.Model):
    __tablename__ = "oplog"
    #__table_args__ = {"useexisting": True}
    id = db.Column(db.Integer, primary_key=True)  # 编号
    admin_id = db.Column(db.Integer, db.ForeignKey('admin.id'))  # 所属管理员
    ip = db.Column(db.String(100))  # 操作IP
    reason = db.Column(db.String(600))  # 操作原因
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 登录时间

    def __repr__(self):
        return "<Oplog %r>" % self.id


# 地区
class Area(db.Model):
    __tablename__ = "area"
    id = db.Column(db.Integer, primary_key=True)  # 编号
    name = db.Column(db.String(100), unique=True)  # 标题
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加景区时间
    is_recommended = db.Column(db.Boolean(), default=0)  # 是否推荐
    introduction = db.Column(db.Text)  # 景区简介
    scenic = db.relationship("Scenic", backref='area')  # 外键关系关联

    def __repr__(self):
        return "<Area %r>" % self.name


# 检测点
class DetectingPoint(db.Model):
    __tablename__ = "detectingpoint"
    id = db.Column(db.Integer, primary_key=True)  # id
    no = db.Column(db.Integer)  # 编号
    name = db.Column(db.String(255))  # 名称
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间
    detectingpoint = db.relationship("Device", backref='detectingpoint')  # 外键关系关联

    def __repr__(self):
        return "<DetectingPoint %r>" % self.name


# 会员登录日志
class Userlog(db.Model):
    __tablename__ = "userlog"
    #__table_args__ = {"useexisting": True}
    id = db.Column(db.Integer, primary_key=True)  # 编号
    # 设置外键
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))  # 所属会员
    ip = db.Column(db.String(100))  # ip地址
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 登录时间

    def __repr__(self):
        return '<User %r>' % self.id


# 景区
class Scenic(db.Model):
    __tablename__ = "scenic"
    id = db.Column(db.Integer, primary_key=True)  # 编号
    title = db.Column(db.String(255), unique=True)  # 标题
    star = db.Column(db.Integer)  # 星级
    logo = db.Column(db.String(255), unique=True)  # 封面
    introduction = db.Column(db.Text)  # 景区简介
    content = db.Column(db.Text)  # 景区内容描述
    address = db.Column(db.Text)  # 景区地址
    is_hot = db.Column(db.Boolean(), default=0)  # 是否热门
    is_recommended = db.Column(db.Boolean(), default=0)  # 是否推荐

    # 设置外键
    area_id = db.Column(db.Integer, db.ForeignKey('area.id'))  # 所属标签
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间
    collect = db.relationship("Collect", backref='scenic')  # 收藏外键关系关联
    travels = db.relationship("Travels", backref='scenic')  # 游记外键关系关联

    def __repr__(self):
        return "<Scenic %r>" % self.title


# 设备类型
class DeviceType(db.Model):
    __tablename__ = "devicetype"
    type = db.Column(db.Integer, primary_key=True)  # 类型
    name = db.Column(db.String(255), unique=True)  # 名称
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间

    device = db.relationship("Device", backref='devicetype')  # 外键关系关联

    def __repr__(self):
        return "<DeviceType %r>" % self.name


# 设备类型
class Notification(db.Model):
    __tablename__ = "notification"
    id = db.Column(db.Integer, primary_key=True)  # id
    type = db.Column(db.String(255))  # 类型
    enable = db.Column(db.Integer)  # 启用标志
    account = db.Column(db.String(255), unique=True)  # 名称
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间

    def __repr__(self):
        return "<Notification %r>" % self.account


# 设备
class Device(db.Model):
    __tablename__ = "device"
    id = db.Column(db.Integer, primary_key=True)  # 编号
    title = db.Column(db.String(255), unique=False)  # 标题
    logo = db.Column(db.String(255), unique=True)  # 封面
    introduction = db.Column(db.Text)  # 简介

    address = db.Column(db.String(255))  # 地址
    threshold = db.Column(db.DECIMAL)  # 阈值

    alarm = db.relationship("Alarm", backref='device')  # 外键关系关联

    # 设置外键
    devicetype_id = db.Column(db.Integer, db.ForeignKey('devicetype.type'))  # 所属标签
    detectingpoint_id = db.Column(db.Integer, db.ForeignKey('detectingpoint.id'))  # 所属标签
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间


    def __repr__(self):
        return "<Device %r>" % self.title


# 报警
class Alarm(db.Model):
    __tablename__ = "alarm"
    id = db.Column(db.Integer, primary_key=True)  # 编号
    msg = db.Column(db.String(255))  # 信息
    img = db.Column(db.String(255))  # 图片
    #handle_user = db.Column(db.String(255))  # 处理人
    handle = db.Column(db.Integer)  # 处理
    preview = db.Column(db.Integer)  # 预览
    remark = db.Column(db.String(255), unique=True)  # 备注

    day = db.Column(db.Integer)
    month = db.Column(db.Integer)
    year = db.Column(db.Integer)

    date = db.Column(db.Date)

    # 设置外键
    device_id = db.Column(db.Integer, db.ForeignKey('device.id'))  # 所属标签
    handle_user = db.Column(db.Integer, db.ForeignKey('admin.name'))  # 所属标签

    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间
    updatetime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间

    def __repr__(self):
        return "<Alarm %r>" % self.id


# 游记
class Travels(db.Model):
    __tablename__ = "travels"
    id = db.Column(db.Integer, primary_key=True)  # 编号
    title = db.Column(db.String(255), unique=True)  # 标题
    author = db.Column(db.String(255))  # 作者
    content = db.Column(db.Text)  # 游记内容
    scenic_id = db.Column(db.Integer, db.ForeignKey('scenic.id'))  # 所属景区
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间


# 景区收藏
class Collect(db.Model):
    __tablename__ = "collect"
    #__table_args__ = {"useexisting": True}
    id = db.Column(db.Integer, primary_key=True)  # 编号

    scenic_id = db.Column(db.Integer, db.ForeignKey('scenic.id'))  # 所属景区
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))  # 所属用户
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 添加时间

    def __repr__(self):
        return "<Collect %r>" % self.id


# 意见建议
class Suggestion(db.Model):
    __tablename__ = "suggestion"
    #__table_args__ = {"useexisting": True}
    id = db.Column(db.Integer, primary_key=True)  # 编号
    name = db.Column(db.String(255))  # 昵称
    email = db.Column(db.String(100))  # 邮箱
    content = db.Column(db.Text)  # 意见内容
    addtime = db.Column(db.DateTime, index=True, default=datetime.now)  # 注册时间

    def __repr__(self):
        return "<Suggestion %r>" % self.id