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
from extensions.ext_login import login_manager
# 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
# DO NOT REMOVE ABOVE
......
......@@ -99,11 +99,15 @@ class NotionOAuth(OAuthDataSource):
# get all authorized pages
pages = self.get_authorized_pages(data_source_binding.access_token)
source_info = data_source_binding.source_info
source_info['pages'] = pages
source_info['total'] = len(pages)
data_source_binding.source_info = source_info
new_source_info = {
'workspace_name': source_info['workspace_name'],
'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
db.session.add(data_source_binding)
db.session.commit()
else:
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