From 505d91718080ec41af600009728d5287439436a1 Mon Sep 17 00:00:00 2001
From: Ruud <ruud@crashdummy.nl>
Date: Fri, 28 Dec 2012 13:34:50 +0100
Subject: [PATCH] Update DB with 3d support

---
 .../migration/versions/001_Add_3D_support.py    | 17 +++++++++++++++++
 couchpotato/core/settings/model.py              |  1 +
 2 files changed, 18 insertions(+)
 create mode 100644 couchpotato/core/migration/versions/001_Add_3D_support.py

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 00000000..5c8c9b54
--- /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 64117fb4..f0ef088e 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')
-- 
GitLab