datastore.py

from models import User, Review, ReviewStatus

class DataStore:
    def __init__(self, db):
        self.db = db

    def get_user_by_id(self, id: str, channel_id: str):
        return self.db.query(User).filter(User.id == id, User.channel_id == channel_id).first()

    def create_user(self, name: str, slack_id: str, channel_id: str, is_reviewer: bool = False):
        new_user = User(name=name, slack_id=slack_id, channel_id=channel_id, is_reviewer=is_reviewer)
        self.db.add(new_user)
        self.db.commit()
        self.db.refresh(new_user)
        return new_user

    def get_all_users_for_channel(self, channel_id: str):
        return self.db.query(User).filter(User.channel_id == channel_id).all()

    def get_all_reviewers_for_channel(self, channel_id: str):
        return self.db.query(User).filter(User.is_reviewer, User.channel_id == channel_id).all()

    def update_user(self, user_id: int, channel_id: str, **kwargs):
        user = self.db.query(User).filter(User.id == user_id, User.channel_id == channel_id).first()
        if user:
            for key, value in kwargs.items():
                setattr(user, key, value)
            self.db.commit()
            self.db.refresh(user)
        return user

    def get_user_by_slack_id(self, slack_id: str, channel_id: str):
        return self.db.query(User).filter(User.slack_id == slack_id, User.channel_id == channel_id).first()

    def create_review(self, user_id: str, url: str, reviewer_id: str, status: str = ReviewStatus.IN_REVIEW):
        new_review = Review(user_id=user_id, url=url, reviewer_id=reviewer_id, status=status)
        self.db.add(new_review)
        self.db.commit()
        self.db.refresh(new_review)
        return new_review

    # TODO: Update these to use channel ID as well.
    def get_reviews_submitted_by(self, user_id: str):
        # Retrieve reviews submitted by a user
        return self.db.query(Review).filter(Review.user_id == user_id).all()

    def get_reviews_assigned_to(self, reviewer_id: str):
        # Retrieve reviews assigned to a reviewer
        return self.db.query(Review).filter(Review.reviewer_id == reviewer_id).all()

    def get_review(self, review_id: int):
        # Retrieve a review by ID
        return self.db.query(Review).filter(Review.id == review_id).first()

    def update_review(self, review_id: int, url: str, reviewer_id: str, status: str):
        review = self.db.query(Review).filter(Review.id == review_id).first()
        if review:
            review.url = url
            review.reviewer_id = reviewer_id
            review.status = status
            self.db.commit()
            self.db.refresh(review)
        return review

    def delete_review(self, review_id):
        review = self.db.query(Review).filter_by(id=review_id).first()
        if not review:
            return False

        try:
            self.db.delete(review)
            self.db.commit()
            return True
        except Exception as e:
            self.db.rollback()
            return False