"""Cookie管理数据模型"""
import datetime
from applications.extensions import db


class CookieUser(db.Model):
    """登录用户 - 保存已登录用户的Cookie信息"""
    __tablename__ = 'admin_cookie_user'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    username = db.Column(db.String(100), nullable=False, comment='用户名')
    nickname = db.Column(db.String(100), default='', comment='昵称')
    website_url = db.Column(db.String(255), nullable=False, comment='网站URL')
    website_name = db.Column(db.String(100), default='', comment='网站名称')
    cookie_value = db.Column(db.Text, nullable=False, comment='Cookie值')
    domain = db.Column(db.String(255), default='', comment='Cookie域名')
    expiry = db.Column(db.DateTime, comment='过期时间')
    enable = db.Column(db.Integer, default=1, comment='状态(1启用,0禁用)')
    create_by = db.Column(db.String(50), default='admin', comment='创建者')
    create_at = db.Column(db.DateTime, default=datetime.datetime.now, comment='创建时间')
    update_at = db.Column(db.DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now, comment='更新时间')

    def to_dict(self):
        return {
            'id': self.id,
            'username': self.username,
            'nickname': self.nickname,
            'website_url': self.website_url,
            'website_name': self.website_name,
            'cookie_value': self.cookie_value,
            'domain': self.domain,
            'expiry': self.expiry.strftime('%Y-%m-%d %H:%M:%S') if self.expiry else '',
            'enable': self.enable,
            'create_by': self.create_by,
            'create_at': self.create_at.strftime('%Y-%m-%d %H:%M:%S') if self.create_at else '',
            'update_at': self.update_at.strftime('%Y-%m-%d %H:%M:%S') if self.update_at else '',
        }


class CookieSite(db.Model):
    """登录网站 - 保存网站的Cookie信息"""
    __tablename__ = 'admin_cookie_site'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    name = db.Column(db.String(100), nullable=False, comment='网站名称')
    url = db.Column(db.String(255), nullable=False, comment='网站URL')
    cookie_value = db.Column(db.Text, nullable=False, comment='Cookie值')
    domain = db.Column(db.String(255), default='', comment='Cookie域名')
    path = db.Column(db.String(255), default='/', comment='Cookie路径')
    expiry = db.Column(db.DateTime, comment='过期时间')
    secure = db.Column(db.Integer, default=0, comment='是否安全')
    http_only = db.Column(db.Integer, default=0, comment='是否HTTPOnly')
    enable = db.Column(db.Integer, default=1, comment='状态(1启用,0禁用)')
    create_by = db.Column(db.String(50), default='admin', comment='创建者')
    create_at = db.Column(db.DateTime, default=datetime.datetime.now, comment='创建时间')
    update_at = db.Column(db.DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now, comment='更新时间')

    def to_dict(self):
        return {
            'id': self.id,
            'name': self.name,
            'url': self.url,
            'cookie_value': self.cookie_value,
            'domain': self.domain,
            'path': self.path,
            'expiry': self.expiry.strftime('%Y-%m-%d %H:%M:%S') if self.expiry else '',
            'secure': self.secure,
            'http_only': self.http_only,
            'enable': self.enable,
            'create_by': self.create_by,
            'create_at': self.create_at.strftime('%Y-%m-%d %H:%M:%S') if self.create_at else '',
            'update_at': self.update_at.strftime('%Y-%m-%d %H:%M:%S') if self.update_at else '',
        }