store the day migrations get sent in msg profile

mamasbm/mamasbm/models.py

@@ -75,12 +75,14 @@ class MessageProfile(Base):
     name = Column(Text)
     profile_id = Column(UUID, ForeignKey('profiles.uuid'))
     messages = relationship('Message', backref='message_profile')
+    send_day = Column(Integer)
 
     def to_dict(self):
         return {
             'uuid': str(self.uuid),
             'profile_id': str(self.profile_id),
             'name': self.name,
+            'send_day': self.send_day,
             'messages': [m.to_dict() for m in self.messages]
         }
 

mamasbm/mamasbm/web/factory.py

@@ -8,11 +8,12 @@
 def build_message_profiles(name, temp_file, profile_uuid):
     with transaction.manager:
         profile = DBSession.query(Profile).get(profile_uuid)
-        num_days = len(profile.get_send_days())
-        days = CsvImporter(num_days).import_csv(temp_file)
+        send_days = profile.get_send_days()
+        days = CsvImporter(num_days=len(send_days)).import_csv(temp_file)
         for k, v in days.items():
             day_name = calendar.day_name[k]
             msg_profile = MessageProfile(name='%s - %s' % (name, day_name))
+            msg_profile.send_day = send_days[k]
             messages = [
                 Message(week=wk, text=txt)
                 for wk, txt in v.items()

mamasbm/mamasbm/web/tests.py

@@ -227,6 +227,8 @@ def test_message_profile_factory(self):
         self.assertEquals(
             resp.json[0]['message_profiles'][0]['name'],
             'English - Monday')
+        self.assertEquals(resp.json[0]['message_profiles'][0]['send_day'], 1)
+        self.assertEquals(resp.json[0]['message_profiles'][1]['send_day'], 4)
         self.assertEquals(
             len(resp.json[0]['message_profiles'][0]['messages']),
             36)

mamasbm/migrations/versions/1c9e829abd8e_add_send_day_to_message_profile.py

@@ -0,0 +1,27 @@
+"""add send_day to message profile
+
+Revision ID: 1c9e829abd8e
+Revises: 29fb24fdc151
+Create Date: 2014-07-16 14:23:04.006251
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '1c9e829abd8e'
+down_revision = '29fb24fdc151'
+
+from alembic import op
+import sqlalchemy as sa
+
+
+def upgrade():
+    ### commands auto generated by Alembic - please adjust! ###
+    op.add_column(
+        'message_profiles', sa.Column('send_day', sa.Integer(), nullable=True))
+    ### end Alembic commands ###
+
+
+def downgrade():
+    ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('message_profiles', 'send_day')
+    ### end Alembic commands ###