open_pr_comment.py

from os import environ
import requests

pr = environ.get('PULL_REQUEST_ID')

def make_request(method, path, *args, **kwargs):
  url = f'https://api.github.com/repos/{environ["GITHUB_REPOSITORY"]}/{path}'
  kwargs['headers'] = {
    'Accept': 'application/vnd.github.v3+json',
    'Authorization': 'Bearer ' + environ['GITHUB_TOKEN'],
  }
  r = requests.request(method, url, *args, **kwargs)
  if not r.ok:
    print(r.status_code, r.text)
  r.raise_for_status()
  return r.json()

def get_pr_comments(author):
  comments = make_request('GET', f'issues/{pr}/comments')
  return [comment for comment in comments if comment['user']['login'] == author]

def edit_pr_comment(comment_id, new_body):
  return make_request('PATCH', f'issues/comments/{comment_id}', json={'body': new_body})

def create_pr_comment(body):
  return make_request('POST', f'issues/{pr}/comments', json={'body': body})

def create_or_edit_pr_comment(comment_body):
  existing_comments = get_pr_comments('github-actions[bot]')
  if existing_comments:
    edit_pr_comment(existing_comments[0]['id'], comment_body)
  else:
    create_pr_comment(comment_body)

def create_issue(title, body=''):
  return make_request('POST', 'issues', json={'title': title, 'body': body})