diff --git a/couchpotato/core/migration/versions/001_Add_3D_support.py b/couchpotato/core/migration/versions/001_Add_3D_support.py new file mode 100644 index 0000000000000000000000000000000000000000..5c8c9b5407c49371a80edda46c365193f320bffa --- /dev/null +++ b/couchpotato/core/migration/versions/001_Add_3D_support.py @@ -0,0 +1,17 @@ +from migrate.changeset.schema import create_column +from sqlalchemy.schema import MetaData, Column, Table +from sqlalchemy.types import Integer + +meta = MetaData() + + +def upgrade(migrate_engine): + meta.bind = migrate_engine + + threed_column = Column('threed', Integer) + resource = Table('profiletype', meta, threed_column) + + create_column(threed_column, resource) + +def downgrade(migrate_engine): + pass diff --git a/couchpotato/core/settings/model.py b/couchpotato/core/settings/model.py index 64117fb48dc41ad32c26e38a43a64ccb367ac942..f0ef088e46027b1bb51aa1b7b74760f41688f26f 100644 --- a/couchpotato/core/settings/model.py +++ b/couchpotato/core/settings/model.py @@ -173,6 +173,7 @@ class ProfileType(Entity): order = Field(Integer, index = True) finish = Field(Boolean) + threed = Field(Boolean) wait_for = Field(Integer) quality = ManyToOne('Quality')