Provide Github tokens to GraphQL

This will support a change in the upstream githubgql library to remove
hardcoded token retrieval.

Author: mbestavros

src/assign

@@ -145,9 +145,10 @@ def update_assignees(number_to_update=None):
     "Updates assignees on all repo PRs by default, or a specified PR if input."
 
     org, repo = os.environ["GITHUB_REPOSITORY"].split("/")
+    token = os.environ.get('BOT_TOKEN', None)
 
     try:
-        pr_data = githubgql.graphql(QUERY, org=org, repo=repo, cursor=QUERY_CURSORS)
+        pr_data = githubgql.graphql(QUERY, token=token, org=org, repo=repo, cursor=QUERY_CURSORS)
     except githubgql.TokenError as e:
         print(e.error)
         sys.exit(0)
@@ -190,6 +191,7 @@ def update_assignees(number_to_update=None):
                     }
                 }
                 """,
+                token=token,
                 input=input
             )
 
@@ -207,6 +209,7 @@ def update_assignees(number_to_update=None):
                     }
                 }
                 """,
+                token=token,
                 input=input
             )
 

src/copy-labels-linked

@@ -88,11 +88,13 @@ def get_related_issues(body, commits):
 
 def copy_labels_linked(id):
     owner, repo = os.environ["GITHUB_REPOSITORY"].split("/")
+    token = os.environ.get('BOT_TOKEN', None)
 
     # Get PR data and open issues in the repo.
     try:
         result = githubgql.graphql(
             QUERY,
+            token=token,
             id=id,
             org=owner,
             repo=repo,
@@ -145,6 +147,7 @@ def copy_labels_linked(id):
                 }
             }
             """,
+            token=token,
             input={
                 "labelableId": result['node']['id'],
                 "labelIds": unset_labels_ids

src/merge

@@ -33,10 +33,11 @@ mutation($input: MergePullRequestInput!) {
 
 def merge(number_to_merge=None):
     owner, repo = os.environ["GITHUB_REPOSITORY"].split("/")
+    token = os.environ.get('BOT_TOKEN', None)
     cursors = {"cursor": ["repository", "pullRequests"]}
 
     try:
-        result = githubgql.graphql(MERGEABLE_QUERY, accept="application/vnd.github.merge-info-preview+json", cursors=cursors, owner=owner, repo=repo)
+        result = githubgql.graphql(MERGEABLE_QUERY, token=token, accept="application/vnd.github.merge-info-preview+json", cursors=cursors, owner=owner, repo=repo)
     except githubgql.TokenError as e:
         print(e.error)
         sys.exit(0)
@@ -64,7 +65,7 @@ def merge(number_to_merge=None):
 
         # Attempt to merge the PR if ready.
         if merged_indicator == "✓":
-            githubgql.graphql(MERGE_MUTATION, input={
+            githubgql.graphql(MERGE_MUTATION, token=token, input={
                     "pullRequestId": id,
                     "mergeMethod": "REBASE"
                 }

src/request

@@ -110,7 +110,7 @@ def iterusers(nested_dictionary):
         elif type(value) is dict:
             yield from iterusers(value)
 
-def auto_request(pr, org):
+def auto_request(pr, org, token):
     team_name = sys.argv[1]
     num_to_request = sys.argv[2]
 
@@ -132,7 +132,7 @@ def auto_request(pr, org):
 
     # Query the PR for all relevant users and status info.
     try:
-        pr_data = githubgql.graphql(QUERY_PR, pr=pr, team=team_name, cursors=QUERY_PR_CURSORS)
+        pr_data = githubgql.graphql(QUERY_PR, token=token, pr=pr, team=team_name, cursors=QUERY_PR_CURSORS)
     except githubgql.TokenError as e:
         print(e.error)
         sys.exit(0)
@@ -175,7 +175,7 @@ def auto_request(pr, org):
 
     if len(to_request) > 0:
         # Make the request to Github.
-        githubgql.graphql(MUTATION_REQUEST_REVIEW, input={
+        githubgql.graphql(MUTATION_REQUEST_REVIEW, token=token, input={
             "pullRequestId": pr,
             "userIds": to_request,
             "teamIds": [],
@@ -195,8 +195,9 @@ def main():
     # Extract relevant data from event/environment.
     pr = event['pull_request']['node_id']
     org = event['organization']['node_id']
+    token = os.environ.get('BOT_TOKEN', None)
 
-    auto_request(pr, org)
+    auto_request(pr, org, token)
 
 if __name__ == "__main__":
     main()