import requests


class GraphQL:

    def __init__(self, token):
        super(GraphQL, self).__init__()
        self.token = token

    def headers(self):
        return {"Authorization": f"token {self.token}"}

    def run_mutation(self, mutation):
        request = requests.post('https://api.github.com/graphql',
                                json={'query': mutation},
                                headers=self.headers())
        if request.status_code == 200:
            return request.json()
        else:
            raise Exception(
                "Query failed to run by returning code of {}. {}".format(
                    request.status_code, mutation))

    def mark_ready_for_review(self, pull_request_id):
        mutation = """
           mutation {
              markPullRequestReadyForReview(input:{pullRequestId: %s}) {
                  pullRequest{id, isDraft}
              }
           }
        """ % pull_request_id

        result = self.run_mutation(mutation)
        return result