| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
1 file changed
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,18 +1,20 @@ | |||
| 1 | 1 | # Monkey patches BigQuery client creation to use proxy. | |
| 2 | 2 | import os | |
| 3 | 3 | ||
| 4 | - kaggle_proxy_token = os.getenv("KAGGLE_PROXY_TOKEN") | ||
| 4 | + kaggle_proxy_token = os.getenv("KAGGLE_DATA_PROXY_TOKEN") | ||
| 5 | 5 | if kaggle_proxy_token: | |
| 6 | 6 | from google.auth import credentials | |
| 7 | 7 | from google.cloud import bigquery | |
| 8 | 8 | from google.cloud.bigquery._http import Connection | |
| 9 | 9 | ||
| 10 | - Connection.API_BASE_URL = os.getenv("KAGGLE_PROXY_URL") | ||
| 10 | + Connection.API_BASE_URL = os.getenv("KAGGLE_DATA_PROXY_URL") | ||
| 11 | 11 | Connection._EXTRA_HEADERS["X-KAGGLE-PROXY-DATA"] = kaggle_proxy_token | |
| 12 | 12 | ||
| 13 | 13 | bq_client = bigquery.Client | |
| 14 | 14 | bigquery.Client = lambda *args, **kwargs: bq_client( | |
| 15 | 15 | *args, | |
| 16 | 16 | credentials=credentials.AnonymousCredentials(), | |
| 17 | - project=os.getenv("KAGGLE_PROXY_PROJECT"), | ||
| 18 | - **kwargs) | ||
| 17 | + project=os.getenv("KAGGLE_DATA_PROXY_PROJECT"), | ||
| 18 | + **kwargs) | ||
| 19 | + | ||
| 20 | + credentials.AnonymousCredentials.refresh = lambda *args: None | ||
| Back | FazBrowse Home | New Git URL |
0 commit comments