|
19 | 19 | DEFAULT_CREATION_DATE = datetime(2020, 1, 13, 23, 58, 59) |
20 | 20 |
|
21 | 21 | RSC_SERVER_URL = os.getenv("CONNECT_SERVER") |
22 | | -_RSC_API_KEYS = None |
23 | 22 |
|
24 | 23 | DATABRICKS_VOLUME = "/Volumes/workshops/my-board/my-volume/test" |
25 | 24 |
|
@@ -75,24 +74,17 @@ def wrapper(*args, **kwargs): |
75 | 74 | return wrapper |
76 | 75 |
|
77 | 76 |
|
78 | | -def set_rsc_api_keys(keys): |
79 | | - global _RSC_API_KEYS |
80 | | - _RSC_API_KEYS = keys |
81 | | - |
82 | | - |
83 | | -def rsc_from_key(name): |
| 77 | +def rsc_from_key(name, api_keys): |
84 | 78 | from pins.rsconnect.api import RsConnectApi |
85 | 79 |
|
86 | | - if _RSC_API_KEYS is None: |
87 | | - raise RuntimeError("RSC API keys not initialized. This should be set by the pytest fixture.") |
88 | | - api_key = _RSC_API_KEYS[name] |
| 80 | + api_key = api_keys[name] |
89 | 81 | return RsConnectApi(RSC_SERVER_URL, api_key) |
90 | 82 |
|
91 | 83 |
|
92 | | -def rsc_fs_from_key(name): |
| 84 | +def rsc_fs_from_key(name, api_keys): |
93 | 85 | from pins.rsconnect.fs import RsConnectFs |
94 | 86 |
|
95 | | - rsc = rsc_from_key(name) |
| 87 | + rsc = rsc_from_key(name, api_keys) |
96 | 88 |
|
97 | 89 | return RsConnectFs(rsc) |
98 | 90 |
|
@@ -211,14 +203,15 @@ class RscBoardBuilder(BoardBuilder): |
211 | 203 |
|
212 | 204 | # TODO: could loop back once initializing all boards is clear |
213 | 205 |
|
214 | | - def __init__(self, fs_name, path=None, *args, **kwargs): |
| 206 | + def __init__(self, fs_name, path=None, api_keys=None, *args, **kwargs): |
215 | 207 | self.fs_name = fs_name |
216 | 208 | self.path = None |
| 209 | + self.api_keys = api_keys |
217 | 210 |
|
218 | 211 | def create_tmp_board(self, src_board=None, versioned=True): |
219 | 212 | from pins.rsconnect.fs import PinBundleManifest # noqa |
220 | 213 |
|
221 | | - board = BoardRsConnect("", rsc_fs_from_key("derek"), versioned=versioned) |
| 214 | + board = BoardRsConnect("", rsc_fs_from_key("derek", self.api_keys), versioned=versioned) |
222 | 215 | if src_board is None: |
223 | 216 | return board |
224 | 217 |
|
|
0 commit comments