from sqlalchemy import Column, Integer, String, JSON, Float, ForeignKey, DateTime
from app.db.database import Base
import datetime

class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    username = Column(String, unique=True, index=True)
    hashed_password = Column(String)
    display_name = Column(String, nullable=True)
    email_id = Column(String, nullable=True)
    mobile_no = Column(String, nullable=True)
    gender = Column(String, nullable=True)
    sector_map = Column(JSON, nullable=True)

class PortfolioItem(Base):
    __tablename__ = "portfolio_items"

    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(Integer, ForeignKey("users.id"))
    symbol = Column(String, index=True)
    quantity = Column(Integer, default=1)
    buy_price = Column(Float)
    purchase_date = Column(DateTime, default=datetime.datetime.utcnow)