add option to get multiple results
All checks were successful
Gitnet/scraper/pipeline/head This commit looks good
All checks were successful
Gitnet/scraper/pipeline/head This commit looks good
This commit is contained in:
parent
bdd90b0e09
commit
b680003549
|
@ -38,6 +38,8 @@ const options = {
|
|||
method: 'GET', // Optional, default is `GET`
|
||||
}
|
||||
|
||||
const isMultiple = false // get the first result, `true` to get an array of results
|
||||
|
||||
const selector = '.repository-content .numbers-summary li:nth-child(4) a'
|
||||
|
||||
const filters = {
|
||||
|
@ -56,7 +58,8 @@ scraper(
|
|||
},
|
||||
function(error) {
|
||||
console.log(error)
|
||||
}
|
||||
},
|
||||
isMultiple
|
||||
)
|
||||
```
|
||||
|
||||
|
|
14
src/cli.js
14
src/cli.js
|
@ -34,7 +34,8 @@ Optional parameters
|
|||
--method [METHOD] HTTP Method
|
||||
--accept-http-error Accepts all status codes (like 404)
|
||||
|
||||
--verbose, -v Show message of error
|
||||
--verbose, -v Show message of error
|
||||
--multiple, -m The output must contain all the selector targets
|
||||
`
|
||||
|
||||
if (input.has('help')) {
|
||||
|
@ -62,6 +63,7 @@ const selector = input.get('selector')
|
|||
const method = input.has('method') ? input.get('method') : 'GET'
|
||||
const acceptAllStatus = input.has('accept-http-error')
|
||||
const verbose = input.has('verbose') || input.has('v')
|
||||
const isMultiple = input.has('multiple') || input.has('m')
|
||||
|
||||
let filtersToApply = {}
|
||||
|
||||
|
@ -86,7 +88,13 @@ const options = {
|
|||
}
|
||||
|
||||
const onSuccess = function(value) {
|
||||
output.write(value)
|
||||
if (isMultiple && value instanceof Array) {
|
||||
for (let item of value) {
|
||||
output.write(item)
|
||||
}
|
||||
} else {
|
||||
output.write(value)
|
||||
}
|
||||
}
|
||||
|
||||
const onError = function(error) {
|
||||
|
@ -99,4 +107,4 @@ const onError = function(error) {
|
|||
process.exit(1)
|
||||
}
|
||||
|
||||
scraper(options, selector, filtersToApply, onSuccess, onError)
|
||||
scraper(options, selector, filtersToApply, onSuccess, onError, isMultiple)
|
||||
|
|
22
src/index.js
22
src/index.js
|
@ -7,7 +7,7 @@ const filters = {
|
|||
trim: require('./filter/trim'),
|
||||
}
|
||||
|
||||
const scraper = function(options, selector, filtersToApply, callbackSuccess, callbackError) {
|
||||
const scraper = function(options, selector, filtersToApply, callbackSuccess, callbackError, isMultiple) {
|
||||
filtersToApply = filtersToApply || {}
|
||||
|
||||
rq({
|
||||
|
@ -17,10 +17,26 @@ const scraper = function(options, selector, filtersToApply, callbackSuccess, cal
|
|||
})
|
||||
.then(function(body) {
|
||||
const $ = cheerio.load(body)
|
||||
let value = $(selector).html()
|
||||
let value = []
|
||||
|
||||
if (isMultiple) {
|
||||
let nodes = $(selector)
|
||||
|
||||
nodes.each(function(i, node) {
|
||||
value.push($(node).html())
|
||||
})
|
||||
} else {
|
||||
value = $(selector).html()
|
||||
}
|
||||
|
||||
for (let filter in filtersToApply) {
|
||||
value = filters[filter](value, filtersToApply[filter])
|
||||
if (isMultiple) {
|
||||
for (let i in value) {
|
||||
value[i] = filters[filter](value[i], filtersToApply[filter])
|
||||
}
|
||||
} else {
|
||||
value = filters[filter](value, filtersToApply[filter])
|
||||
}
|
||||
}
|
||||
|
||||
if (callbackSuccess) {
|
||||
|
|
Loading…
Reference in a new issue