jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-09 19:47:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: get correct movie episode id

Browse Source
This commit is contained in:
Simon Ding
2025-04-20 10:41:58 +08:00
parent ec3f9b2f96
commit 8e9ba101c8
2 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
engine/resources.go
View File

@@ -165,7 +165,7 @@ func (c *Engine) downloadTorrent(m *ent.Media, r1 torznab.Result, seasonNum int,
c.db.SetSeasonAllEpisodeStatus(m.ID, seasonNum, episode.StatusDownloading) c.db.SetSeasonAllEpisodeStatus(m.ID, seasonNum, episode.StatusDownloading)
} }
} else { } else {//movie download
ep, _ := c.db.GetMovieDummyEpisode(m.ID) ep, _ := c.db.GetMovieDummyEpisode(m.ID)
if ep.Status == episode.StatusMissing { if ep.Status == episode.StatusMissing {
c.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading) c.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)

34
engine/scheduler.go
View File

@@ -204,35 +204,37 @@ func getSeasonNum(h *ent.History) int {
} }
func (c *Engine) GetEpisodeIds(r *ent.History) []int { func (c *Engine) GetEpisodeIds(r *ent.History) []int {
var episodeIds []int
seasonNum := getSeasonNum(r)
// if r.EpisodeID > 0 {
// episodeIds = append(episodeIds, r.EpisodeID)
// }
series, err := c.db.GetMediaDetails(r.MediaID) series, err := c.db.GetMediaDetails(r.MediaID)
if err != nil { if err != nil {
log.Errorf("get media details error: %v", err) log.Errorf("get media details error: %v", err)
return []int{} return []int{}
} }
if series.MediaType == media.MediaTypeMovie { //movie
ep, _ := c.db.GetMovieDummyEpisode(series.ID)
return []int{ep.ID}
} else { //tv
var episodeIds []int
seasonNum := getSeasonNum(r)
if len(r.EpisodeNums) > 0 { if len(r.EpisodeNums) > 0 {
for _, epNum := range r.EpisodeNums { for _, epNum := range r.EpisodeNums {
for _, ep := range series.Episodes {
if ep.SeasonNumber == seasonNum && ep.EpisodeNumber == epNum {
episodeIds = append(episodeIds, ep.ID)
}
}
}
} else {
for _, ep := range series.Episodes { for _, ep := range series.Episodes {
if ep.SeasonNumber == seasonNum && ep.EpisodeNumber == epNum { if ep.SeasonNumber == seasonNum {
episodeIds = append(episodeIds, ep.ID) episodeIds = append(episodeIds, ep.ID)
} }
} }
}
} else {
for _, ep := range series.Episodes {
if ep.SeasonNumber == seasonNum {
episodeIds = append(episodeIds, ep.ID)
}
}
}
return episodeIds
} }
return episodeIds
} }
func (c *Engine) moveCompletedTask(id int) (err1 error) { func (c *Engine) moveCompletedTask(id int) (err1 error) {