def edit_serie(req, serie):
if req.POST:
episode = req.POST.get('episode', None)
if episode and serie:
s = Serie.get_by(name=serie)
e = Episode(name=episode)
s.episodes.append(e)
session.commit()
return Redirect(full_path_redirect(req.ENV, "/series/edit/%s" % serie))
s = Serie.get_by(name=serie)
return render_to_response('episode_new.html', {'serie': s})
def test_list_series(self):
with Mock() as Serie:
from tvseries.model import Serie
s = Serie.query.order_by(Serie.name).all() >> ["dexter", "house"]
with Mock() as render_to_response:
from tvseries.templates import render_to_response
render_to_response('series.html', data={'series':["dexter", "house"]})
tvseries.routes.series(Request({}))
Serie.validate()
render_to_response.validate()
def delete_episode(req, serie, episode):
if serie and episode:
s = Serie.get_by(name=serie)
e = Episode.get_by(serie=s, name=episode)
s.episodes.remove(e)
session.commit()
return Redirect(full_path_redirect(req.ENV, "/series/edit/%s" % serie))
def next_episode(req, serie):
if serie and req.POST:
s = Serie.get_by(name=serie)
last_ep = s.last_episode()
if last_ep:
next = last_ep.next()
if next:
s.episodes.append(next)
session.commit()
return render_to_response('episode_new.html', {'serie': s})
return Redirect(full_path_redirect(req.ENV, '/'))
def test_get_last_episode_empty_serie(self):
s = Serie(name="dexter", episodes=[Episode(name="s01e02"), Episode(name="s01e05")])
self.assertEquals("s01e05", s.last_episode().name)
def test_get_last_episode(self):
s = Serie(name="dexter")
self.assertEquals("s01e00", s.last_episode().name)
def delete_serie(req, serie):
s = Serie.get_by(name=serie)
if s:
s.delete()
session.commit()
return Redirect(full_path_redirect(req.ENV, "/series"))
