import datetime

from sqlalchemy import Column, DateTime, Integer, String

from db.db import Base


class InvestorTable(Base):
    __tablename__ = "investors"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, nullable=False)
    aum = Column(Integer, nullable=False)
    check_size = Column(String, nullable=False)
    sector_focus = Column(String, nullable=False)
    stage_focus = Column(String, nullable=False)
    region = Column(String, nullable=False)
    created_at = Column(DateTime, default=datetime.datetime.now(datetime.UTC))
    updated_at = Column(
        DateTime,
        default=datetime.datetime.now(datetime.UTC),
        onupdate=datetime.datetime.now(datetime.UTC),
    )