utils.py

import json
from requests import Session

from exceptions import NoGitHubTokenException
from config import get_token, get_credentials, get_proxy, APP_NAME


class PullRequest:
    def __init__(self, event=None):
        self.event = event
        self._session = Session()
        self.github_proxy = get_proxy()
        try:
            self._session.headers.update({"Authorization": f"token {get_token()}"})
        except NoGitHubTokenException:
            self._session.auth = get_credentials()
        self._session.headers.update({"User-Agent": APP_NAME})
        if event is not None:
            self.issue_url = event['pull_request']['issue_url']

    @property
    def labels(self):
        return self.request_labels_json()

    def request_labels_json(self):
        response = self._session.get(self.label_url)
        if response.status_code >= 300:
            print("Got a non-2xx status: ", response.url, response.headers,
                  response.content)
        return response.json()

    @property
    def label_url(self):
        if self.github_proxy is not None:
           return "{}/labels".format(self.issue_url).replace("https://api.github.com/", self.github_proxy)
        return "{}/labels".format(self.issue_url)

    def compute_and_post_status(self, required_any, required_all, banned, status_text, status_target_url):
        return self.post_status(self.create_status_json(required_any, required_all, banned, status_text, status_target_url))

    def post_status(self, status_json):
        response = self._session.post(self.statuses_url, data=status_json)
        return response.status_code

    @property
    def statuses_url(self):
        if self.github_proxy is not None:
           return self.event['pull_request']['statuses_url'].replace("https://api.github.com/", self.github_proxy)
        return self.event['pull_request']['statuses_url']

    def create_status_json(self, required_any, required_all, banned, status_text, status_target_url):
        passes_label_requirements = self.validate_labels(required_any, required_all, banned)
        if passes_label_requirements:
            description = "Label requirements satisfied."
        else:
            description = status_text
        response_json = {
            "state": "success" if passes_label_requirements else "failure",
            "target_url": status_target_url,
            "description": description,
            "context": APP_NAME,
        }
        return json.dumps(response_json)

    def validate_labels(self, required_any, required_all, banned):
        try:
            labels_json = self.labels
            labels_list = [l['name'] for l in labels_json]
            if required_any is not None and not any(l in required_any for l in labels_list):
                return False
            if required_all is not None and any(l not in labels_list for l in required_all):
                return False
            if banned is not None and any(l in labels_list for l in banned):
                return False
            return True
        except TypeError:
            print('self.labels was of unexpected format for PR event {}: {}'.format(self.issue_url, labels_json))
            return False