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
