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

feat: check all in dir

Browse Source
This commit is contained in:
Simon Ding
2024-07-14 20:24:33 +08:00
parent 6a5ff9707e
commit 6f91157d40
5 changed files with 122 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
db/db.go
View File

@@ -5,6 +5,7 @@ import (
"encoding/json"
"fmt"
"os"
"path/filepath"
"polaris/ent"
"polaris/ent/downloadclients"
"polaris/ent/episode"
@@ -14,6 +15,8 @@ import (
"polaris/ent/settings"
"polaris/ent/storage"
"polaris/log"
"polaris/pkg/utils"
"strings"
"time"
"entgo.io/ent/dialect"
@@ -86,6 +89,16 @@ func (c *Client) AddWatchlist(storageId int, nameCn, nameEn string, detail *tmdb
storageId = r.ID
}
}
st := c.GetStorage(storageId)
targetDir := fmt.Sprintf("%s %s (%v)", nameCn, nameEn, strings.Split(detail.FirstAirDate, "-")[0])
if !utils.IsChineseChar(nameCn) {
log.Warnf("name cn is not chinese name: %v", nameCn)
targetDir = fmt.Sprintf("%s (%v)", nameEn, strings.Split(detail.FirstAirDate, "-")[0])
}
targetDir = filepath.Join(st.GetPath(), targetDir)
r, err := c.ent.Series.Create().
SetTmdbID(int(detail.ID)).
@@ -97,6 +110,7 @@ func (c *Client) AddWatchlist(storageId int, nameCn, nameEn string, detail *tmdb
SetPosterPath(detail.PosterPath).
SetAirDate(detail.FirstAirDate).
SetResolution(res.String()).
SetTargetDir(targetDir).
AddEpisodeIDs(episodes...).
Save(context.TODO())
return r, err
@@ -311,7 +325,12 @@ func (s *Storage) ToWebDavSetting() WebdavSetting {
var webdavSetting WebdavSetting
json.Unmarshal([]byte(s.Settings), &webdavSetting)
return webdavSetting
}
func (s *Storage) GetPath() string {
var m map[string]string
json.Unmarshal([]byte(s.Settings), &m)
return m["path"]
}
func (c *Client) GetStorage(id int) *Storage {
@@ -387,3 +406,12 @@ func (c *Client) GetDownloadDir() string {
}
return r.Value
}
func (c *Client) UpdateEpisodeFile(seriesID int, seasonNum, episodeNum int, file string) error {
ep, err := c.ent.Episode.Query().Where(episode.SeriesID(seriesID)).Where(episode.EpisodeNumber(episodeNum)).
Where(episode.SeasonNumber(seasonNum)).First(context.TODO())
if err != nil {
return errors.Wrap(err, "finding episode")
}
return ep.Update().SetFileInStorage(file).Exec(context.TODO())
}