Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Seperate out issue/pr logic into their own functions and add !gh that will get

Browse source 
info on either an issue or a pull request (github.py)
This commit is contained in:
jesopo 2019-01-10 13:32:19 +00:00
parent 8da6036894
commit 2d346e73bc
1 changed files with 45 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
modules/github.py
View file

@ -52,33 +52,30 @@ class Module(ModuleManager.BaseModule):
raise utils.EventError("Issue number must be a number") raise utils.EventError("Issue number must be a number")
return username, repository, number return username, repository, number
@utils.hook("received.command.ghissue", min_args=1) def _gh_issue(self, event, page):
def github_issue(self, event):
username, repository, number = self._parse_ref(
event["target"], event["args_split"][0])
page = utils.http.request(
API_ISSUE_URL % (username, repository, number),
json=True)
if page and page.code == 200:
labels = [label["name"] for label in page.data["labels"]] labels = [label["name"] for label in page.data["labels"]]
url = self._short_url(page.data["html_url"]) url = self._short_url(page.data["html_url"])
event["stdout"].write("(%s/%s issue#%s, %s) %s [%s] %s" % ( event["stdout"].write("(%s/%s issue#%s, %s) %s [%s] %s" % (
username, repository, number, page.data["state"], username, repository, number, page.data["state"],
page.data["title"], ", ".join(labels), url)) page.data["title"], ", ".join(labels), url))
else: def _gh_get_issue(self, username, repository, number):
event["stderr"].write("Could not find issue") return utils.http.request(
API_ISSUE_URL % (username, repository, number),
json=True)
@utils.hook("received.command.ghpull", min_args=1) @utils.hook("received.command.ghissue", min_args=1)
def github_pull(self, event): def github_issue(self, event):
username, repository, number = self._parse_ref( username, repository, number = self._parse_ref(
event["target"], event["args_split"][0]) event["target"], event["args_split"][0])
page = utils.http.request( page = self._gh_get_issue(username, repository, number)
API_PULL_URL % (username, repository, number),
json=True)
if page and page.code == 200: if page and page.code == 200:
self._gh_issue(event, page)
else:
event["stderr"].write("Could not find issue")
def _gh_pull(self, event, page):
repo_from = page.data["head"]["label"] repo_from = page.data["head"]["label"]
repo_to = page.data["base"]["label"] repo_to = page.data["base"]["label"]
added = self._added(page.data["additions"]) added = self._added(page.data["additions"])
@ -89,9 +86,35 @@ class Module(ModuleManager.BaseModule):
"(%s/%s pull#%s, %s) [%s/%s] %s→%s - %s %s" % ( "(%s/%s pull#%s, %s) [%s/%s] %s→%s - %s %s" % (
username, repository, number, page.data["state"], username, repository, number, page.data["state"],
added, removed, repo_from, repo_to, page.data["title"], url)) added, removed, repo_from, repo_to, page.data["title"], url))
def _gh_get_pull(self, username, repository, number):
return utils.http.request(
API_PULL_URL % (username, repository, number),
json=True)
@utils.hook("received.command.ghpull", min_args=1)
def github_pull(self, event):
username, repository, number = self._parse_ref(
event["target"], event["args_split"][0])
page = self._gh_get_pull(username, repository, number)
if page and page.code == 200:
self._gh_pull(event, page)
else: else:
event["stderr"].write("Could not find pull request") event["stderr"].write("Could not find pull request")
@utils.hook("received.command.gh", min_args=1)
def github(self, event):
username, repository, number = self._parse_ref(
event["target"], event["args_split"][0])
page = self._gh_get_issue(username, repository, number)
if page and page.code == 200:
if "pull_request" in page.data:
pull = self._gh_get_pull(username, repository, number)
self._gh_pull(event, pull)
else:
self._gh_issue(event, page)
else:
event["stderr"].write("Issue/PR not found")
@utils.hook("api.post.github") @utils.hook("api.post.github")
def webhook(self, event): def webhook(self, event):
payload = event["data"].decode("utf8") payload = event["data"].decode("utf8")