Add eventvote owner

Author: chaosagent

migrations/versions/daf5b5301c75_.py

@@ -0,0 +1,30 @@
+"""empty message
+
+Revision ID: daf5b5301c75
+Revises: f79ba2365689
+Create Date: 2016-06-30 10:08:05.129124
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = 'daf5b5301c75'
+down_revision = 'f79ba2365689'
+
+from alembic import op
+import sqlalchemy as sa
+
+
+def upgrade():
+### commands auto generated by Alembic - please adjust! ###
+    op.add_column('eventvotes', sa.Column('user_id', sa.Integer(), nullable=True))
+    op.create_unique_constraint('eventvote_user_event_uc', 'eventvotes', ['user_id', 'event_id'])
+    op.create_foreign_key('eventvote_user_id_fk', 'eventvotes', 'users', ['user_id'], ['id'])
+    ### end Alembic commands ###
+
+
+def downgrade():
+### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint('eventvote_user_id_fk', 'eventvotes', type_='foreignkey')
+    op.drop_constraint('eventvote_user_event_uc', 'eventvotes', type_='unique')
+    op.drop_column('eventvotes', 'user_id')
+    ### end Alembic commands ###

models.py

@@ -1,5 +1,6 @@
 from flask_login import LoginManager
 from flask_sqlalchemy import SQLAlchemy
+from sqlalchemy import UniqueConstraint
 from sqlalchemy.orm import backref
 from sqlalchemy.ext.hybrid import hybrid_property
 
@@ -95,6 +96,9 @@ class Event(db.Model):
 class EventVote(db.Model):
     __tablename__ = 'eventvotes'
     id = db.Column(db.Integer, primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('users.id', name='eventvote_user_id_fk'))
+    user = db.relationship('User', backref='event_votes')
     event_id = db.Column(db.Integer, db.ForeignKey('events.id', name='vote_event_id_fk'), index=True)
     event = db.relationship('Event', backref='votes')
     direction = db.Column(db.Boolean)
+    __table_args__ = (UniqueConstraint('user_id', 'event_id', name='eventvote_user_event_uc'),)