class Song(ContentBase):
rating = peewee.CharField(null=True)
genre = peewee.CharField(null=True)
year = peewee.CharField(null=True)
path = peewee.CharField(unique=True)
duration = peewee.FloatField(default=0)
class Rating(peewee.Model):
name = peewee.CharField(unique=True)
internal = peewee.CharField()
value = peewee.IntegerField(default=0)
system = peewee.CharField()
class Meta:
database = DB
class RatingSystem(peewee.Model):
name = peewee.CharField()
context = peewee.CharField(null=True)
regEx = peewee.CharField()
regions = peewee.CharField(null=True)
class Meta:
database = DB
def migrateDB(DB, version):
util.LOG('Migrating database from version {0} to {1}'.format(
version, DATABASE_VERSION))
from peewee.playhouse import migrate
migrator = migrate.SqliteMigrator(DB)
migratorW = migrate.SqliteMigrator(W_DB)
if version < 1:
try:
migrate.migrate(
migrator.add_column('RatingsBumpers', 'style',
peewee.CharField(default='Classic')))
except peewee.OperationalError:
util.MINOR_ERROR('Migration (RatingsBumpers: Add style column)')
except:
util.ERROR()
return False
if version < 2:
try:
migrate.migrate(
migrator.add_column('RatingSystem', 'regions',
peewee.CharField(null=True)),
migrator.drop_not_null('RatingSystem', 'context'),
)
except peewee.OperationalError:
util.MINOR_ERROR('Migration (RatingSystem: Add region column)')
except:
util.ERROR()
return False
if version < 5:
try:
migrate.migrate(
migratorW.add_column('Trailers', 'is3D',
peewee.BooleanField(default=False)), )
except peewee.OperationalError:
util.MINOR_ERROR('Migration (Trailers: Add is3D column)')
except:
util.ERROR()
return False
if version < 6:
try:
migrate.migrate(
migratorW.add_column('Trailers', 'verified',
peewee.BooleanField(default=True)), )
except peewee.OperationalError:
util.MINOR_ERROR('Migration (Trailers: Add verified column)')
except:
util.ERROR()
return False
return True
class WatchedBase(peewee.Model):
WID = peewee.CharField(unique=True)
watched = peewee.BooleanField(default=False)
date = peewee.DateTimeField(default=datetime.date(1900, 1, 1))
class Meta:
database = W_DB
class ContentBase(peewee.Model):
name = peewee.CharField()
accessed = peewee.DateTimeField(null=True)
pack = peewee.TextField(null=True)
class Meta:
database = DB
class Trailers(WatchedBase):
source = peewee.CharField()
rating = peewee.CharField(null=True)
genres = peewee.CharField(null=True)
title = peewee.CharField()
release = peewee.DateTimeField(default=datetime.date(1900, 1, 1))
url = peewee.CharField(null=True)
userAgent = peewee.CharField(null=True)
thumb = peewee.CharField(null=True)
broken = peewee.BooleanField(default=False)
is3D = peewee.BooleanField(default=False)
verified = peewee.BooleanField(default=True)
class VideoBumpers(BumperBase):
type = peewee.CharField()
rating = peewee.CharField(null=True)
genre = peewee.CharField(null=True)
year = peewee.CharField(null=True)
class RatingsBumpers(BumperBase):
system = peewee.CharField(default='MPAA')
style = peewee.CharField(default='Classic')
class AudioFormatBumpers(BumperBase):
format = peewee.CharField()
class BumperBase(ContentBase):
is3D = peewee.BooleanField(default=False)
isImage = peewee.BooleanField(default=False)
path = peewee.CharField(unique=True)
class Trivia(ContentBase):
type = peewee.CharField()
TID = peewee.CharField(unique=True)
rating = peewee.CharField(null=True)
genre = peewee.CharField(null=True)
year = peewee.CharField(null=True)
duration = peewee.FloatField(default=0)
questionPath = peewee.CharField(unique=True, null=True)
cluePath0 = peewee.CharField(unique=True, null=True)
cluePath1 = peewee.CharField(unique=True, null=True)
cluePath2 = peewee.CharField(unique=True, null=True)
cluePath3 = peewee.CharField(unique=True, null=True)
cluePath4 = peewee.CharField(unique=True, null=True)
cluePath5 = peewee.CharField(unique=True, null=True)
cluePath6 = peewee.CharField(unique=True, null=True)
cluePath7 = peewee.CharField(unique=True, null=True)
cluePath8 = peewee.CharField(unique=True, null=True)
cluePath9 = peewee.CharField(unique=True, null=True)
answerPath = peewee.CharField(unique=True, null=True)
