Browse Source

check_account_info方法增加
account_name参数, 该参数每日从 aigc 数据库和 gh_id一起获取
若是一个新的账号,则返回该 account_name

luojunhui 8 months ago
parent
commit
1453b5ae64
1 changed files with 7 additions and 5 deletions
  1. 7 5
      updatePublishedMsgDaily.py

+ 7 - 5
updatePublishedMsgDaily.py

@@ -232,9 +232,10 @@ def update_each_account(db_client, account_info, account_name, latest_update_tim
         return
 
 
-def check_account_info(db_client, gh_id):
+def check_account_info(db_client, gh_id, account_name):
     """
     通过 gh_id查询视频信息
+    :param account_name:
     :param db_client:
     :param gh_id:
     :return:
@@ -247,15 +248,15 @@ def check_account_info(db_client, gh_id):
         """
     result = db_client.select(sql)
     if result:
-        account_name, update_time = result[0]
+        old_account_name, update_time = result[0]
         return {
-            "account_name": account_name,
+            "account_name": old_account_name,
             "update_time": update_time,
             "account_type": "history"
         }
     else:
         return {
-            "account_name": "",
+            "account_name": account_name,
             "update_time": int(time.time()) - 30 * 24 * 60 * 60,
             "account_type": "new"
         }
@@ -269,7 +270,8 @@ def update_single_account(db_client, account_info):
     :return:
     """
     gh_id = account_info['ghId']
-    account_detail = check_account_info(db_client, gh_id)
+    account_name = account_info['name']
+    account_detail = check_account_info(db_client, gh_id, account_name)
     account_name = account_detail['account_name']
     update_time = account_detail['update_time']
     update_each_account(