Commit fefa3482 authored by jyong's avatar jyong

fix sync source info update

parent 58d769b3
...@@ -18,7 +18,7 @@ from extensions.ext_database import db ...@@ -18,7 +18,7 @@ from extensions.ext_database import db
from extensions.ext_login import login_manager from extensions.ext_login import login_manager
# DO NOT REMOVE BELOW # DO NOT REMOVE BELOW
from models import model, account, dataset, web, task from models import model, account, dataset, web, task, source
from events import event_handlers from events import event_handlers
# DO NOT REMOVE ABOVE # DO NOT REMOVE ABOVE
......
...@@ -99,11 +99,15 @@ class NotionOAuth(OAuthDataSource): ...@@ -99,11 +99,15 @@ class NotionOAuth(OAuthDataSource):
# get all authorized pages # get all authorized pages
pages = self.get_authorized_pages(data_source_binding.access_token) pages = self.get_authorized_pages(data_source_binding.access_token)
source_info = data_source_binding.source_info source_info = data_source_binding.source_info
source_info['pages'] = pages new_source_info = {
source_info['total'] = len(pages) 'workspace_name': source_info['workspace_name'],
data_source_binding.source_info = source_info 'workspace_icon': source_info['workspace_icon'],
'workspace_id': source_info['workspace_id'],
'pages': pages,
'total': len(pages)
}
data_source_binding.source_info = new_source_info
data_source_binding.disabled = False data_source_binding.disabled = False
db.session.add(data_source_binding)
db.session.commit() db.session.commit()
else: else:
raise ValueError('Data source binding not found') raise ValueError('Data source binding not found')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment