1
0
mirror of https://github.com/RipMeApp/ripme.git synced 2025-08-15 02:14:10 +02:00

Added support for folder type urls.

This commit is contained in:
Tushar
2019-09-02 12:08:01 +05:30
parent 4711f4f7cb
commit 9cd2c1f984
2 changed files with 123 additions and 39 deletions

View File

@@ -11,17 +11,25 @@ import java.util.regex.Matcher;
import java.util.regex.Pattern; import java.util.regex.Pattern;
import org.jsoup.nodes.Document; import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element; import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;
import com.rarchives.ripme.ripper.AbstractHTMLRipper; import com.rarchives.ripme.ripper.AbstractHTMLRipper;
import com.rarchives.ripme.ripper.DownloadThreadPool; import com.rarchives.ripme.ripper.DownloadThreadPool;
import com.rarchives.ripme.utils.Http; import com.rarchives.ripme.utils.Http;
/**
* @author Tushar
*
*/
public class ListalRipper extends AbstractHTMLRipper { public class ListalRipper extends AbstractHTMLRipper {
private Pattern p = Pattern.compile("https://www.listal.com/list/([a-zA-Z0-9-]+)"); private Pattern p1 = Pattern.compile("https:\\/\\/www.listal.com\\/list\\/([a-zA-Z0-9-]+)");
private Pattern p2 =
Pattern.compile("https:\\/\\/www.listal.com\\/((?:(?:[a-zA-Z0-9-]+)\\/?)+)");
private String listId = null; // listId to get more images via POST. private String listId = null; // listId to get more images via POST.
private String postUrl = "https://www.listal.com/item-list/"; //to load more images. private String postUrl = "https://www.listal.com/item-list/"; //to load more images.
private UrlType urlType = UrlType.UNKNOWN;
private DownloadThreadPool listalThreadPool = new DownloadThreadPool("listalThreadPool"); private DownloadThreadPool listalThreadPool = new DownloadThreadPool("listalThreadPool");
@@ -42,21 +50,22 @@ public class ListalRipper extends AbstractHTMLRipper {
@Override @Override
public Document getFirstPage() throws IOException { public Document getFirstPage() throws IOException {
Document doc = Http.url(url).get(); Document doc = Http.url(url).get();
listId = doc.select("#customlistitems").first().attr("data-listid"); if (urlType == UrlType.LIST) {
listId = doc.select("#customlistitems").first().attr("data-listid"); // Used for list types.
}
return doc; return doc;
} }
@Override @Override
public List<String> getURLsFromPage(Document page) { public List<String> getURLsFromPage(Document page) {
Matcher m = p.matcher(url.toExternalForm()); if (urlType == UrlType.LIST) {
if (m.matches()) { // for url of type LIST, https://www.listal.com/list/my-list
// for url of type https://www.listal.com/list/my-list return getURLsForListType(page);
return getURLsFromList(page); } else if (urlType == UrlType.FOLDER) {
} else { // for url of type FOLDER, https://www.listal.com/jim-carrey/pictures
// for url of type https://www.listal.com/jim-carrey/pictures return getURLsForFolderType(page);
//TODO need to write
return null;
} }
return null;
} }
@Override @Override
@@ -66,54 +75,111 @@ public class ListalRipper extends AbstractHTMLRipper {
@Override @Override
public String getGID(URL url) throws MalformedURLException { public String getGID(URL url) throws MalformedURLException {
Pattern p = Pattern.compile("https://www.listal.com/list/([a-zA-Z0-9-]+)"); Matcher m1 = p1.matcher(url.toExternalForm());
Matcher m = p.matcher(url.toExternalForm()); if (m1.matches()) {
if (m.matches()) {
// Return the text contained between () in the regex // Return the text contained between () in the regex
return m.group(1); urlType = UrlType.LIST;
return m1.group(1);
}
Matcher m2 = p2.matcher(url.toExternalForm());
if (m2.matches()) {
// Return only gid from capturing group of type listal.com/tvOrSomething/dexter/pictures
urlType = UrlType.FOLDER;
return getFolderTypeGid(m2.group(1));
} }
//TODO match /../celebrity_name/images
throw new MalformedURLException("Expected listal.com URL format: " throw new MalformedURLException("Expected listal.com URL format: "
+ "listal.com/list/my-list-name - got " + url + " instead."); + "listal.com/list/my-list-name - got " + url + " instead.");
} }
@Override @Override
public Document getNextPage(Document doc) throws IOException { public Document getNextPage(Document page) throws IOException {
// TODO Auto-generated method stub Document nextPage = super.getNextPage(page);
return super.getNextPage(doc); switch (urlType) {
case LIST:
if (!page.select(".loadmoreitems").isEmpty()) {
// All items are not loaded.
// Load remaining items using postUrl.
String offSet = page.select(".loadmoreitems").last().attr("data-offset");
Map<String, String> postParams = new HashMap<>();
postParams.put("listid", listId);
postParams.put("offset", offSet);
try {
nextPage = Http.url(postUrl).data(postParams).retries(3).post();
} catch (IOException e1) {
LOGGER.error("Failed to load more images after " + offSet, e1);
throw e1;
}
}
break;
case FOLDER:
Elements pageLinks = page.select(".pages a");
if (!pageLinks.isEmpty() && pageLinks.last().text().startsWith("Next")) {
String nextUrl = pageLinks.last().attr("abs:href");
nextPage = Http.url(nextUrl).retries(3).get();
}
break;
case UNKNOWN:
default:
}
return nextPage;
} }
private List<String> getURLsFromList(Document page) {
// recursive method for url type: https://www.listal.com/list/my-list
@Override
public DownloadThreadPool getThreadPool() {
return listalThreadPool;
}
/**
* Returns the image urls for UrlType LIST.
*/
private List<String> getURLsForListType(Document page) {
List<String> list = new ArrayList<>(); List<String> list = new ArrayList<>();
for (Element e : page.select(".pure-g a[href*=viewimage]")) { for (Element e : page.select(".pure-g a[href*=viewimage]")) {
//list.add("https://www.listal.com" + e.attr("href") + "h"); //list.add("https://www.listal.com" + e.attr("href") + "h");
list.add(e.attr("abs:href") + "h"); list.add(e.attr("abs:href") + "h");
} }
if (!page.select(".loadmoreitems").isEmpty()) {
// All items are not loaded.
// Load remaining items using postUrl.
String offSet = page.select(".loadmoreitems").last().attr("data-offset");
Map<String, String> postParams = new HashMap<>();
postParams.put("listid", listId);
postParams.put("offset", offSet);
try {
list.addAll(getURLsFromList(Http.url(postUrl).data(postParams).retries(3).post()));
} catch (IOException e1) {
LOGGER.error("Failed to load more images after " + offSet, e1);
}
}
return list; return list;
} }
@Override /**
public DownloadThreadPool getThreadPool() { * Returns the image urls for UrlType FOLDER.
return listalThreadPool; */
private List<String> getURLsForFolderType(Document page) {
List<String> list = new ArrayList<>();
for (Element e : page.select("#browseimagescontainer .imagewrap-outer a")) {
list.add(e.attr("abs:href") + "h");
}
return list;
}
/**
* Returns the gid for url type listal.com/tvOrSomething/dexter/pictures
*/
public String getFolderTypeGid(String group) throws MalformedURLException {
String[] folders = group.split("/");
try {
if (folders.length == 2 && folders[1].equals("pictures")) {
// Url is probably for an actor.
return folders[0];
}
if (folders.length == 3 && folders[2].equals("pictures")) {
// Url if for a folder(like movies, tv etc).
Document doc = Http.url(url).get();
return doc.select(".itemheadingmedium").first().text();
}
} catch (Exception e) {
LOGGER.error(e);
}
throw new MalformedURLException("Unable to fetch the gid for given url.");
} }
private class ListalImageDownloadThread extends Thread { private class ListalImageDownloadThread extends Thread {
@@ -161,4 +227,8 @@ public class ListalRipper extends AbstractHTMLRipper {
return name + ".jpg"; return name + ".jpg";
} }
} }
private static enum UrlType {
LIST, FOLDER, UNKNOWN
}
} }

View File

@@ -6,10 +6,24 @@ import com.rarchives.ripme.ripper.rippers.ListalRipper;
public class ListalRipperTest extends RippersTest { public class ListalRipperTest extends RippersTest {
public void testRip() throws IOException { /**
* Test for list type url.
* @throws IOException
*/
public void testRipListType() throws IOException {
ListalRipper ripper = ListalRipper ripper =
new ListalRipper(new URL("https://www.listal.com/list/evolution-emma-stone")); new ListalRipper(new URL("https://www.listal.com/list/evolution-emma-stone"));
testRipper(ripper); testRipper(ripper);
} }
/**
* Test for folder type url.
* @throws IOException
*/
public void testRipFolderType() throws IOException {
ListalRipper ripper =
new ListalRipper(new URL("https://www.listal.com/chet-atkins/pictures"));
testRipper(ripper);
}
} }