Support adding an /etc/host entry for the data proxy, if asked. · HpcDataLab/docker-python@062f975 · GitHub

HpcDataLab / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 0

• Code
• Pull requests
• Actions
• Projects
• Security and quality
• Insights

Additional navigation options

Commit 062f975

Browse filesBrowse the repository at this point in the historyBrowse files

Seb Boving

committed

Support adding an /etc/host entry for the data proxy, if asked.

1 parent a364dfe commit 062f975
Copy full SHA for 062f975

1 file changed

+15Lines changed: 15 additions & 0 deletions

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• patches
  • sitecustomize.py

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎patches/sitecustomize.py‎

Copy file name to clipboardExpand all lines: patches/sitecustomize.py

+15Lines changed: 15 additions & 0 deletions

Original file line number	Diff line number	Diff line change
@@ -5,12 +5,27 @@
5	5	import torch
6	6	import os
7	7
	8	+ _HOST_FILE = "/etc/hosts"
	9	+
8	10	kaggle_proxy_token = os.getenv("KAGGLE_DATA_PROXY_TOKEN")
9	11	if kaggle_proxy_token:
10	12	from google.auth import credentials
11	13	from google.cloud import bigquery
12	14	from google.cloud.bigquery._http import Connection
13	15
	16	+ host_entry = os.getenv("KAGGLE_DATA_PROXY_HOST_ENTRY")
	17	+ if host_entry:
	18	+ host_entry = host_entry.strip()
	19	+ with open(_HOST_FILE, "r") as host_file:
	20	+ for line in host_file:
	21	+ if line.strip() == host_entry:
	22	+ break
	23	+ else:
	24	+ with open(_HOST_FILE, "a") as host_file_append:
	25	+ host_file_append.write("\n" # Just in case it wasn't newline terminated.
	26	+ + host_entry
	27	+ + "\n")
	28	+
14	29	Connection.API_BASE_URL = os.getenv("KAGGLE_DATA_PROXY_URL")
15	30	Connection._EXTRA_HEADERS["X-KAGGLE-PROXY-DATA"] = kaggle_proxy_token
16	31

0 commit comments

Comments

0 (0)