From 1b426fb23fd9bde8350dc1e49db38a75b7b81388 Mon Sep 17 00:00:00 2001 From: evazion Date: Tue, 3 Sep 2019 22:33:09 -0500 Subject: [PATCH] Fix #4150: Nijie strategy fails for mp4 files. --- app/logical/sources/strategies/nijie.rb | 3 ++- test/unit/sources/nijie_test.rb | 9 +++++++++ 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/app/logical/sources/strategies/nijie.rb b/app/logical/sources/strategies/nijie.rb index 0ddc161a4..5435c229a 100644 --- a/app/logical/sources/strategies/nijie.rb +++ b/app/logical/sources/strategies/nijie.rb @@ -50,6 +50,7 @@ module Sources # https://pic03.nijie.info/nijie_picture/28310_20131101215959.jpg # https://pic03.nijie.info/nijie_picture/236014_20170620101426_0.png + # http://pic.nijie.net/03/nijie_picture/829001_20190620004513_0.mp4 # https://pic05.nijie.info/nijie_picture/diff/main/559053_20180604023346_1.png FILENAME1 = %r!(?\d+)_(?\d{14})(?:_\d+)?!i @@ -77,7 +78,7 @@ module Sources end def image_urls - images = page&.search("div#gallery a > img").to_a.map do |img| + images = page&.search("div#gallery a > .mozamoza").to_a.map do |img| "https:#{img["src"]}" end diff --git a/test/unit/sources/nijie_test.rb b/test/unit/sources/nijie_test.rb index 143a6cef1..305f991a2 100644 --- a/test/unit/sources/nijie_test.rb +++ b/test/unit/sources/nijie_test.rb @@ -220,6 +220,15 @@ module Sources end end + context "An mp4 post" do + should "find the mp4 file" do + site = Sources::Strategies.find("https://nijie.info/view.php?id=319485") + + assert_equal(["https://pic.nijie.net/03/nijie_picture/829001_20190620004513_0.mp4"], site.image_urls) + assert_equal("https://pic.nijie.net/03/nijie_picture/829001_20190620004513_0.mp4", site.image_url) + end + end + context "An artist profile url" do should "not fail" do site = Sources::Strategies.find("https://nijie.info/members_illust.php?id=236014")