Skip to content

Commit 382c209

Browse files
committed
Add User Agent headers to hgmo requests
1 parent fc19091 commit 382c209

File tree

2 files changed

+16
-4
lines changed

2 files changed

+16
-4
lines changed

backend/code_coverage_backend/hgmo.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ def hgmo_revision_details(repository, changeset):
2929
return __hgmo[key]
3030

3131
url = HGMO_REVISION_URL.format(repository=repository, revision=changeset)
32-
resp = requests.get(url)
32+
resp = requests.get(url, headers={"User-Agent": "code-coverage-backend"})
3333
resp.raise_for_status()
3434
assert "changesets" in resp.json(), "Missing changesets"
3535
data = resp.json()["changesets"][-1]
@@ -52,7 +52,11 @@ def hgmo_pushes(repository, min_push_id, nb_pages, chunk_size=8):
5252
params["endID"] = min_push_id + chunk_size
5353

5454
for page in range(nb_pages):
55-
r = requests.get(HGMO_PUSHES_URL.format(repository=repository), params=params)
55+
r = requests.get(
56+
HGMO_PUSHES_URL.format(repository=repository),
57+
params=params,
58+
headers={"User-Agent": "code-coverage-backend"},
59+
)
5660
data = r.json()
5761

5862
# Sort pushes to go from oldest to newest

bot/code_coverage_bot/hgmo.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,14 +78,22 @@ def get_pushes(
7878
if changeset is not None:
7979
params["changeset"] = changeset
8080

81-
r = requests.get("{}/json-pushes".format(self.server_address), params=params)
81+
r = requests.get(
82+
"{}/json-pushes".format(self.server_address),
83+
params=params,
84+
headers={"User-Agent": "code-coverage-bot"},
85+
)
8286

8387
r.raise_for_status()
8488
return r.json()
8589

8690
def get_automation_relevance_changesets(self, changeset):
8791
r = requests.get(
88-
"{}/json-automationrelevance/{}".format(self.server_address, changeset)
92+
"{}/json-automationrelevance/{}".format(
93+
self.server_address,
94+
changeset,
95+
),
96+
headers={"User-Agent": "code-coverage-bot"},
8997
)
9098
r.raise_for_status()
9199
return r.json()["changesets"]

0 commit comments

Comments
 (0)