罗俊辉 11 месяцев назад
Родитель
Сommit
042833219d
2 измененных файлов с 12 добавлено и 5 удалено
  1. 10 3
      routes/AccountArticleRank.py
  2. 2 2
      test/rank_dev.py

+ 10 - 3
routes/AccountArticleRank.py

@@ -76,9 +76,15 @@ class AccountArticleRank(object):
         try:
             rank_info = ArticleRank().rank(account_list=[self.accountName], text_list=self.title_list)
             score_list = rank_info[self.accountName]['score_list']
-            for index, item in enumerate(self.publishArticleList):
-                item['score'] = score_list[index]
-            sorted_list = sorted(self.publishArticleList, key=lambda x: x['score'], reverse=True)
+            title_score_dict = {}
+            for index, item in enumerate(self.title_list):
+                title_score_dict[item] = score_list[index]
+            result_list = []
+            for obj in self.publishArticleList:
+                if title_score_dict.get(obj['title']):
+                    obj['score'] = title_score_dict[obj['title']]
+                    result_list.append(obj)
+            sorted_list = sorted(result_list, key=lambda x: x['score'], reverse=True)
             result = {
                 "accountId": self.accountId,
                 "accountName": self.accountName,
@@ -88,6 +94,7 @@ class AccountArticleRank(object):
                 "rank_list": sorted_list[:self.publishNum]
             }
         except Exception as e:
+            print("error", e)
             result = {
                 "accountId": self.accountId,
                 "accountName": self.accountName,

Разница между файлами не показана из-за своего большого размера
+ 2 - 2
test/rank_dev.py


Некоторые файлы не были показаны из-за большого количества измененных файлов