|
@@ -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,
|