ODWRTW/canape
2
2
Fork 0
Code Issues 3 Pull Requests 1 Releases Activity

Update Show.IsTracked

Browse Source
This commit is contained in:
Nicolas Duhamel 2016-02-16 20:17:15 +01:00
parent 6234231e71
commit d0bf7b4354
2 changed files with 39 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
shows.go
View File

@ -44,7 +44,9 @@ var Schema = sqly.Schema{
Sql: ` Sql: `
CREATE TABLE shows_tracked ( CREATE TABLE shows_tracked (
shows_id integer NOT NULL REFERENCES shows (id) ON DELETE CASCADE, shows_id integer NOT NULL REFERENCES shows (id) ON DELETE CASCADE,
users_id integer NOT NULL REFERENCES users (id) ON DELETE CASCADE users_id integer NOT NULL REFERENCES users (id) ON DELETE CASCADE,
season integer NOT NULL,
episode integer NOT NULL
); );
`}, `},
}, },
@ -64,17 +66,22 @@ const (
getEpisodesQuery = `SELECT title, season, episode FROM episodes WHERE shows_id=$1;` getEpisodesQuery = `SELECT title, season, episode FROM episodes WHERE shows_id=$1;`
getShowWithUserQuery = ` getShowWithUserQuery = `
SELECT id, imdbid, title, SELECT
EXISTS (SELECT 1 FROM shows_tracked WHERE shows_id=shows.id AND users_id=$2) AS tracked shows.id,
FROM shows WHERE imdbid=$1; shows.title,
` shows.imdbid,
COALESCE(shows_tracked.season,0) AS trackedseason,
COALESCE(shows_tracked.episode,0) AS trackedepisode
FROM shows LEFT JOIN shows_tracked ON shows.id=shows_tracked.shows_id AND shows_tracked.users_id=$2
WHERE shows.imdbid=$1;`
) )
type Show struct { type Show struct {
sqly.BaseTable sqly.BaseTable
polochon.Show polochon.Show
Episodes []*Episode Episodes []*Episode
Tracked bool TrackedSeason int
TrackedEpisode int
} }
func Get(db *sqlx.DB, imdbID string) (*Show, error) { func Get(db *sqlx.DB, imdbID string) (*Show, error) {
@ -96,6 +103,27 @@ func GetAsUser(db *sqlx.DB, user *users.User, imdbID string) (*Show, error) {
return s, nil return s, nil
} }
// GetDetails retrieves details for the show, first try to
// get info from db, if not exists, use polochon.Detailer
// and save informations in the database for future use
func (s *Show) GetDetails(db *sqlx.DB) error {
return nil
}
// GetDetailsAsUser like GetDetails but with User context
func (s *Show) GetDetailsAsUser(db *sqlx.DB, user *users.User) error {
return nil
}
// IsTracked returns true if the show is tracked use this function
// after retrieve the show with GetAsUser or other *AsUser functions
func (s *Show) IsTracked() bool {
if s.TrackedSeason != 0 && s.TrackedEpisode != 0 {
return true
}
return false
}
func (s *Show) Add(db *sqlx.DB) error { func (s *Show) Add(db *sqlx.DB) error {
var id int var id int
r, err := db.NamedQuery(addShowQuery, s) r, err := db.NamedQuery(addShowQuery, s)

8
shows_test.go
View File

@ -160,12 +160,12 @@ func TestTrackedShow(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if show1.Tracked { if show1.IsTracked() {
t.Fatal("Tracked must be false here") t.Fatal("Tracked must be false here")
} }
q := `INSERT INTO shows_tracked (shows_id, users_id) VALUES ($1, $2);` q := `INSERT INTO shows_tracked (shows_id, users_id, season, episode) VALUES ($1, $2, $3, $4);`
_, err = db.Exec(q, show1.ID, u.ID) _, err = db.Exec(q, show1.ID, u.ID, 1, 1)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -173,7 +173,7 @@ func TestTrackedShow(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if !show2.Tracked { if !show2.IsTracked() {
t.Fatal("Tracked must be true here") t.Fatal("Tracked must be true here")
} }
}) })