Refactor file retrieval to use GitHub library instead of requests

usegalaxy-eu · Nov 6, 2024 · 11402ce · 11402ce
1 parent 3d8ba73
commit 11402ce
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/github_run.py b/github_run.py
@@ -4,7 +4,6 @@
 import sys
 
 import github
-import requests
 import yaml
 
 from lib.galaxy_social import galaxy_social
@@ -64,12 +63,14 @@ def get_files(self):
         for file in self.pr.get_files():
             file_path = file.filename
             if file_path.endswith(".md"):
-                response = requests.get(file.raw_url)
-                if response.status_code == 200:
-                    os.makedirs(os.path.dirname(file_path), exist_ok=True)
-                    with open(file_path, "w") as f:
-                        f.write(response.text)
-                    files_to_process.append(file_path)
+                head_branch = self.pr.head
+                pr_content = head_branch.repo.get_contents(
+                    file_path, ref=head_branch.ref
+                ).decoded_content.decode()
+                os.makedirs(os.path.dirname(file_path), exist_ok=True)
+                with open(file_path, "w") as f:
+                    f.write(pr_content)
+                files_to_process.append(file_path)
 
         return files_to_process