define model

Author: sydowma

backend/app/__init__.py

@@ -0,0 +1,23 @@
+from authlib.integrations.starlette_client import OAuth
+from starlette.config import Config
+from ..config import config
+
+oauth = OAuth(Config(".env"))
+
+oauth.register(
+    name="github",
+    client_id=config.GITHUB_CLIENT_ID,
+    client_secret=config.GITHUB_CLIENT_SECRET,
+    authorize_url="https://github.com/login/oauth/authorize",
+    access_token_url="https://github.com/login/oauth/access_token",
+    api_base_url="https://api.github.com/",
+)
+
+oauth.register(
+    name="gitlab",
+    client_id=config.GITLAB_CLIENT_ID,
+    client_secret=config.GITLAB_CLIENT_SECRET,
+    authorize_url="https://gitlab.com/oauth/authorize",
+    access_token_url="https://gitlab.com/oauth/token",
+    api_base_url="https://gitlab.com/api/v4/",
+)

backend/app/auth/__init__.py

@@ -0,0 +1,13 @@
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+class Config:
+    GITHUB_CLIENT_ID = os.getenv("GITHUB_CLIENT_ID")
+    GITHUB_CLIENT_SECRET = os.getenv("GITHUB_CLIENT_SECRET")
+    GITLAB_CLIENT_ID = os.getenv("GITLAB_CLIENT_ID")
+    GITLAB_CLIENT_SECRET = os.getenv("GITLAB_CLIENT_SECRET")
+    DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///database.db")
+
+config = Config()

backend/app/auth/oauth.py

@@ -0,0 +1,6 @@
+from sqlmodel import Session
+from ..main import engine
+
+def get_session():
+    with Session(engine) as session:
+        yield session

backend/app/config.py

@@ -0,0 +1,24 @@
+from sqlmodel import SQLModel, Field, Relationship
+from typing import Optional, List
+from datetime import datetime
+import json
+from ..user.model import User  # Import User from its new location
+
+class Repository(SQLModel, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True, index=True)
+    owner_id: int = Field(foreign_key="user.id")
+    name: str
+    platform: str = Field(regex="^(github|gitlab)$")
+    api_token: str
+    custom_prompt: Optional[str] = Field(default=None)
+    settings: str = Field(default="{}")
+    created_at: datetime = Field(default_factory=datetime.utcnow)
+    
+    owner: User = Relationship(back_populates="repositories")
+    reviews: List["Review"] = Relationship(back_populates="repository")
+
+    def get_settings(self) -> dict:
+        return json.loads(self.settings)
+    
+    def set_settings(self, settings: dict):
+        self.settings = json.dumps(settings)

backend/app/dashboard/__init__.py

@@ -0,0 +1,16 @@
+from fastapi import APIRouter, Depends
+from sqlmodel import Session
+from ..dependencies import get_session
+from .model import Repository
+from .schema import RepositoryCreate, RepositoryResponse
+
+router = APIRouter()
+
+@router.post("/", response_model=RepositoryResponse)
+def create_repository(repo: RepositoryCreate, session: Session = Depends(get_session)):
+    db_repo = Repository.from_orm(repo, update={"owner_id": 1})  # Hardcoded owner_id for now
+    db_repo.set_settings(repo.settings.dict())
+    session.add(db_repo)
+    session.commit()
+    session.refresh(db_repo)
+    return db_repo

backend/app/dashboard/model.py

@@ -0,0 +1,28 @@
+from pydantic import Field
+from typing import Optional
+from datetime import datetime
+from sqlmodel import SQLModel as SQLModelBase
+
+class RepositorySettings(SQLModelBase):
+    ai_provider: Optional[str] = "openai"
+    model: Optional[str] = "gpt-4"
+
+class RepositoryCreate(SQLModelBase):
+    name: str
+    platform: str = Field(..., regex="^(github|gitlab)$")
+    api_token: str
+    custom_prompt: Optional[str] = None
+    settings: Optional[RepositorySettings] = Field(default_factory=lambda: RepositorySettings())
+
+class RepositoryResponse(SQLModelBase):
+    id: int
+    owner_id: int
+    name: str
+    platform: str
+    custom_prompt: Optional[str]
+    settings: RepositorySettings
+    created_at: datetime
+    
+    class Config:
+        orm_mode = True
+        property_getters = {"settings": lambda obj: obj.get_settings()}

backend/app/dashboard/route.py

@@ -0,0 +1,20 @@
+from sqlmodel import SQLModel, Field, Relationship
+from typing import Optional, List
+from datetime import datetime
+from ..user.model import User
+from ..repository.model import Repository
+
+class Review(SQLModel, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True, index=True)
+    repository_id: int = Field(foreign_key="repository.id")
+    reviewer_id: Optional[int] = Field(default=None, foreign_key="user.id")
+    pr_or_mr_id: str
+    title: str
+    summary: Optional[str] = Field(default=None)
+    status: str = Field(default="pending")
+    created_at: datetime = Field(default_factory=datetime.utcnow)
+    updated_at: datetime = Field(default_factory=datetime.utcnow)
+    
+    repository: Repository = Relationship(back_populates="reviews")
+    reviewer: Optional[User] = Relationship(back_populates="reviews")
+    comments: List["ReviewComment"] = Relationship(back_populates="review")

