kuiki
/
bilibili-rename
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

添加skip的集

master
kuiki 2019-02-19 11:43:51 +08:00
parent 2c592ed54e
commit 6799e50ab1
1 changed files with 22 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
main.go
View File

@ -3,6 +3,8 @@ package main
import (
"fmt"
"strconv"
"github.com/yinhui87/go-component/language"
)
type episode struct {
@ -13,17 +15,17 @@ type episode struct {
func main() {
episodeProvider := new(mockProvider)
// epdatas, err := getEPData(episodeProvider, "av1742161")
epdatas, err := getEPData(episodeProvider, "ep173262")
epdatas, err := getEPData(episodeProvider, "av1742161")
// epdatas, err := getEPData(episodeProvider, "ep173262")
if err != nil {
panic(err)
}
for i, epdata := range epdatas {
fmt.Printf("ep %d Cid: %v, Title: %s\n", i, epdata.Cid, epdata.Title)
fmt.Printf("%dep %d Cid: %v, Title: %s\n", i, epdata.Index, epdata.Cid, epdata.Title)
}
}
func getEPData(provider EpisodeDataProvider, videoNo string) (episodes []episode, err error) {
func getEPData(provider EpisodeDataProvider, videoNo string, skipIndex ...int64) (episodes []episode, err error) {
videoType := videoNo[0:2]
videoID, err := strconv.ParseInt(videoNo[2:], 10, 64)
if err != nil {
@ -37,24 +39,30 @@ func getEPData(provider EpisodeDataProvider, videoNo string) (episodes []episode
return episodes, err
}
for i, avdata := range avdatas {
index := int64(i) + 1
if language.ArrayIn(skipIndex, index) == -1 {
episodes = append(episodes, episode{
Index: int64(i),
Index: index,
Cid: avdata.Cid,
Title: avdata.Part,
})
}
}
case "ep":
epdatas, err := provider.GetEPData(videoID)
if err != nil {
return episodes, err
}
for i, epdata := range epdatas {
index := int64(i) + 1
if language.ArrayIn(skipIndex, index) == -1 {
episodes = append(episodes, episode{
Index: int64(i),
Index: index,
Cid: epdata.Cid,
Title: epdata.LongTitle,
})
}
}
}
return
}