From 224da8a7dace706a8f97372cb92ddd38bee542aa Mon Sep 17 00:00:00 2001 From: Toks Date: Wed, 25 Jun 2014 11:14:08 -0400 Subject: [PATCH] Prevent pixiv downloader from returning wrong image --- app/logical/sources/strategies/pixiv.rb | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/app/logical/sources/strategies/pixiv.rb b/app/logical/sources/strategies/pixiv.rb index 2add709ce..753b34074 100644 --- a/app/logical/sources/strategies/pixiv.rb +++ b/app/logical/sources/strategies/pixiv.rb @@ -50,9 +50,12 @@ module Sources end def get_image_url_from_page(page) - element = page.search("div.works_display a img").first - if element - element.attr("src").sub(/_[ms]\./, ".") + elements = page.search("div.works_display a img").find_all do |node| + node["src"] !~ /source\.pixiv\.net/ + end + + if elements.any? + elements.first.attr("src").sub(/_[ms]\./, ".") else nil end