From 7b4aab71051f6c80dbb899b9c3458c49507eea6e Mon Sep 17 00:00:00 2001
From: alstr <alstr@users.noreply.github.com>
Date: Fri, 22 Nov 2024 15:02:37 +0000
Subject: [PATCH] Add User-Agent header

Fixes #248
---
 GitHubClient.py | 9 ++++++---
 TodoParser.py   | 8 ++++++--
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/GitHubClient.py b/GitHubClient.py
index f543601..7ea5b87 100644
--- a/GitHubClient.py
+++ b/GitHubClient.py
@@ -25,11 +25,13 @@ def __init__(self):
         self.issue_headers = {
             'Content-Type': 'application/json',
             'Authorization': f'token {self.token}',
-            'X-GitHub-Api-Version': '2022-11-28'
+            'X-GitHub-Api-Version': '2022-11-28',
+            'User-Agent': 'TODOToIssue/5.1.5'
         }
         self.graphql_headers = {
             'Authorization': f'Bearer {os.getenv("INPUT_PROJECTS_SECRET", "")}',
-            'Accept': 'application/vnd.github.v4+json'
+            'Accept': 'application/vnd.github.v4+json',
+            'User-Agent': 'TODOToIssue/5.1.5'
         }
         auto_p = os.getenv('INPUT_AUTO_P', 'true') == 'true'
         self.line_break = '\n\n' if auto_p else '\n'
@@ -79,7 +81,8 @@ def get_last_diff(self):
         diff_headers = {
             'Accept': 'application/vnd.github.v3.diff',
             'Authorization': f'token {self.token}',
-            'X-GitHub-Api-Version': '2022-11-28'
+            'X-GitHub-Api-Version': '2022-11-28',
+            'User-Agent': 'TODOToIssue/5.1.5'
         }
         diff_request = requests.get(url=diff_url, headers=diff_headers)
         if diff_request.status_code == 200:
diff --git a/TodoParser.py b/TodoParser.py
index eefd1bd..9f3f94b 100644
--- a/TodoParser.py
+++ b/TodoParser.py
@@ -8,6 +8,10 @@
 from urllib.parse import urlparse
 import itertools
 
+headers = {
+    'User-Agent': 'TODOToIssue/5.1.5'
+}
+
 class TodoParser(object):
     """Parser for extracting information from a given diff file."""
     FILE_HUNK_PATTERN = r'(?<=diff)(.*?)(?=diff\s--git\s)'
@@ -54,7 +58,7 @@ def __init__(self, options=dict()):
         if os.getenv('INPUT_NO_STANDARD', 'false') != 'true':
             # Load the languages data for ascertaining file types.
             languages_url = 'https://raw.githubusercontent.com/github/linguist/master/lib/linguist/languages.yml'
-            languages_request = requests.get(url=languages_url)
+            languages_request = requests.get(url=languages_url, headers=headers)
             if languages_request.status_code == 200:
                 languages_data = languages_request.text
                 yaml = YAML(typ='safe')
@@ -64,7 +68,7 @@ def __init__(self, options=dict()):
 
             # Load the comment syntax data for identifying comments.
             syntax_url = 'https://raw.githubusercontent.com/alstr/todo-to-issue-action/master/syntax.json'
-            syntax_request = requests.get(url=syntax_url)
+            syntax_request = requests.get(url=syntax_url, headers=headers)
             if syntax_request.status_code == 200:
                 self.syntax_dict = syntax_request.json()
             else: