def faketrack(name, season, chapter): name = name.lower() try: serie = Series.get(Series.name == name) except Series.DoesNotExist: serie = Series.create(name=name, last_season=season, last_chapter=chapter, tracking=True) else: serie.tracking = True serie.save() print "Tracking %s" % name
def test_stringmaker(self): faketrack("The Wire", 1, 0) serie = Series.get(Series.name == "the wire") seriestring = self.daemon.stringmaker(serie) self.assertEqual(seriestring, ("the wire S01E01", "01", "01")) seriestring = self.daemon.stringmaker(serie, newseason=1) self.assertEqual(seriestring, (u"the wire S02E01", "01", "02"))
def fakejunky(name, season, chapter): name = name.lower() try: serie = Series.get(Series.name == name) except Series.DoesNotExist: serie = Series.create(name=name, last_season=season, last_chapter=chapter, tracking=True, junky=True) else: serie.tracking = True serie.junky = True if serie.last_season <= season: serie.last_season = season if serie.last_chapter < chapter: serie.last_chapter = chapter serie.save() finally: # No more than 1 tv series in junky mode at the time series = Series.select().where(Series.name != name, Series.junky == True) for serie in series: serie.junky = False serie.save() print "%s is not more in Junky Mode" % serie.name print "%s is on Junky Mode" % name