File size: 1,362 Bytes
a496aae 71c1c20 a496aae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | from sqlalchemy import Column, Integer, String, DateTime, Enum
from sqlalchemy.sql import func
from sqlalchemy.orm import relationship
from app.core.database import Base
import enum
class RepositoryStatusEnum(enum.Enum):
"""Repository processing status"""
PENDING = "PENDING"
PROCESSING = "PROCESSING"
READY = "READY"
FAILED = "FAILED"
class Repository(Base):
"""Repository model with user ownership"""
__tablename__ = "repositories"
id = Column(Integer, primary_key=True, index=True)
user_id = Column(String, nullable=False, index=True) # ✅ Added back!
github_url = Column(String, nullable=False, unique=True)
name = Column(String, nullable=False)
status = Column(Enum(RepositoryStatusEnum), default=RepositoryStatusEnum.PENDING)
error_message = Column(String, nullable=True)
created_at = Column(DateTime(timezone=True), server_default=func.now())
updated_at = Column(DateTime(timezone=True), onupdate=func.now())
# Relationships
conversations = relationship("Conversation", back_populates="repository", cascade="all, delete-orphan")
code_files = relationship("CodeFile", back_populates="repository", cascade="all, delete-orphan")
def __repr__(self):
return f"<Repository(id={self.id}, user_id='{self.user_id}', name='{self.name}', status={self.status.value})>" |