backend/app/dashboard/schema.py

@@ -0,0 +1,15 @@
+from fastapi import APIRouter, Depends
+from sqlmodel import Session
+from ..dependencies import get_session
+from .model import Review
+from .schema import ReviewCreate, ReviewResponse
+
+router = APIRouter()
+
+@router.post("/", response_model=ReviewResponse)
+def create_review(review: ReviewCreate, session: Session = Depends(get_session)):
+    db_review = Review.from_orm(review)
+    session.add(db_review)
+    session.commit()
+    session.refresh(db_review)
+    return db_review

backend/app/dependencies.py

@@ -0,0 +1,24 @@
+from typing import Optional, List
+from datetime import datetime
+from sqlmodel import SQLModel as SQLModelBase
+from ..review_comment.schema import ReviewCommentResponse
+
+class ReviewCreate(SQLModelBase):
+    repository_id: int
+    pr_or_mr_id: str
+    title: str
+
+class ReviewResponse(SQLModelBase):
+    id: int
+    repository_id: int
+    reviewer_id: Optional[int]
+    pr_or_mr_id: str
+    title: str
+    summary: Optional[str]
+    status: str
+    created_at: datetime
+    updated_at: datetime
+    comments: List[ReviewCommentResponse] = []
+    
+    class Config:
+        orm_mode = True

backend/app/repository/__init__.py

@@ -0,0 +1,14 @@
+from sqlmodel import SQLModel, Field, Relationship
+from typing import Optional
+from datetime import datetime
+from ..review.model import Review
+
+class ReviewComment(SQLModel, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True, index=True)
+    review_id: int = Field(foreign_key="review.id")
+    file_path: str
+    line_number: int
+    comment: str
+    created_at: datetime = Field(default_factory=datetime.utcnow)
+    
+    review: Review = Relationship(back_populates="comments")

backend/app/repository/model.py

@@ -0,0 +1,15 @@
+from fastapi import APIRouter, Depends
+from sqlmodel import Session
+from ..dependencies import get_session
+from .model import ReviewComment
+from .schema import ReviewCommentCreate, ReviewCommentResponse
+
+router = APIRouter()
+
+@router.post("/", response_model=ReviewCommentResponse)
+def create_review_comment(comment: ReviewCommentCreate, session: Session = Depends(get_session)):
+    db_comment = ReviewComment.from_orm(comment)
+    session.add(db_comment)
+    session.commit()
+    session.refresh(db_comment)
+    return db_comment

backend/app/repository/route.py

@@ -0,0 +1,19 @@
+from typing import Optional
+from datetime import datetime
+from sqlmodel import SQLModel as SQLModelBase
+
+class ReviewCommentCreate(SQLModelBase):
+    file_path: str
+    line_number: int
+    comment: str
+
+class ReviewCommentResponse(SQLModelBase):
+    id: int
+    review_id: int
+    file_path: str
+    line_number: int
+    comment: str
+    created_at: datetime
+    
+    class Config:
+        orm_mode = True

backend/app/repository/schema.py

@@ -0,0 +1,4 @@
+# app/user/__init__.py
+from .model import User
+from .schema import UserCreate, UserResponse
+from .route import router as user_router

backend/app/review/__init__.py

@@ -0,0 +1,16 @@
+from sqlmodel import SQLModel, Field, Relationship
+from typing import Optional, List
+from datetime import datetime
+
+class User(SQLModel, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True, index=True)
+    email: str = Field(unique=True, index=True)
+    username: str = Field(unique=True)
+    github_access_token: Optional[str] = Field(default=None)
+    github_refresh_token: Optional[str] = Field(default=None)
+    gitlab_access_token: Optional[str] = Field(default=None)
+    gitlab_refresh_token: Optional[str] = Field(default=None)
+    created_at: datetime = Field(default_factory=datetime.utcnow)
+    
+    repositories: List["Repository"] = Relationship(back_populates="owner")
+    reviews: List["Review"] = Relationship(back_populates="reviewer")

backend/app/review/model.py

@@ -0,0 +1,15 @@
+from fastapi import APIRouter, Depends
+from sqlmodel import Session, select
+from ..dependencies import get_session
+from .model import User
+from .schema import UserCreate, UserResponse
+
+router = APIRouter()
+
+@router.post("/", response_model=UserResponse)
+def create_user(user: UserCreate, session: Session = Depends(get_session)):
+    db_user = User.from_orm(user)
+    session.add(db_user)
+    session.commit()
+    session.refresh(db_user)
+    return db_user

backend/app/review/route.py

@@ -0,0 +1,17 @@
+from pydantic import EmailStr
+from typing import Optional
+from datetime import datetime
+from sqlmodel import SQLModel as SQLModelBase
+
+class UserCreate(SQLModelBase):
+    email: EmailStr
+    username: str
+
+class UserResponse(SQLModelBase):
+    id: int
+    email: EmailStr
+    username: str
+    created_at: datetime
+    
+    class Config:
+        orm_mode = True

backend/app/review/schema.py

@@ -11,6 +11,7 @@ dependencies = [
     "pygithub>=2.6.1",
     "python-dotenv>=1.0.1",
     "python-gitlab>=5.6.0",
+    "sqlmodel>=0.0.23",
     "tiktoken>=0.9.0",
     "uvicorn>=0.34.0",
 ]