init

backend/app/models/pull_request.py

@@ -0,0 +1,43 @@
+"""Pull Request 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 app.database import Base
+
+
+class PRStatusEnum(str, enum.Enum):
+    """Pull Request status"""
+    OPEN = "open"
+    REVIEWING = "reviewing"
+    REVIEWED = "reviewed"
+    CLOSED = "closed"
+
+
+class PullRequest(Base):
+    """Pull Request model"""
+    
+    __tablename__ = "pull_requests"
+    
+    id = Column(Integer, primary_key=True, index=True)
+    repository_id = Column(Integer, ForeignKey("repositories.id"), nullable=False)
+    pr_number = Column(Integer, nullable=False)
+    title = Column(String, nullable=False)
+    author = Column(String, nullable=False)
+    source_branch = Column(String, nullable=False)
+    target_branch = Column(String, nullable=False)
+    status = Column(Enum(PRStatusEnum), default=PRStatusEnum.OPEN)
+    url = Column(String, nullable=False)
+    created_at = Column(DateTime, default=datetime.utcnow, server_default=func.now())
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, server_default=func.now())
+    
+    # Relationships
+    repository = relationship("Repository", back_populates="pull_requests")
+    reviews = relationship("Review", back_populates="pull_request", cascade="all, delete-orphan")
+    
+    def __repr__(self):
+        return f"<PullRequest(id={self.id}, pr_number={self.pr_number}, title={self.title})>"
+