Skip to content

Commit 3243c72

Browse files
committed
feat(rsconnect): speed up pin search, and enable log printing
1 parent 2e3981c commit 3243c72

File tree

3 files changed

+14
-4
lines changed

3 files changed

+14
-4
lines changed

pins/boards.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -717,12 +717,13 @@ def pin_search(self, search=None, as_df=True):
717717

718718
paged_res = self.fs.api.misc_get_applications("content_type:pin", search=search)
719719
results = paged_res.results
720-
names = [f"{cont['owner_username']}/{cont['name']}" for cont in results]
721720

722721
res = []
723-
for pin_name in names:
722+
for content in results:
723+
pin_name = f"{content['owner_username']}/{content['name']}"
724+
version = str(content["bundle_id"])
724725
try:
725-
meta = self.pin_meta(pin_name)
726+
meta = self.pin_meta(pin_name, version)
726727
res.append(meta)
727728

728729
except RsConnectApiRequestError as e:

pins/config.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,3 +29,12 @@ def get_allow_pickle_read(flag):
2929
flag = bool(env_int)
3030

3131
return flag
32+
33+
34+
def _enable_logs():
35+
import logging
36+
37+
format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
38+
handlers = [logging.FileHandler("filename.log"), logging.StreamHandler()]
39+
40+
logging.basicConfig(level=logging.INFO, format=format, handlers=handlers)

pins/rsconnect/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,7 @@ def _raw_query(self, url, method="GET", return_request=False, **kwargs):
222222

223223
headers = self._get_headers()
224224

225+
logging.info(f"RSConnect API {method}: {url} -- {kwargs}")
225226
r = self.session.request(method, url, headers=headers, **kwargs)
226227

227228
if return_request:
@@ -247,7 +248,6 @@ def walk_paginated_offsets(self, f_query, endpoint, method, params=None, **kwarg
247248
all_results.extend(data["results"])
248249

249250
while data["results"]:
250-
logging.info(f"RSConnect API {method}: {endpoint}")
251251
page_kwargs = {"page_number": data["current_page"] + 1}
252252
new_params = {**params, **page_kwargs}
253253
data = f_query(endpoint, method, params=new_params)

0 commit comments

Comments
 (0)