gabehf/sonarr-anime-importer
1
0
Fork 0
mirror of https://github.com/gabehf/sonarr-anime-importer.git synced 2026-03-19 03:56:25 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix limit param

Browse source
This commit is contained in:
Gabe Farrell 2025-03-04 11:11:56 -05:00
parent 78c2173197
commit 84b1184061
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
main.go
View file

@ -48,9 +48,15 @@ func handleAnimeSearch(malToTvdb map[int]int) func(w http.ResponseWriter, r *htt
func getAnimeSearch(malToTvdb map[int]int, r *http.Request) (string, error) { func getAnimeSearch(malToTvdb map[int]int, r *http.Request) (string, error) {
q := r.URL.Query() q := r.URL.Query()
limit, err := strconv.Atoi(q.Get("limit"))
if err != nil {
limit = 9999 // limit not specified or invalid
}
hasNextPage := true hasNextPage := true
page := 0 page := 0
resp := []ResponseItem{} resp := []ResponseItem{}
count := 0
for hasNextPage { for hasNextPage {
page++ page++
q.Set("page", strconv.Itoa(page)) q.Set("page", strconv.Itoa(page))
@ -62,6 +68,10 @@ func getAnimeSearch(malToTvdb map[int]int, r *http.Request) (string, error) {
// map the data // map the data
for _, item := range result.Data { for _, item := range result.Data {
count++
if count > limit {
break
}
resp = append(resp, resp = append(resp,
ResponseItem{ ResponseItem{
item.Title, item.Title,
@ -70,6 +80,9 @@ func getAnimeSearch(malToTvdb map[int]int, r *http.Request) (string, error) {
}) })
} }
hasNextPage = result.Pagination.HasNextPage hasNextPage = result.Pagination.HasNextPage
if count > limit {
break
}
if hasNextPage { if hasNextPage {
time.Sleep(1 * time.Second) // sleep between requests for new page to try and avoid rate limits time.Sleep(1 * time.Second) // sleep between requests for new page to try and avoid rate limits
} }