Add year field on TVSerie model

Author: rafaelhenrique

migrations/versions/e66884bb260f_.py

@@ -0,0 +1,30 @@
+"""empty message
+
+Revision ID: e66884bb260f
+Revises: c23d2a740fc7
+Create Date: 2016-11-28 22:19:19.418769
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'e66884bb260f'
+down_revision = 'c23d2a740fc7'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('tv_serie', sa.Column('year', sa.Date(), nullable=True))
+    op.create_unique_constraint(None, 'tv_serie', ['id'])
+    ### end Alembic commands ###
+
+
+def downgrade():
+    ### commands auto generated by Alembic - please adjust! ###
+    # op.drop_constraint(None, 'tv_serie', type_='unique')
+    op.drop_column('tv_serie', 'year')
+    ### end Alembic commands ###

tvseries/core/models.py

@@ -9,6 +9,7 @@ class TVSerie(db.Model):
     description = db.Column(db.Text, nullable=True)
     episodies_number = db.Column(db.Integer, nullable=False, default=1)
     author = db.Column(db.String(50), nullable=False)
+    year = db.Column(db.Date)
 
     def __repr__(self):
         if self.description:

tvseries/tests/test_core.py

@@ -1,5 +1,6 @@
 import pytest
 
+from datetime import date
 from tvseries.core.models import TVSerie
 from tvseries.config import TestConfig
 
@@ -79,7 +80,8 @@ def test_insert_on_model_tvserie(self, db):
         )
         serie = TVSerie(name="Game of Thrones",
                         description=description,
-                        author="George R.R. Martin")
+                        author="George R.R. Martin",
+                        year=date(2011, 1, 1))
         db.session.add(serie)
         db.session.commit()
         assert TVSerie.query.count() == 1
@@ -102,7 +104,8 @@ def test_repr_on_model_tvserie(self, db):
         )
         serie = TVSerie(name="Game of Thrones",
                         description=description,
-                        author="George R.R. Martin")
+                        author="George R.R. Martin",
+                        year=date(2011, 1, 1))
         assert repr(serie) == (
             "TVSerie(id=None, name='Game of Thrones', "
             "description='Há muito t...', episodies_number=None)"