fantasticbin/go-crawler-anime
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加章节自动获取机制

Browse Source
This commit is contained in:
fantasticbin 2024-04-04 10:11:13 +08:00
parent f496801c4d
commit 0776f86f7e
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/cmd/root.go
View File

@ -32,5 +32,5 @@ func CommandInit(rootCmd *cobra.Command) {
rootCmd.PersistentFlags().IntVarP(&config.max, "max", "m", 1, "图片最大值") rootCmd.PersistentFlags().IntVarP(&config.max, "max", "m", 1, "图片最大值")
rootCmd.PersistentFlags().StringVarP(&config.output, "output", "o", "anime", "设置漫画抓取结果的保存位置，默认为当前用户的主目录下的 anime 文件夹") rootCmd.PersistentFlags().StringVarP(&config.output, "output", "o", "anime", "设置漫画抓取结果的保存位置，默认为当前用户的主目录下的 anime 文件夹")
rootCmd.PersistentFlags().StringVarP(&config.folderTitleUrl, "folderTitleUrl", "t", "", "文件夹标题抓取链接") rootCmd.PersistentFlags().StringVarP(&config.folderTitleUrl, "folderTitleUrl", "t", "", "文件夹标题抓取链接")
rootCmd.PersistentFlags().IntVarP(&config.chapter, "chapter", "c", 1, "指定章节，用于文件夹标题抓取") rootCmd.PersistentFlags().IntVarP(&config.chapter, "chapter", "c", 0, "指定章节，用于文件夹标题抓取")
} }

7
src/cmd/start.go
View File

@ -24,6 +24,13 @@ func NewCrawler() *Crawler {
func (c *Crawler) Start() { func (c *Crawler) Start() {
c.path = config.output c.path = config.output
if config.folderTitleUrl != "" { if config.folderTitleUrl != "" {
if config.chapter == 0 {
// 章节自动获取
urlSlice := strings.Split(config.url, "_")
urlSlice = strings.Split(urlSlice[0], "/")
config.chapter, _ = strconv.Atoi(urlSlice[len(urlSlice)-1])
}
chapterTitle := FindChapterTitle(config.folderTitleUrl) chapterTitle := FindChapterTitle(config.folderTitleUrl)
title := strings.Join([]string{ title := strings.Join([]string{
"第", "第",