Coverage for src/crawler/by_source/amp_crawler.py: 91%

1import lingua 

2from bs4 import BeautifulSoup 

3from lingua import LanguageDetectorBuilder 

4from ptf.model_data import create_abstract, create_articledata 

5 

6from crawler.base_crawler import BaseCollectionCrawler 

7from crawler.types import CitationLiteral 

8 

9 

10class AmpCrawler(BaseCollectionCrawler): 

11 source_name = "Annals of Mathematics website" 

12 source_domain = "AMP" 

13 source_website = "https://annals.math.princeton.edu" 

14 

15 language_detector = LanguageDetectorBuilder.from_languages( 

16 lingua.Language.ENGLISH, 

17 lingua.Language.FRENCH, 

18 ).build() 

19 

20 def parse_collection_content(self, content): 

21 """ 

22 Parse the HTML page of Annals of Math and returns a list of xissue. 

23 Each xissue has its volume/number/year metadata + its url 

24 """ 

25 soup = BeautifulSoup(content, "html.parser") 

26 xissues = [] 

27 

28 # Extract the list of issues 

29 issue_nodes = soup.find_all("div", {"class": "cat-item-2"}) 

30 

31 for issue_node in issue_nodes: 

32 issue_link_node = issue_node.find("a") 

33 if issue_link_node: 33 ↛ 31line 33 didn't jump to line 31 because the condition on line 33 was always true

34 url = issue_link_node.get("href") 

35 xissue = self.create_amp_xissue(url) 

36 if xissue: 36 ↛ 31line 36 didn't jump to line 31 because the condition on line 36 was always true

37 xissues.append(xissue) 

38 

39 return xissues 

40 

41 def create_amp_xissue(self, url): 

42 if url.endswith("/"): 

43 url = url[:-1] 

44 parts = url.split("/") 

45 

46 last_part = parts[-1] 

47 exceptions = last_part.split("-") 

48 if len(exceptions) > 2: 

49 year = exceptions[0] 

50 volume = exceptions[0] 

51 number = exceptions[1] 

52 else: 

53 year = parts[-2] 

54 if len(year) < 4: 

55 # The links are different with volumes before 2015 

56 year = parts[-3] 

57 

58 volume_number = parts[-1] 

59 volume_number_parts = volume_number.split("-") 

60 volume = volume_number_parts[0] 

61 number = volume_number_parts[1] 

62 

63 return self.create_xissue(url, year, volume, number) 

64 

65 def parse_issue_content(self, content, xissue): 

66 soup = BeautifulSoup(content, "html.parser") 

67 article_nodes = soup.find_all("h2", {"class": "entry-title"}) 

68 

69 for index_article, article_node in enumerate(article_nodes): 

70 article_link_node = article_node.find("a") 

71 if article_link_node: 71 ↛ 69line 71 didn't jump to line 69 because the condition on line 71 was always true

72 url = article_link_node.get("href") 

73 xarticle = create_articledata() 

74 xarticle.pid = "a" + str(index_article) 

75 xarticle.url = url 

76 xissue.articles.append(xarticle) 

77 

78 def parse_article_content(self, content, xissue, xarticle, url): 

79 """ 

80 Parse the content with Beautifulsoup and returns an ArticleData 

81 """ 

82 xarticle.lang = "en" 

83 

84 soup = BeautifulSoup(content, "html.parser") 

85 

86 what: list[CitationLiteral] = ["author", "abstract", "page"] 

87 

88 doi_tag = soup.select_one("meta[name='citation_doi']") 

89 if doi_tag and not doi_tag.text.startswith("https://doi.org/"): 89 ↛ 92line 89 didn't jump to line 92 because the condition on line 89 was always true

90 what.append("pdf") 

91 

92 title_node = soup.find("h1", {"class": "entry-title"}) 

93 if title_node: 93 ↛ 96line 93 didn't jump to line 96 because the condition on line 93 was always true

94 what.append("title") 

95 

96 if url != "https://annals.math.princeton.edu/2010/172-3/p06": 96 ↛ 102line 96 didn't jump to line 102 because the condition on line 96 was always true

97 # Exception with Annals of Math: 2 articles have the same DOI ! 

98 # https://annals.math.princeton.edu/2010/172-3/p06 and https://annals.math.princeton.edu/2011/173-1/p14 

99 # we ignore DOI/ZBMATH/MR for the first one 

100 what.append("doi") 

101 

102 self.get_metadata_using_citation_meta(xarticle, xissue, soup, what) 

103 

104 abstract_node = soup.select_one("div.entry-content") 

105 if abstract_node is not None: 105 ↛ 116line 105 didn't jump to line 116 because the condition on line 105 was always true

106 abstract_section_node = abstract_node.select_one("p") 

107 if abstract_section_node: 107 ↛ 116line 107 didn't jump to line 116 because the condition on line 107 was always true

108 abstract = str(abstract_section_node) 

109 xarticle.abstracts.append( 

110 create_abstract( 

111 value_tex=abstract, lang=self.detect_language(abstract, xarticle) 

112 ) 

113 ) 

114 

115 # ZBMATH 

116 metadata_header_nodes = soup.find_all("div", {"class": "metadata-headers"}) 

117 for metadata_header_node in metadata_header_nodes: 

118 text = metadata_header_node.get_text() 

119 

120 if text == "zbMATH": 

121 link_node = metadata_header_node.parent.find("a") 

122 if link_node: 122 ↛ 117line 122 didn't jump to line 117 because the condition on line 122 was always true

123 zblid = link_node.get("href") 

124 pos = zblid.find("?q=an:") 

125 if pos > 0: 125 ↛ 127line 125 didn't jump to line 127 because the condition on line 125 was always true

126 zblid = zblid[pos + 6 :] 

127 xarticle.extids.append(("zbl-item-id", zblid)) 

128 elif text == "MR": 

129 link_node = metadata_header_node.parent.find("a") 

130 if link_node: 130 ↛ 117line 130 didn't jump to line 117 because the condition on line 130 was always true

131 mrid = link_node.get("href") 

132 pos = mrid.find("?mr=") 

133 if pos > 0: 133 ↛ 135line 133 didn't jump to line 135 because the condition on line 133 was always true

134 mrid = mrid[pos + 4 :] 

135 xarticle.extids.append(("mr-item-id", mrid)) 

136 

137 return xarticle