From 4d66a865791c83dd3a8ed7632b98e44777993031 Mon Sep 17 00:00:00 2001 From: yalei <269870927@qq.com> Date: Fri, 26 Apr 2024 18:04:37 +0800 Subject: [PATCH] fix: fetch page name of notion wiki (#3847) --- api/libs/oauth_data_source.py | 23 ++++++----------------- 1 file changed, 6 insertions(+), 17 deletions(-) diff --git a/api/libs/oauth_data_source.py b/api/libs/oauth_data_source.py index 7891b01182..a865ee85ab 100644 --- a/api/libs/oauth_data_source.py +++ b/api/libs/oauth_data_source.py @@ -153,23 +153,12 @@ class NotionOAuth(OAuthDataSource): # get page detail for page_result in page_results: page_id = page_result['id'] - if 'Name' in page_result['properties']: - if len(page_result['properties']['Name']['title']) > 0: - page_name = page_result['properties']['Name']['title'][0]['plain_text'] - else: - page_name = 'Untitled' - elif 'title' in page_result['properties']: - if len(page_result['properties']['title']['title']) > 0: - page_name = page_result['properties']['title']['title'][0]['plain_text'] - else: - page_name = 'Untitled' - elif 'Title' in page_result['properties']: - if len(page_result['properties']['Title']['title']) > 0: - page_name = page_result['properties']['Title']['title'][0]['plain_text'] - else: - page_name = 'Untitled' - else: - page_name = 'Untitled' + page_name = 'Untitled' + for key in ['Name', 'title', 'Title', 'Page']: + if key in page_result['properties']: + if len(page_result['properties'][key].get('title', [])) > 0: + page_name = page_result['properties'][key]['title'][0]['plain_text'] + break page_icon = page_result['icon'] if page_icon: icon_type = page_icon['type']