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

class Membership(Base):
    __tablename__ = "membership"
    
    id = Column(Integer, primary_key=True, index=True, comment='主键ID')
    company_code = Column(String(50), nullable=False, server_default="", index=True, comment='公司编码')
    level = Column(Integer, nullable=False, server_default="0", comment='会员等级')
    start_date = Column(DateTime, nullable=True, comment='有效期开始')
    end_date = Column(DateTime, nullable=True, 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='更新时间')