mirror of
https://github.com/idanoo/autobrr
synced 2025-07-22 16:29:12 +00:00
feat(filters): smart episode (#563)
* feat(filters): initial smart episode * feat(smart-episode): pseudo-logic * feat(filters): check releases * feat(filters): update logic * feat(web): smart episode (#562) * add frontend part for smart episode feature * change description for smart episode help text * fix wording * feat(filters): smart-episode logic Co-authored-by: Kyle Sanderson <kyle.leet@gmail.com> Co-authored-by: xoaaC <35452459+xoaaC@users.noreply.github.com>
This commit is contained in:
parent
45e03c10b6
commit
38795be9ea
11 changed files with 132 additions and 17 deletions
|
@ -447,3 +447,70 @@ func (repo *ReleaseRepo) Delete(ctx context.Context) error {
|
|||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (repo *ReleaseRepo) CanDownloadShow(ctx context.Context, title string, season int, episode int) (bool, error) {
|
||||
// TODO support non season episode shows
|
||||
// if rls.Day > 0 {
|
||||
// // Maybe in the future
|
||||
// // SELECT '' FROM release WHERE Title LIKE %q AND ((Year == %d AND Month == %d AND Day > %d) OR (Year == %d AND Month > %d) OR (Year > %d))"
|
||||
// qs := sql.Query("SELECT torrent_name FROM release WHERE Title LIKE %q AND Year >= %d", rls.Title, rls.Year)
|
||||
//
|
||||
// for q := range qs.Rows() {
|
||||
// r := rls.ParseTitle(q)
|
||||
// if r.Year > rls.Year {
|
||||
// return false, fmt.Errorf("stale release year")
|
||||
// }
|
||||
//
|
||||
// if r.Month > rls.Month {
|
||||
// return false, fmt.Errorf("stale release month")
|
||||
// }
|
||||
//
|
||||
// if r.Month == rls.Month && r.Day > rls.Day {
|
||||
// return false, fmt.Errorf("stale release day")
|
||||
// }
|
||||
// }
|
||||
//}
|
||||
|
||||
queryBuilder := repo.db.squirrel.
|
||||
Select("COUNT(*)").
|
||||
From("release").
|
||||
Where("title LIKE ?", fmt.Sprint("%", title, "%"))
|
||||
|
||||
if season > 0 && episode > 0 {
|
||||
queryBuilder = queryBuilder.Where(sq.Or{
|
||||
sq.And{
|
||||
sq.Eq{"season": season},
|
||||
sq.Gt{"episode": episode},
|
||||
},
|
||||
sq.Gt{"season": season},
|
||||
})
|
||||
} else if season > 0 && episode == 0 {
|
||||
queryBuilder = queryBuilder.Where(sq.Gt{"season": season})
|
||||
} else {
|
||||
/* No support for this scenario today. Specifically multi-part specials.
|
||||
* The Database presently does not have Subtitle as a field, but is coming at a future date. */
|
||||
return true, nil
|
||||
}
|
||||
|
||||
query, args, err := queryBuilder.ToSql()
|
||||
if err != nil {
|
||||
return false, errors.Wrap(err, "error building query")
|
||||
}
|
||||
|
||||
row := repo.db.handler.QueryRowContext(ctx, query, args...)
|
||||
if err := row.Err(); err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
var count int
|
||||
|
||||
if err := row.Scan(&count); err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
if count > 0 {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
return true, nil
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue