Browse Source

improve: Refresh updated_at field of DataSourceOauthBinding model (#9173)

Signed-off-by: Xiaoguang Sun <sunxiaoguang@gmail.com>
Xiaoguang Sun 6 months ago
parent
commit
be2f1e59f2
1 changed files with 4 additions and 0 deletions
  1. 4 0
      api/libs/oauth_data_source.py

+ 4 - 0
api/libs/oauth_data_source.py

@@ -1,3 +1,4 @@
+import datetime
 import urllib.parse
 
 import requests
@@ -69,6 +70,7 @@ class NotionOAuth(OAuthDataSource):
         if data_source_binding:
             data_source_binding.source_info = source_info
             data_source_binding.disabled = False
+            data_source_binding.updated_at = datetime.datetime.now(datetime.timezone.utc).replace(tzinfo=None)
             db.session.commit()
         else:
             new_data_source_binding = DataSourceOauthBinding(
@@ -104,6 +106,7 @@ class NotionOAuth(OAuthDataSource):
         if data_source_binding:
             data_source_binding.source_info = source_info
             data_source_binding.disabled = False
+            data_source_binding.updated_at = datetime.datetime.now(datetime.timezone.utc).replace(tzinfo=None)
             db.session.commit()
         else:
             new_data_source_binding = DataSourceOauthBinding(
@@ -138,6 +141,7 @@ class NotionOAuth(OAuthDataSource):
             }
             data_source_binding.source_info = new_source_info
             data_source_binding.disabled = False
+            data_source_binding.updated_at = datetime.datetime.now(datetime.timezone.utc).replace(tzinfo=None)
             db.session.commit()
         else:
             raise ValueError("Data source binding not found")