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

sources: use http client from base strategy.

Browse Source
This commit is contained in:
evazion
2020-08-17 15:30:01 -05:00
parent e196514de8
commit 160b099445
8 changed files with 12 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/logical/sources/strategies/nijie.rb
View File

@@ -190,10 +190,10 @@ module Sources
memoize :page
def client
http = Danbooru::Http.new.timeout(60)
nijie = http.timeout(60).use(retriable: { max_retries: 20 })
cookie = Cache.get("nijie-session-cookie", 1.week) do
login_page = http.use(retriable: { max_retries: 20 }).get("https://nijie.info/login.php").parse
login_page = nijie.get("https://nijie.info/login.php").parse
form = {
email: Danbooru.config.nijie_login,
password: Danbooru.config.nijie_password,
@@ -201,14 +201,14 @@ module Sources
save: "on",
ticket: ""
}
response = http.use(retriable: { max_retries: 20 }).post("https://nijie.info/login_int.php", form: form)
response = nijie.post("https://nijie.info/login_int.php", form: form)
DanbooruLogger.info "Nijie login failed (#{url}, #{response.status})" if response.status != 200
return nil unless response.status == 200
response.cookies.select { |c| c.name == "NIJIEIJIEID" }.compact.first
end
http.cookies(NIJIEIJIEID: cookie, R18: 1)
nijie.cookies(NIJIEIJIEID: cookie, R18: 1)
end
memoize :client
end