Explorar el Código

[O] Allow github token in accept_upstream.py

Azalea (on HyDEV-Daisy) hace 2 años
padre
commit
b4494cab21
Se han modificado 1 ficheros con 7 adiciones y 2 borrados
  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_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():
     # Get commit SHA
@@ -42,11 +47,11 @@ if __name__ == '__main__':
     print(f'Accepting pull request {pr}...')
 
     # 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']
 
     # 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']
 
     # Create commit message