Fixed Modelmayhem ripper
This commit is contained in:
cyian-1756
parent
13162bed73
commit
0845efe328
@ -3,134 +3,64 @@ package com.rarchives.ripme.ripper.rippers;
|
||||
import java.io.IOException;
|
||||
import java.net.MalformedURLException;
|
||||
import java.net.URL;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
import org.json.JSONArray;
|
||||
import org.json.JSONObject;
|
||||
import org.jsoup.Connection.Method;
|
||||
import org.jsoup.Connection.Response;
|
||||
import org.jsoup.Jsoup;
|
||||
import org.jsoup.nodes.Document;
|
||||
import org.jsoup.nodes.Element;
|
||||
|
||||
import com.rarchives.ripme.ripper.AlbumRipper;
|
||||
import com.rarchives.ripme.ui.RipStatusMessage.STATUS;
|
||||
import com.rarchives.ripme.utils.Utils;
|
||||
import com.rarchives.ripme.ripper.AbstractHTMLRipper;
|
||||
import com.rarchives.ripme.utils.Http;
|
||||
|
||||
public class ModelmayhemRipper extends AlbumRipper {
|
||||
|
||||
private static final String DOMAIN = "modelmayhem.com",
|
||||
HOST = "modelmayhem";
|
||||
public class ModelmayhemRipper extends AbstractHTMLRipper {
|
||||
|
||||
public ModelmayhemRipper(URL url) throws IOException {
|
||||
super(url);
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean canRip(URL url) {
|
||||
return (url.getHost().endsWith(DOMAIN));
|
||||
}
|
||||
|
||||
@Override
|
||||
public URL sanitizeURL(URL url) throws MalformedURLException {
|
||||
return url;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void rip() throws IOException {
|
||||
Map<String,String> cookies = null,
|
||||
postData = new HashMap<>();
|
||||
String gid = getGID(this.url),
|
||||
ref = "http://www.modelmayhem.com/" + gid;
|
||||
|
||||
Response resp = null;
|
||||
String theurl = "http://www.modelmayhem.com/" + gid;
|
||||
logger.info("Loading " + theurl);
|
||||
resp = Jsoup.connect(theurl)
|
||||
.timeout(5000)
|
||||
.referrer("")
|
||||
.userAgent(USER_AGENT)
|
||||
.method(Method.GET)
|
||||
.execute();
|
||||
cookies = resp.cookies();
|
||||
|
||||
resp = Jsoup.connect("http://www.modelmayhem.com/includes/js/auth.php")
|
||||
.cookies(cookies)
|
||||
.ignoreContentType(true)
|
||||
.referrer(ref)
|
||||
.userAgent(USER_AGENT)
|
||||
.method(Method.GET)
|
||||
.execute();
|
||||
String authText = resp.parse().html();
|
||||
String mmservice = authText.substring(authText.indexOf("token = '") + 9);
|
||||
mmservice = mmservice.substring(0, mmservice.indexOf("'"));
|
||||
|
||||
cookies.putAll(resp.cookies());
|
||||
|
||||
cookies.put("worksafe", "0");
|
||||
theurl = "http://www.modelmayhem.com/services/photo_viewer/albums/" + gid;
|
||||
postData.put("MMSERVICE", mmservice);
|
||||
resp = Jsoup.connect(theurl)
|
||||
.data(postData)
|
||||
.cookies(cookies)
|
||||
.referrer(ref)
|
||||
.userAgent(USER_AGENT)
|
||||
.method(Method.POST)
|
||||
.execute();
|
||||
cookies.putAll(resp.cookies());
|
||||
|
||||
theurl = "http://www.modelmayhem.com/services/photo_viewer/pictures/" + gid + "/0/0/1/0";
|
||||
this.sendUpdate(STATUS.LOADING_RESOURCE, theurl);
|
||||
logger.info("Loading " + theurl);
|
||||
resp = Jsoup.connect(theurl)
|
||||
.data(postData)
|
||||
.cookies(cookies)
|
||||
.referrer(ref)
|
||||
.userAgent(USER_AGENT)
|
||||
.method(Method.POST)
|
||||
.execute();
|
||||
|
||||
Document doc = resp.parse();
|
||||
String jsonText = doc.body().html();
|
||||
jsonText = jsonText.replace(""", "\"");
|
||||
System.err.println(jsonText);
|
||||
JSONObject json = new JSONObject(jsonText);
|
||||
JSONArray pictures = json.getJSONArray("pictures");
|
||||
for (int i = 0; i < pictures.length(); i++) {
|
||||
JSONObject picture = pictures.getJSONObject(i);
|
||||
String bigImage = picture.getString("big_image");
|
||||
if (bigImage.trim().equals("")) {
|
||||
logger.info("Got empty image for " + picture.toString(2));
|
||||
continue;
|
||||
}
|
||||
String prefix = "";
|
||||
if (Utils.getConfigBoolean("download.save_order", true)) {
|
||||
prefix = String.format("%03d_", i + 1);
|
||||
}
|
||||
addURLToDownload(new URL(bigImage), prefix);
|
||||
|
||||
if (isThisATest()) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
waitForThreads();
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getHost() {
|
||||
return HOST;
|
||||
return "modelmayhem";
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getDomain() {
|
||||
return "modelmayhem.com";
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getGID(URL url) throws MalformedURLException {
|
||||
Pattern p = Pattern.compile("^https?://[w.]*modelmayhem.com.*/([0-9]+)/?.*$");
|
||||
Pattern p = Pattern.compile("https?://www\\.modelmayhem\\.com/portfolio/(\\d+)/viewall");
|
||||
Matcher m = p.matcher(url.toExternalForm());
|
||||
if (m.matches()) {
|
||||
return m.group(1);
|
||||
}
|
||||
throw new MalformedURLException("Modelmayhem user ID not found in " + url + ", expected http://modelmayhem.com/userid");
|
||||
throw new MalformedURLException("Expected modelmayhem URL format: " +
|
||||
"modelmayhem.com/portfolio/ID/viewall - got " + url + " instead");
|
||||
}
|
||||
|
||||
@Override
|
||||
public Document getFirstPage() throws IOException {
|
||||
// "url" is an instance field of the superclass
|
||||
return Http.url(url).get();
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<String> getURLsFromPage(Document doc) {
|
||||
List<String> result = new ArrayList<>();
|
||||
for (Element el : doc.select("tr.a_pics > td > div > a")) {
|
||||
String image_URL = el.select("img").attr("src").replaceAll("_m", "");
|
||||
if (image_URL.contains("http")) {
|
||||
result.add(image_URL);
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void downloadURL(URL url, int index) {
|
||||
addURLToDownload(url, getPrefix(index));
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user