25 lines
538 B
Go
25 lines
538 B
Go
package extract
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/PuerkitoBio/goquery"
|
|
)
|
|
|
|
var ErrAttrNotExists = errors.New("attribute not exist")
|
|
|
|
// Attr returns HTML attribute value of provided selector
|
|
type Attr struct {
|
|
Name string
|
|
Selector string
|
|
Attr string
|
|
}
|
|
|
|
// Extract returns HTML attribute value of provided selector
|
|
func (e *Attr) Extract(doc *goquery.Document) (interface{}, error) {
|
|
attr, exists := doc.Find(e.Selector).Attr(e.Attr)
|
|
if !exists {
|
|
return nil, ErrAttrNotExists
|
|
}
|
|
return map[string]string{e.Name: attr}, nil
|
|
}
|