From 5ea86301d9a2f376a6417a29ca3030c884b86cae Mon Sep 17 00:00:00 2001 From: Jet Li Date: Fri, 31 Jan 2025 03:52:09 +0000 Subject: [PATCH] Hack solution to update selected --- .../backend/content/refresh_sharepoint_content.py | 12 ++++++++++-- .../scheduler/refresh_sharepoint_content_job.py | 6 ++++-- 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/apps/content/backend/content/refresh_sharepoint_content.py b/apps/content/backend/content/refresh_sharepoint_content.py index 6aca88b..6475e3a 100644 --- a/apps/content/backend/content/refresh_sharepoint_content.py +++ b/apps/content/backend/content/refresh_sharepoint_content.py @@ -6,12 +6,15 @@ from backend.content.content_service import ContentService from backend.content.models import ContentFolderDoc from backend.content.constants import ContentSource + class SharePointContentRefresher: def __init__(self) -> None: self.content_sharepoint_manager = ContentSharePointManager() + @log_entry_exit_async async def refresh_all_in_database(self) -> None: await self.content_sharepoint_manager.retrieve_directorys_for_all_folders() + @log_entry_exit_async async def refresh_selected(self) -> None: selected_folders = [ @@ -23,9 +26,14 @@ class SharePointContentRefresher: {"folder": "career", "region": UserRegion.ZH_CN}, {"folder": "contact", "region": UserRegion.OTHER}, {"folder": "contact", "region": UserRegion.ZH_CN}, + {"folder": "testimony", "region": UserRegion.ZH_CN}, + {"folder": "testimony", "region": UserRegion.ZH_CN}, + {"folder": "development", "region": UserRegion.ZH_CN}, + {"folder": "development", "region": UserRegion.ZH_CN}, + {"folder": "legal", "region": UserRegion.ZH_CN}, + {"folder": "legal", "region": UserRegion.ZH_CN}, ] for folder in selected_folders: await self.content_sharepoint_manager.retrieve_directories_for_folder( - folder_name=folder["folder"], - region=folder["region"] + folder_name=folder["folder"], region=folder["region"] ) diff --git a/apps/content/scheduler/refresh_sharepoint_content_job.py b/apps/content/scheduler/refresh_sharepoint_content_job.py index 2e94c5f..2fa5e84 100644 --- a/apps/content/scheduler/refresh_sharepoint_content_job.py +++ b/apps/content/scheduler/refresh_sharepoint_content_job.py @@ -5,11 +5,13 @@ from backend.content.refresh_sharepoint_content import SharePointContentRefreshe async def refresh_sharepoint_content_job(): - if await acquire_lock(ScheduleJobLocker.REFRESH_SHAREPOINT_CONTENT_JOB_LOCKER, 3600): + if await acquire_lock( + ScheduleJobLocker.REFRESH_SHAREPOINT_CONTENT_JOB_LOCKER, 3600 + ): try: logging.info("Starting job to refresh sharepoint content job.") sharepoint_refresher = SharePointContentRefresher() - await sharepoint_refresher.refresh_all_in_database() + await sharepoint_refresher.refresh_selected() logging.info("Exiting job to refresh sharepoint content job.") finally: await release_lock(ScheduleJobLocker.REFRESH_SHAREPOINT_CONTENT_JOB_LOCKER)