Add MergePullRequest class for merge_pull_request function with JSON definition

This commit is contained in:
2024-08-19 15:48:17 -05:00
parent b67ab7a847
commit 28b793a2ef
@@ -0,0 +1,82 @@
import requests
import logging
class MergePullRequest:
def __init__(self, base_url, token, repo):
self.base_url = base_url
self.headers = {
"Authorization": f"token {token}",
"Accept": "application/vnd.github.v3+json"
}
self.repo = repo
# Set up logging
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.INFO)
# Create a file handler
file_handler = logging.FileHandler('merge_pull_request.log')
file_handler.setLevel(logging.INFO)
# Create a console handler
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.INFO)
# Create a formatting for the logs
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levellevel)s - %(message)s')
file_handler.setFormatter(formatter)
console_handler.setFormatter(formatter)
# Add the handlers to the logger
self.logger.addHandler(file_handler)
self.logger.addHandler(console_handler)
def __call__(self, pull_number, commit_title, commit_message, merge_method):
self.logger.info(f"Merging pull request: {pull_number}")
url = f"{self.base_url}/repos/{self.repo}/pulls/{pull_number}/merge"
data = {
"commit_title": commit_title,
"commit_message": commit_message,
"merge_method": merge_method
}
response =.requests.put(url, headers=self.headers, json=data)
if response.status_code == 200:
success_message = f"Pull request {pull_number} merged successfully"
self.logger.info(success_message)
return success_message
else:
error_message = f"Error merging pull request: {response.status_code}\nResponse: {response.text}"
self.logger.error(error_message)
return error_message
# JSON definition for the merge_pull_request function
merge_pull_request_definition = {
"name": "merge_pull_request",
"description": "Merge a pull request",
"parameters": {
"type": "object",
"properties": {
"pull_number": {
"type": "integer",
"description": "The number of the pull request"
},
"commit_title": {
"type": "string",
"description": "Title for the automatic commit message",
"default": "Merge pull request"
},
"commit_message": {
"type": "string",
"description": "Extra detail to append to automatic commit message",
"default": ""
},
"merge_method": {
"type": "string",
"description": "Merge method to use",
"enum": ["merge", "squash", "rebase"],
"default": "merge"
}
},
"required": ["pull_number"]
}
}