#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
任务模板模型
对应vwed_tasktemplate表
"""

from sqlalchemy import Column, String, Integer
from data.models.base import BaseModel


class VWEDTaskTemplate(BaseModel):
    """
    任务模板模型
    对应vwed_tasktemplate表
    功能：定义可用的任务模板类型
    """
    __tablename__ = 'vwed_tasktemplate'

    __table_args__ = {
        'mysql_engine': 'InnoDB',
        'mysql_charset': 'utf8mb4',
        'mysql_collate': 'utf8mb4_general_ci',
        'info': {'order_by': 'created_at DESC'}
    }

    id = Column(String(255), primary_key=True, nullable=False, comment='模板ID')
    template_name = Column(String(255), nullable=False, comment='模板名称')
    template_description = Column(String(255), nullable=False, comment='模板描述')
    template_if_enable = Column(Integer, nullable=False, default=0, comment='是否启用')
    template_dir = Column(String(255), comment='模板目录')
    
    def __repr__(self):
        return f"<VWEDTaskTemplate(id='{self.id}', template_name='{self.template_name}')>" 