first commit
This commit is contained in:
Peter Kleiweg
126
NU/cmd/metadata/metadata.go
Normal file
126
NU/cmd/metadata/metadata.go
Normal file
@@ -0,0 +1,126 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"github.com/pebbe/util"
|
||||
|
||||
"bufio"
|
||||
"encoding/xml"
|
||||
"fmt"
|
||||
"html"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Item struct {
|
||||
XMLName xml.Name `xml:"item"`
|
||||
UnixTime int64 `xml:"unixTime"`
|
||||
}
|
||||
|
||||
var (
|
||||
x = util.CheckErr
|
||||
escape = html.EscapeString
|
||||
data = make(map[string][]string)
|
||||
location *time.Location
|
||||
)
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
location, err = time.LoadLocation("Europe/Amsterdam")
|
||||
x(err)
|
||||
|
||||
files, err := os.ReadDir("..")
|
||||
x(err)
|
||||
for _, file := range files {
|
||||
filename := file.Name()
|
||||
if strings.HasSuffix(filename, ".txt") {
|
||||
doText(filename)
|
||||
} else if strings.HasSuffix(filename, ".xml") {
|
||||
doXml(filename)
|
||||
}
|
||||
}
|
||||
|
||||
files, err = os.ReadDir("xml")
|
||||
x(err)
|
||||
for _, file := range files {
|
||||
filename := file.Name()
|
||||
if !strings.HasSuffix(filename, ".xml") {
|
||||
continue
|
||||
}
|
||||
aa := strings.Split(filename, ".")
|
||||
base := strings.Join(aa[1:len(aa)-2], ".")
|
||||
b, err := os.ReadFile("xml/" + filename)
|
||||
x(err)
|
||||
s := string(b)
|
||||
i := strings.Index(s, "<alpino") + 1
|
||||
i += strings.Index(s[i:], "<")
|
||||
fp, err := os.Create("xml/" + filename + ".tmp")
|
||||
x(err)
|
||||
_, err = fp.WriteString(s[:i])
|
||||
x(err)
|
||||
_, err = fp.WriteString("<metadata>\n <meta type=\"text\" name=\"source\" value=\"NU\"/>\n")
|
||||
x(err)
|
||||
for _, m := range data[base] {
|
||||
_, err = fp.WriteString(" " + m + "\n")
|
||||
x(err)
|
||||
}
|
||||
_, err = fp.WriteString(" </metadata>\n ")
|
||||
x(err)
|
||||
_, err = fp.WriteString(stripMeta(s[i:]))
|
||||
x(err)
|
||||
x(fp.Close())
|
||||
x(os.Rename("xml/"+filename+".tmp", "xml/"+filename))
|
||||
}
|
||||
}
|
||||
|
||||
func doText(filename string) {
|
||||
base := filename[:len(filename)-4]
|
||||
if _, ok := data[base]; !ok {
|
||||
data[base] = make([]string, 0)
|
||||
}
|
||||
fp, err := os.Open("../" + filename)
|
||||
x(err)
|
||||
defer func() { x(fp.Close()) }()
|
||||
scanner := bufio.NewScanner(fp)
|
||||
for scanner.Scan() {
|
||||
line := scanner.Text()
|
||||
if !strings.HasPrefix(line, "##META") {
|
||||
continue
|
||||
}
|
||||
aa := strings.Fields(line)
|
||||
if len(aa) > 4 {
|
||||
data[base] = append(data[base],
|
||||
fmt.Sprintf(`<meta type="%s" name="%s" value="%s"/>`,
|
||||
aa[1],
|
||||
escape(aa[2]),
|
||||
escape(strings.Join(aa[4:], " "))))
|
||||
}
|
||||
}
|
||||
x(scanner.Err())
|
||||
}
|
||||
|
||||
func doXml(filename string) {
|
||||
base := filename[:len(filename)-4]
|
||||
if _, ok := data[base]; !ok {
|
||||
data[base] = make([]string, 0)
|
||||
}
|
||||
b, err := os.ReadFile("../" + filename)
|
||||
x(err)
|
||||
var item Item
|
||||
x(xml.Unmarshal(b, &item))
|
||||
t := time.Unix(item.UnixTime, 0).In(location)
|
||||
data[base] = append(data[base],
|
||||
fmt.Sprintf(`<meta type="date" name="pubdate" value="%d-%02d-%02d"/>`,
|
||||
t.Year(),
|
||||
int(t.Month()),
|
||||
t.Day()))
|
||||
}
|
||||
|
||||
func stripMeta(s string) string {
|
||||
i1 := strings.Index(s, "<metadata>")
|
||||
if i1 < 0 {
|
||||
return s
|
||||
}
|
||||
i2 := i1 + strings.Index(s[i1:], "</metadata>") + 11
|
||||
return s[:i1] + strings.TrimLeft(s[i2:], " \t\r\n")
|
||||
}
|
||||
240
NU/cmd/nu/nu.go
Normal file
240
NU/cmd/nu/nu.go
Normal file
@@ -0,0 +1,240 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"github.com/pebbe/util"
|
||||
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"encoding/xml"
|
||||
"fmt"
|
||||
"html"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Rss struct {
|
||||
XMLName xml.Name `xml:"rss"`
|
||||
Items []ItemT `xml:"channel>item"`
|
||||
}
|
||||
|
||||
type ItemT struct {
|
||||
PubDate string `xml:"pubDate"`
|
||||
UnixTime int64 `xml:"unixTime"`
|
||||
Guid string `xml:"guid"`
|
||||
Link string `xml:"link"`
|
||||
Data []byte `xml:",innerxml"`
|
||||
}
|
||||
|
||||
type Doc struct {
|
||||
Graph []GItem `json:"@graph"`
|
||||
}
|
||||
|
||||
type GItem struct {
|
||||
ArticleBody string `json:"articleBody"`
|
||||
ArticleSection []string `json:"articleSection"`
|
||||
}
|
||||
|
||||
var (
|
||||
x = util.CheckErr
|
||||
w = util.WarnErr
|
||||
agent = "AhrefsBot/7.0"
|
||||
)
|
||||
|
||||
func exists(filename string) bool {
|
||||
_, err := os.Stat(filename)
|
||||
return err == nil
|
||||
}
|
||||
|
||||
func fileDate(filename string) string {
|
||||
b, err := os.ReadFile(filename)
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
s := string(b)
|
||||
i1 := strings.Index(s, "<unixTime>") + 10
|
||||
i2 := strings.Index(s, "</unixTime>")
|
||||
return s[i1:i2]
|
||||
}
|
||||
|
||||
func main() {
|
||||
resp, err := http.Get("https://www.nu.nl/rss")
|
||||
x(err)
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
x(err)
|
||||
x(resp.Body.Close())
|
||||
|
||||
var rss Rss
|
||||
x(xml.Unmarshal(body, &rss))
|
||||
|
||||
if len(rss.Items) == 0 {
|
||||
x(fmt.Errorf("len(rss.Items) == 0"))
|
||||
}
|
||||
|
||||
for _, item := range rss.Items {
|
||||
t, err := time.Parse(time.RFC1123Z, item.PubDate)
|
||||
if err != nil {
|
||||
t, err = time.Parse(time.RFC1123, item.PubDate)
|
||||
}
|
||||
x(err)
|
||||
year, week := t.ISOWeek()
|
||||
dirname := fmt.Sprintf("/net/corpora/nlnieuws/NU/%d/%02d", year, week)
|
||||
filename := dirname + "/" + url.PathEscape(item.Guid)
|
||||
|
||||
ts := fmt.Sprintf("%d", t.Unix())
|
||||
needUpdate := fileDate(filename+".xml") != ts
|
||||
|
||||
x(os.MkdirAll(dirname, 0777))
|
||||
fp, err := os.Create(filename + ".xml")
|
||||
x(err)
|
||||
_, err = fp.WriteString("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n<item>\n")
|
||||
x(err)
|
||||
_, err = fmt.Fprintf(fp, "<unixTime>%d</unixTime>", t.Unix())
|
||||
x(err)
|
||||
_, err = fp.Write(item.Data)
|
||||
x(err)
|
||||
_, err = fp.WriteString("</item>\n")
|
||||
x(err)
|
||||
x(fp.Close())
|
||||
x(os.Chtimes(filename+".xml", t, t))
|
||||
if !doArticle(filename, item.Link, t, needUpdate) {
|
||||
x(os.Remove(filename + ".xml"))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func doArticle(filename string, url string, timestamp time.Time, needUpdate bool) bool {
|
||||
if needUpdate {
|
||||
_ = os.Remove(filename + ".err")
|
||||
_ = os.Remove(filename + ".html")
|
||||
_ = os.Remove(filename + ".json")
|
||||
_ = os.Remove(filename + ".txt")
|
||||
_ = os.Remove(filename + ".skip")
|
||||
} else {
|
||||
// voor sommige berichten is geen .txt, alleen .json
|
||||
if exists(filename+".json") || exists(filename+".skip") {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
req, err := http.NewRequest("GET", url, nil)
|
||||
x(err)
|
||||
req.Header.Set("User-Agent", agent)
|
||||
|
||||
client := &http.Client{}
|
||||
resp, err := client.Do(req)
|
||||
x(err)
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
x(err)
|
||||
x(resp.Body.Close())
|
||||
|
||||
s := string(body)
|
||||
ok := true
|
||||
i1 := strings.Index(s, `<script type="application/ld+json"`)
|
||||
if i1 < 0 {
|
||||
ok = false
|
||||
} else {
|
||||
i1 += strings.Index(s[i1:], `>`) + 1
|
||||
i2 := i1 + strings.Index(s[i1:], `</script>`)
|
||||
if i2 < i1 {
|
||||
ok = false
|
||||
} else {
|
||||
s = s[i1:i2]
|
||||
}
|
||||
}
|
||||
if !ok {
|
||||
_ = w(fmt.Errorf("script jsonld not found: %s", url))
|
||||
|
||||
fp, err := os.Create(filename + ".err")
|
||||
x(err)
|
||||
_, err = fmt.Fprintf(fp, "script jsonld not found: %s\n", url)
|
||||
x(err)
|
||||
x(fp.Close())
|
||||
x(os.Chtimes(filename+".err", timestamp, timestamp))
|
||||
|
||||
fp, err = os.Create(filename + ".html")
|
||||
x(err)
|
||||
_, err = fp.Write(body)
|
||||
x(err)
|
||||
x(fp.Close())
|
||||
x(os.Chtimes(filename+".html", timestamp, timestamp))
|
||||
|
||||
return false
|
||||
}
|
||||
fp, err := os.Create(filename + ".json")
|
||||
x(err)
|
||||
_, err = fp.WriteString(s)
|
||||
x(err)
|
||||
x(fp.Close())
|
||||
x(os.Chtimes(filename+".json", timestamp, timestamp))
|
||||
|
||||
var doc Doc
|
||||
if err = json.Unmarshal([]byte(s), &doc); err != nil {
|
||||
_ = w(err, url)
|
||||
|
||||
fp, err := os.Create(filename + ".err")
|
||||
x(err)
|
||||
_, err = fmt.Fprintf(fp, "%s: %v\n", url, err)
|
||||
x(err)
|
||||
x(fp.Close())
|
||||
x(os.Chtimes(filename+".err", timestamp, timestamp))
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
cats := make([]string, 0)
|
||||
var buffer bytes.Buffer
|
||||
for _, i := range doc.Graph {
|
||||
_, err = buffer.WriteString(html.UnescapeString(i.ArticleBody))
|
||||
x(err)
|
||||
cats = append(cats, i.ArticleSection...)
|
||||
}
|
||||
text := buffer.String()
|
||||
|
||||
// sommige berichten bevatten geen tekst, maar een video bijvoorbeeld
|
||||
// dit is geen fout
|
||||
if len(text) > 0 {
|
||||
|
||||
// text bevat kopjes zonder punt aan het eind
|
||||
lines := strings.Split(text, "\n")
|
||||
for i, line := range lines {
|
||||
n := len(line)
|
||||
if n > 0 {
|
||||
if strings.ContainsAny(line[n-1:], ".!?") {
|
||||
continue
|
||||
}
|
||||
}
|
||||
if n > 1 {
|
||||
s := line[n-2:]
|
||||
if s == `."` || s == `!"` || s == `?"` {
|
||||
continue
|
||||
}
|
||||
}
|
||||
lines[i] = line + "."
|
||||
}
|
||||
text = strings.Join(lines, "\n") + "\n"
|
||||
|
||||
fp, err := os.Create(filename + ".txt")
|
||||
x(err)
|
||||
if len(cats) == 0 {
|
||||
_, err := fmt.Fprintln(fp, "##META text cat =")
|
||||
x(err)
|
||||
} else {
|
||||
for _, cat := range cats {
|
||||
_, err := fmt.Fprintf(fp, "##META text cat = %s\n", cat)
|
||||
x(err)
|
||||
}
|
||||
}
|
||||
_, err = fp.WriteString(text)
|
||||
x(err)
|
||||
x(fp.Close())
|
||||
x(os.Chtimes(filename+".txt", timestamp, timestamp))
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
Reference in New Issue
Block a user