From 6e264c60e1df777cfcd47708ea809d2f8e514684 Mon Sep 17 00:00:00 2001
From: Jon Rosebaugh <jon@inklesspen.com>
Date: Tue, 1 Sep 2015 14:07:15 -0400
Subject: [PATCH] Adding some model columns

---
 .../versions/4efa2946f5a_adjustments.py       | 35 +++++++++++++++++++
 mimir/models/__init__.py                      |  4 +++
 2 files changed, 39 insertions(+)
 create mode 100644 mimir/alembic/versions/4efa2946f5a_adjustments.py

diff --git a/mimir/alembic/versions/4efa2946f5a_adjustments.py b/mimir/alembic/versions/4efa2946f5a_adjustments.py
new file mode 100644
index 0000000..a857880
--- /dev/null
+++ b/mimir/alembic/versions/4efa2946f5a_adjustments.py
@@ -0,0 +1,35 @@
+"""adjustments
+
+Revision ID: 4efa2946f5a
+Revises: 4dbff7d28fd
+Create Date: 2015-08-30 22:02:00.790640
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '4efa2946f5a'
+down_revision = '4dbff7d28fd'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.add_column('writeup_posts',
+                  sa.Column('published', sa.Boolean(), nullable=False, server_default=sa.true()))
+    op.add_column('writeup_post_versions',
+                  sa.Column('active', sa.Boolean(), nullable=False, server_default=sa.true()))
+    op.add_column('writeup_post_versions',
+                  sa.Column('version', sa.Integer(), nullable=False, server_default='1'))
+    op.alter_column('writeup_posts', 'published', server_default=None)
+    op.alter_column('writeup_post_versions', 'active', server_default=None)
+    op.alter_column('writeup_post_versions', 'version', server_default=None)
+
+
+def downgrade():
+    op.drop_column('writeup_posts', 'published')
+    op.drop_column('writeup_post_versions', 'version')
+    op.drop_column('writeup_post_versions', 'active')
diff --git a/mimir/models/__init__.py b/mimir/models/__init__.py
index 0adb54d..c94a88f 100644
--- a/mimir/models/__init__.py
+++ b/mimir/models/__init__.py
@@ -141,6 +141,7 @@ class WriteupPost(Base):
     title = Column(Unicode, nullable=True)
     url = Column(Unicode, nullable=False)
     last_fetched = Column(AwareDateTime, nullable=False)
+    published = Column(Boolean, nullable=False, default=False)
 
     versions = relationship("WriteupPostVersion", backref="writeup_post")
 
@@ -161,6 +162,9 @@ class WriteupPostVersion(Base):
     threadpost_id = Column(Integer, ForeignKey("thread_posts.id"), nullable=True)
     extracted_at = Column(AwareDateTime, nullable=False)
 
+    version = Column(Integer, nullable=False, default=1)
+    active = Column(Boolean, nullable=False, default=False)
+
 
 class Image(Base):
     id = Column(String(64), primary_key=True)