|
from sqlalchemy import Boolean, Column, ForeignKey, Integer, String, Float, ARRAY, DateTime |
|
from sqlalchemy.orm import relationship |
|
from core.database import Base |
|
from datetime import datetime |
|
|
|
class User(Base): |
|
__tablename__ = "users" |
|
|
|
id = Column(Integer, primary_key=True, index=True) |
|
email = Column(String, unique=True, index=True, nullable=False) |
|
password = Column(String, nullable=False) |
|
first_name = Column(String, nullable=True) |
|
last_name = Column(String, nullable=True) |
|
age = Column(Integer, nullable=True) |
|
preferences = Column(ARRAY(String), nullable=True) |
|
is_active = Column(Boolean, default=True) |
|
is_admin = Column(Boolean, default=False) |
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow) |
|
created_at = Column(DateTime, default=datetime.utcnow) |
|
|
|
orders = relationship("Order", back_populates="user") |
|
embeddings = relationship("UserEmbeddings", back_populates="user", uselist=False) |
|
|
|
class UserEmbeddings(Base): |
|
__tablename__ = "user_embeddings" |
|
|
|
id = Column(Integer, primary_key=True, index=True) |
|
user_id = Column(Integer, ForeignKey("users.id"), unique=True, nullable=False) |
|
embeddings = Column(ARRAY(Float), nullable=False) |
|
|
|
user = relationship("User", back_populates="embeddings") |
|
|