Add import-magnetico-db
For importing https://github.com/boramalper/magnetico/issues/218pull/27/head
parent
a6715616fb
commit
b6537dbcea
@ -0,0 +1,92 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"encoding/csv"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
"unicode/utf8"
|
||||
|
||||
"github.com/lib/pq"
|
||||
)
|
||||
|
||||
func main() {
|
||||
f, err := os.Open(os.Args[1])
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
db := initDb()
|
||||
|
||||
cr := csv.NewReader(f)
|
||||
cr.LazyQuotes = true
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
for {
|
||||
line, err := cr.Read()
|
||||
if err == io.EOF {
|
||||
break
|
||||
} else if err != nil {
|
||||
if perr, ok := err.(*csv.ParseError); ok && perr.Err == csv.ErrFieldCount {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
infohash := line[0]
|
||||
if len(infohash) != 40 {
|
||||
log.Fatal("bad infohash length " + line[0])
|
||||
}
|
||||
|
||||
name := line[1]
|
||||
if len(name) < 2 {
|
||||
log.Println("bad name length " + line[1])
|
||||
continue
|
||||
}
|
||||
if !utf8.ValidString(name) {
|
||||
log.Println("utf8 invalid name")
|
||||
log.Println(name)
|
||||
continue
|
||||
}
|
||||
|
||||
length := line[2]
|
||||
|
||||
addedUnix, err := strconv.ParseInt(line[3], 10, 0)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
added := time.Unix(addedUnix, 0)
|
||||
|
||||
//fmt.Printf("Ih %v name %v len %v added %v", infohash, name, length, added)
|
||||
_, err = db.Exec("INSERT INTO torrent (infohash, name, length, added) VALUES ($1, $2, $3, $4)", infohash, name, length, added)
|
||||
if err, ok := err.(*pq.Error); ok { //dark magic
|
||||
if err.Code != "23505" {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func initDb() *sql.DB {
|
||||
connStr := "user=nextgen dbname=nextgen host=/var/run/postgresql"
|
||||
db, err := sql.Open("postgres", connStr)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
_, err = db.Exec(`CREATE TABLE IF NOT EXISTS torrent (
|
||||
infohash char(40) PRIMARY KEY NOT NULL,
|
||||
name varchar NOT NULL,
|
||||
length bigint,
|
||||
added timestamp DEFAULT current_timestamp
|
||||
)`)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
return db
|
||||
}
|
Loading…
Reference in New Issue