init

backend/app/models/review.py

@@ -0,0 +1,43 @@
+"""Review model"""
+
+from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Enum
+from sqlalchemy.orm import relationship
+from sqlalchemy.sql import func
+from datetime import datetime
+import enum
+from typing import Optional
+
+from app.database import Base
+
+
+class ReviewStatusEnum(str, enum.Enum):
+    """Review status"""
+    PENDING = "pending"
+    FETCHING = "fetching"
+    ANALYZING = "analyzing"
+    COMMENTING = "commenting"
+    COMPLETED = "completed"
+    FAILED = "failed"
+
+
+class Review(Base):
+    """Code review model"""
+    
+    __tablename__ = "reviews"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    pull_request_id = Column(Integer, ForeignKey("pull_requests.id"), nullable=False)
+    status = Column(Enum(ReviewStatusEnum), default=ReviewStatusEnum.PENDING)
+    started_at = Column(DateTime, default=datetime.utcnow, server_default=func.now())
+    completed_at = Column(DateTime, nullable=True)
+    files_analyzed = Column(Integer, default=0)
+    comments_generated = Column(Integer, default=0)
+    error_message = Column(String, nullable=True)
+    
+    # Relationships
+    pull_request = relationship("PullRequest", back_populates="reviews")
+    comments = relationship("Comment", back_populates="review", cascade="all, delete-orphan")
+    
+    def __repr__(self):
+        return f"<Review(id={self.id}, status={self.status}, pr_id={self.pull_request_id})>"
+