此代码将遍历多个页面以查找和提取页面上的元素。一旦循环完成,它将生成一个包含来自HashMap的这些元素的日志文件,但结果不会被追加,而是被覆盖。
int d = new Integer(0);
for (int i = 0; i <= 100; d += 10) {
String url = Constants.FilterUrl + "&startIndex=" + d;
this.getAuthors();
driver.get(url);
if (!driver.getPageSource().contains("h3")) break;
}
/* Send HashMap values to text file */
File file = new File(Constants.FILEPATH + Constants.dateFormat.format(new Date()) + ".txt");
try{
if(!file.exists()){
System.out.println("We had to make a new file.");
file.createNewFile();
}
PrintWriter out = new PrintWriter(new FileWriter(file), true);
map.forEach((k, v) -> out.println(k + ", " + v));
out.append("************** " + "\n");
out.close();
} catch(IOException e) {
System.out.println("COULD NOT LOG!!");
}
}
public void getAuthors(){
List<WebElement> allElements = driver.findElements(By.tagName("h3"));
/* Create HashMap and store H3 elements in the key set */
this.map = new HashMap<String, String>();
for (WebElement element1 : allElements) {
map.put(element1.getText(), element1.findElement(By.tagName("a")).getAttribute("href"));
}
/* Visit pages for H3 elements and retrieve names of the authors */
for (Map.Entry<String, String> entry : map.entrySet()) {
driver.get(entry.getValue());
entry.setValue(driver.findElement(By.className("userlink-0")).getText());
}
}
有什么想法吗?
答案 0 :(得分:1)
map.put(element1.getText(), element1.findElement(By.tagName( “A”))的getAttribute(的 “href”));
如果HashMap中的任何条目与element1.getText()
具有相同的文本,它将覆盖它。
此外,您还要为每个调用创建地图,每次都会创建一个新地图,导致早期内容的数据丢失。
/* Create HashMap and store H3 elements in the key set */ this.map = new HashMap<String, String>();
您应该在实例级别创建它。
要生成唯一键,请在实例级别定义一个数字变量,并为每个put添加该变量。
long counter = 0;
map.put(counter++, element1.findElement(By.tagName("a")).getAttribute("href"));
可能会将HashMap更改为Key而不是String。
答案 1 :(得分:0)
for (WebElement element1 : allElements) {
i++
map.put(element1.getText()+i, element1.findElement(By.tagName("a")).getAttribute("href"));
}
添加i ++以便它不会覆盖