From 914de78576baf66d8f4c0365d8cedb27c6f70663 Mon Sep 17 00:00:00 2001
From: kevgliss <kevgliss@gmail.com>
Date: Wed, 10 May 2017 12:13:42 -0700
Subject: [PATCH] Adds migration to fix keys on unique index. Closes #743.
 (#785)

---
 lemur/migrations/versions/5e680529b666_.py |  4 ----
 lemur/migrations/versions/8ae67285ff14_.py | 24 ++++++++++++++++++++++
 lemur/models.py                            |  2 +-
 3 files changed, 25 insertions(+), 5 deletions(-)
 create mode 100644 lemur/migrations/versions/8ae67285ff14_.py

diff --git a/lemur/migrations/versions/5e680529b666_.py b/lemur/migrations/versions/5e680529b666_.py
index f5d8489a..d59d996f 100644
--- a/lemur/migrations/versions/5e680529b666_.py
+++ b/lemur/migrations/versions/5e680529b666_.py
@@ -15,16 +15,12 @@ import sqlalchemy as sa
 
 
 def upgrade():
-    ### commands auto generated by Alembic - please adjust! ###
     op.add_column('endpoints', sa.Column('sensitive', sa.Boolean(), nullable=True))
     op.add_column('endpoints', sa.Column('source_id', sa.Integer(), nullable=True))
     op.create_foreign_key(None, 'endpoints', 'sources', ['source_id'], ['id'])
-    ### end Alembic commands ###
 
 
 def downgrade():
-    ### commands auto generated by Alembic - please adjust! ###
     op.drop_constraint(None, 'endpoints', type_='foreignkey')
     op.drop_column('endpoints', 'source_id')
     op.drop_column('endpoints', 'sensitive')
-    ### end Alembic commands ###
\ No newline at end of file
diff --git a/lemur/migrations/versions/8ae67285ff14_.py b/lemur/migrations/versions/8ae67285ff14_.py
new file mode 100644
index 00000000..f45be70d
--- /dev/null
+++ b/lemur/migrations/versions/8ae67285ff14_.py
@@ -0,0 +1,24 @@
+"""Modifies the unique index for the certificate replacements
+
+Revision ID: 8ae67285ff14
+Revises: 5e680529b666
+Create Date: 2017-05-10 11:56:13.999332
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '8ae67285ff14'
+down_revision = '5e680529b666'
+
+from alembic import op
+import sqlalchemy as sa
+
+
+def upgrade():
+    op.drop_index('certificate_replacement_associations_ix')
+    op.create_index('certificate_replacement_associations_ix', 'certificate_replacement_associations', ['replaced_certificate_id', 'certificate_id'], unique=True)
+
+
+def downgrade():
+    op.drop_index('certificate_replacement_associations_ix')
+    op.create_index('certificate_replacement_associations_ix', 'certificate_replacement_associations', ['certificate_id', 'certificate_id'], unique=True)
diff --git a/lemur/models.py b/lemur/models.py
index 60328c75..f189d76c 100644
--- a/lemur/models.py
+++ b/lemur/models.py
@@ -53,7 +53,7 @@ certificate_replacement_associations = db.Table('certificate_replacement_associa
                                                        ForeignKey('certificates.id', ondelete='cascade'))
                                                 )
 
-Index('certificate_replacement_associations_ix', certificate_replacement_associations.c.certificate_id, certificate_replacement_associations.c.certificate_id)
+Index('certificate_replacement_associations_ix', certificate_replacement_associations.c.replaced_certificate_id, certificate_replacement_associations.c.certificate_id)
 
 roles_authorities = db.Table('roles_authorities',
                              Column('authority_id', Integer, ForeignKey('authorities.id')),