Browse Source

[O] Allow github token in accept_upstream.py

Azalea (on HyDEV-Daisy) 2 years ago
parent
commit
b4494cab21
1 changed files with 7 additions and 2 deletions
  1. 7 2
      tools/accept_upstream.py

+ 7 - 2
tools/accept_upstream.py

@@ -12,6 +12,11 @@ upstream = 'dylanaraps/neofetch'
 my_fork = 'hykilpikonna/hyfetch'
 my_fork = 'hykilpikonna/hyfetch'
 my_base = 'master'
 my_base = 'master'
 
 
+http = requests.Session()
+if 'GH_TOKEN' in os.environ:
+    print('Token loaded')
+    http.headers['Authorization'] = f'token {os.environ["GH_TOKEN"]}'
+
 
 
 def copy_comment():
 def copy_comment():
     # Get commit SHA
     # Get commit SHA
@@ -42,11 +47,11 @@ if __name__ == '__main__':
     print(f'Accepting pull request {pr}...')
     print(f'Accepting pull request {pr}...')
 
 
     # Fetch original pr's information
     # Fetch original pr's information
-    info = requests.get(f'https://api.github.com/repos/{upstream}/pulls/{pr}').json()
+    info = http.get(f'https://api.github.com/repos/{upstream}/pulls/{pr}').json()
     user = info['user']['login']
     user = info['user']['login']
 
 
     # Fetch commit information
     # Fetch commit information
-    commits = requests.get(f'https://api.github.com/repos/{upstream}/pulls/{pr}/commits').json()
+    commits = http.get(f'https://api.github.com/repos/{upstream}/pulls/{pr}/commits').json()
     author = commits[0]['commit']['author']
     author = commits[0]['commit']['author']
 
 
     # Create commit message
     # Create commit message