jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix lofter not working with iqdb

Browse Source
This commit is contained in:
nonamethanks
2022-02-05 09:43:17 +01:00
parent e7744cb6e3
commit 1c9014a5bb
2 changed files with 8 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
app/logical/sources/strategies/lofter.rb
View File

@@ -40,17 +40,13 @@ module Sources
parsed_url.domain.in?(domains) || parsed_url.host =~ IMAGE_HOST
end
def image_url
if url =~ IMAGE_HOST
get_full_version(url)
else
image_urls.first
end
end
def image_urls
images = page&.search(".imgclasstag img")
images.to_a.map { |img| get_full_version(img["src"]) }
if url =~ IMAGE_HOST
[get_full_version(url)]
else
images = page&.search(".imgclasstag img")
images.to_a.map { |img| get_full_version(img["src"]) }
end
end
def get_full_version(url)

3
test/unit/sources/lofter_test.rb
View File

@@ -7,6 +7,7 @@ module Sources
@img = "https://imglf4.lf127.net/img/S1d2QlVsWkJhSW1qcnpIS0ZSa3ZJUFczb2RKSVlpMHJkNy9kc3BSQVQvQm5DNzB4eVhxay9nPT0.png?imageView&thumbnail=1680x0&quality=96&stripmeta=0"
@ref = "https://gengar563.lofter.com/post/1e82da8c_1c98dae1b"
@source = Sources::Strategies.find(@img, @ref)
@source2 = Sources::Strategies.find(@ref)
end
should "get the artist name" do
@@ -31,7 +32,7 @@ module Sources
https://imglf3.lf127.net/img/S1d2QlVsWkJhSW1qcnpIS0ZSa3ZJSzFCWFlnUWgzb01DcUdpT1lreG5yQjJVMkhGS09HNGR3PT0.png
]
assert_equal(images, @source.image_urls)
assert_equal(images, @source2.image_urls)
end
should "download the full-size image" do