from sqlalchemy import Column, Integer, String, Text, DateTime
from sqlalchemy.sql import text
from database import Base

class Company(Base):
    __tablename__ = "companies"
    
    id = Column(Integer, primary_key=True, index=True, comment='主键ID')
    company_code = Column(String(50), nullable=False, server_default="", unique=True, index=True, comment='公司编码')
    name = Column(String(100), nullable=False, server_default="", comment='公司名称')
    logo_url = Column(String(255), nullable=False, server_default="", comment='公司logo')
    description = Column(Text, nullable=False, default="", comment='公司描述')
    address = Column(Text, nullable=False, default="", comment='公司地址')
    phone = Column(String(50), nullable=False, server_default="", comment='联系电话')
    email = Column(String(100), nullable=False, server_default="", comment='联系邮箱')
    website = Column(String(255), nullable=False, server_default="", comment='公司网站')
    business_license = Column(String(255), nullable=False, server_default="", comment='营业执照')
    established_date = Column(DateTime, nullable=True, comment='成立日期')
    employees_count = Column(Integer, nullable=False, server_default="0", comment='员工数量')
    business_scope = Column(Text, nullable=False, default="", comment='经营范围')
    level = Column(Integer, nullable=False, server_default="0", comment='会员等级：0-普通会员，1-金卡，2-钻石，3-皇冠')
    template = Column(String(50), nullable=False, server_default="Home", comment='模板选择')
    created_at = Column(DateTime(timezone=True), nullable=False, server_default=text('CURRENT_TIMESTAMP'), comment='创建时间')
    updated_at = Column(DateTime(timezone=True), nullable=False, server_default=text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'), comment='更新时间')