🐞 fix(谷歌学术搜索): 包装search.results()为空可能造成的报错

#423
binary-husky · May 15, 2023 · 3f31fb9 · 3f31fb9
1 parent c0e57e0
commit 3f31fb9
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/crazy_functions/谷歌检索小助手.py b/crazy_functions/谷歌检索小助手.py
@@ -36,14 +36,18 @@ def string_similar(s1, s2):
             max_results = 1,
             sort_by = arxiv.SortCriterion.Relevance,
         )
-        paper = next(search.results())
-        if string_similar(title, paper.title) > 0.90: # same paper
-            abstract = paper.summary.replace('\n', ' ')
-            is_paper_in_arxiv = True
-        else:   # different paper
+        try:
+            paper = next(search.results())
+            if string_similar(title, paper.title) > 0.90: # same paper
+                abstract = paper.summary.replace('\n', ' ')
+                is_paper_in_arxiv = True
+            else:   # different paper
+                abstract = abstract
+                is_paper_in_arxiv = False
+            paper = next(search.results())
+        except:
             abstract = abstract
             is_paper_in_arxiv = False
-        paper = next(search.results())
         print(title)
         print(author)
         print(citation)