fix bug with source::site
This commit is contained in:
@@ -6,7 +6,7 @@ div.list-of-forum-posts {
|
||||
}
|
||||
|
||||
article {
|
||||
border: 1px solid #999;
|
||||
border: 1px solid #AAA;
|
||||
margin: 1em 0em;
|
||||
word-wrap: break-word;
|
||||
box-shadow: 1px 1px 2px #AAA;
|
||||
|
||||
@@ -18,7 +18,6 @@ module Downloads
|
||||
protected
|
||||
def rewrite_html_url(url, headers)
|
||||
return [url, headers] unless Sources::Strategies::ArtStation.url_match?(url)
|
||||
|
||||
source = Sources::Site.new(url)
|
||||
source.get
|
||||
[source.image_url, headers]
|
||||
|
||||
@@ -17,16 +17,14 @@ module Sources
|
||||
end
|
||||
|
||||
def initialize(url, referer_url: nil)
|
||||
@url = url
|
||||
|
||||
Site.strategies.each do |strategy|
|
||||
if strategy.url_match?(url) || strategy.url_match?(referer_url)
|
||||
@strategy = strategy.new(url, referer_url)
|
||||
break
|
||||
return
|
||||
end
|
||||
|
||||
raise NoStrategyError.new
|
||||
end
|
||||
|
||||
raise NoStrategyError.new
|
||||
end
|
||||
|
||||
def referer_url
|
||||
|
||||
@@ -6,9 +6,9 @@ module Sources::Strategies
|
||||
self.project_id(url).present?
|
||||
end
|
||||
|
||||
# https://www.artstation.com/artwork/04XA4"
|
||||
# https://dantewontdie.artstation.com/projects/YZK5q"
|
||||
# https://www.artstation.com/artwork/cody-from-sf"
|
||||
# https://www.artstation.com/artwork/04XA4
|
||||
# https://dantewontdie.artstation.com/projects/YZK5q
|
||||
# https://www.artstation.com/artwork/cody-from-sf
|
||||
def self.project_id(url)
|
||||
if url =~ %r!\Ahttps?://\w+\.artstation\.com/(?:artwork|projects)/(?<project_id>[a-z0-9-]+)\z!i
|
||||
$~[:project_id]
|
||||
|
||||
@@ -11,7 +11,7 @@ module Downloads
|
||||
end
|
||||
|
||||
should "download the large image instead" do
|
||||
assert_equal("https://cdnb3.artstation.com/p/assets/images/images/003/716/071/original/aoi-ogata-hate-city.jpg?1476754974", @download.source)
|
||||
assert_equal("https://cdnb3.artstation.com/p/assets/images/images/003/716/071/large/aoi-ogata-hate-city.jpg?1476754974", @download.source)
|
||||
end
|
||||
end
|
||||
|
||||
@@ -37,7 +37,7 @@ module Downloads
|
||||
end
|
||||
|
||||
should "download the original image instead" do
|
||||
assert_equal("https://cdna.artstation.com/p/assets/images/images/006/066/534/original/yinan-cui-reika.jpg?1495781565", @download.source)
|
||||
assert_equal("https://cdna.artstation.com/p/assets/images/images/006/066/534/large/yinan-cui-reika.jpg?1495781565", @download.source)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
@@ -11,7 +11,7 @@ module Downloads
|
||||
end
|
||||
|
||||
should "set the html page as the source" do
|
||||
assert_equal("http://orig00.deviantart.net/82ef/f/2016/271/7/1/aaaaaa_by_starbitt-daj8b46.gif", @download.source)
|
||||
assert_equal("https://orig00.deviantart.net/82ef/f/2016/271/7/1/aaaaaa_by_starbitt-daj8b46.gif", @download.source)
|
||||
end
|
||||
|
||||
should "work" do
|
||||
|
||||
Reference in New Issue
Block a user