Skip to content

Commit 067b237

Browse files
committed
Add github token
1 parent 5414163 commit 067b237

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

.github/workflows/test.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,8 @@ jobs:
126126
environments: ${{ matrix.environment }}
127127
- name: Download data
128128
if: needs.setup.outputs.code_change == 'true'
129+
env:
130+
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
129131
run: |
130132
pixi run -e ${{ matrix.environment }} python examples/download.py
131133
- name: Test Unit

examples/download.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import os
12
from shutil import rmtree
23

34
try:
@@ -7,6 +8,12 @@
78
raise ImportError("requests and platformdirs are needed to download data") from None
89

910

11+
if os.environ.get("GITHUB_TOKEN"):
12+
HEADERS = {"Authorization": f"token {os.environ['GITHUB_TOKEN']}"}
13+
else:
14+
HEADERS = None
15+
16+
1017
def download_map(dataset):
1118
if dataset not in ("naturalearth_lowres", "naturalearth_cities"):
1219
raise ValueError(
@@ -18,7 +25,7 @@ def download_map(dataset):
1825
if local_dir.exists():
1926
return local_dir
2027

21-
response = requests.get(url)
28+
response = requests.get(url, headers=HEADERS)
2229
if response.ok:
2330
files = response.json()
2431
else:
@@ -32,7 +39,7 @@ def download_map(dataset):
3239
for file in files:
3340
file_url = file["download_url"]
3441
file_name = file["name"]
35-
file_response = requests.get(file_url)
42+
file_response = requests.get(file_url, headers=HEADERS)
3643
if not file_response.ok:
3744
rmtree(local_dir)
3845
raise ValueError(f"Failed to download file: {file_name}, \n{file_response.text}")

0 commit comments

Comments
 (0)