metaforecast/src/goodjudmentopen-fetch.js
2021-01-12 13:43:41 +01:00

145 lines
4.0 KiB
JavaScript

/* Imports */
import fs from 'fs'
import axios from "axios"
/* Definitions */
let htmlEndPoint = 'https://www.gjopen.com/questions?page='
/* Support functions */
function getcookie(){
try {
let rawcookie = fs.readFileSync("./src/privatekeys.json")
let cookie = JSON.parse(rawcookie).goodjudmentopencookie
if(cookie == undefined){
throw new Error('No cookie for Good Judgment Open!');
}
return cookie
} catch(error) {
console.log("Error: No cookies for Good Judgment Open on src/privatekeys.json! See the README.md")
process.exit()
}
}
async function fetchPage(page, cookie){
let response = await axios({
url: htmlEndPoint+page,
method: 'GET',
headers: ({
'Content-Type': 'text/html',
'Cookie': cookie
}),
})
.then(res => res.data)
//console.log(response)
return response
}
async function fetchStats(questionUrl, cookie){
let response = await axios({
url: questionUrl+"/stats",
method: 'GET',
headers: ({
'Content-Type': 'text/html',
'Cookie': cookie,
'Referer': questionUrl,
}),
})
.then(res => res.data)
//console.log(response)
// Is binary?
let isbinary = response.includes("binary?":true")
let percentage = "none"
if(isbinary){
// Crowd percentage
let htmlElements = response.split("\n")
let h3Element = htmlElements.filter(str => str.includes("h3"))[0]
let crowdpercentage = h3Element.split(">")[1].split("<")[0]
percentage = crowdpercentage
}
// Number of forecasts
let numforecasts = response.split("prediction_sets_count&quot;:")[1].split(",")[0]
//console.log(numforecasts)
// Number of predictors
let numforecasters = response.split("predictors_count&quot;:")[1].split(",")[0]
//console.log(numpredictors)
let result = {
"Binary question?": isbinary,
"Percentage": percentage,
"# Forecasts": numforecasts,
"# Forecasters": numforecasters
}
return result
}
function isEnd(html){
return html.includes("No questions match your filter")
}
function sleep(ms) {
return new Promise(resolve => setTimeout(resolve, ms));
}
/* Body */
export async function goodjudgmentopen(){
let cookie = getcookie()
let i=1
let response = await fetchPage(i, cookie)
let questions = []
let init = Date.now()
console.log("Downloading... This might take a couple of minutes. Results will be shown.")
while(!isEnd(response)){
console.log(`Page #${i}`)
let htmlLines = response.split("\n")
let h5elements = htmlLines.filter(str => str.includes("<h5><a href="))
for(let h5element of h5elements){
let h5elementSplit = h5element.split('"><span>')
let url = h5elementSplit[0].split('<a href="')[1]
let title = h5elementSplit[1].replace('</span></a></h5>', "")
await sleep(1000 + Math.random()*1000) // don't be as noticeable
try{
let moreinfo = await fetchStats(url, cookie)
if(moreinfo.isbinary){
if(!moreinfo.crowdpercentage){ // then request again.
moreinfo = await fetchStats(url, cookie)
}
}
let question = ({
"Title": title,
"URL": url,
"Platform": "Good Judgment Open",
...moreinfo
})
console.log(question)
questions.push(question)
} catch(error){
console.log(`We encountered some error when fetching the URL: ${url}, so it won't appear on the final json`)
}
}
i=i+1
console.log("Sleeping for 5secs so as to not be as noticeable to the gjopen servers")
await sleep(5000 + Math.random()*1000) // don't be a dick to gjopen server
try{
response = await fetchPage(i,cookie)
}catch(error){
console.log(`We encountered some error when fetching page #${i}, so it won't appear on the final json`)
}
}
let string = JSON.stringify(questions,null, 2)
fs.writeFileSync('./data/goodjudmentopen-questions.json', string);
let end = Date.now()
let difference = end-init
console.log(`Took ${difference/1000} seconds, or ${difference/(1000*60)} minutes.`)
}