From e276d87d19c95350a96805e27a48687af413338b Mon Sep 17 00:00:00 2001
From: Luciano Almeida <lucianoalmeida@interlegis.leg.br>
Date: Mon, 3 Apr 2017 13:53:45 -0300
Subject: [PATCH] =?UTF-8?q?Arruma=20problema=20da=20recria=C3=A7=C3=A3o=20?=
 =?UTF-8?q?das=20constraints?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Luciano Almeida <lucianoalmeida@interlegis.leg.br>
---
 .../migrations/0002_auto_20170331_1900.py     | 51 +++++++++++++++++++
 sapl/legacy/migration.py                      |  6 ++-
 2 files changed, 55 insertions(+), 2 deletions(-)
 create mode 100644 sapl/base/migrations/0002_auto_20170331_1900.py

diff --git a/sapl/base/migrations/0002_auto_20170331_1900.py b/sapl/base/migrations/0002_auto_20170331_1900.py
new file mode 100644
index 000000000..5ee726647
--- /dev/null
+++ b/sapl/base/migrations/0002_auto_20170331_1900.py
@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.9.12 on 2017-03-31 19:00
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('base', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Argumento',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('argumento', models.CharField(max_length=50, verbose_name='Argumento')),
+            ],
+            options={
+                'verbose_name': 'Argumento da constraint',
+                'verbose_name_plural': 'Argumentos da constraint',
+            },
+        ),
+        migrations.CreateModel(
+            name='Constraint',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('nome_tabela', models.CharField(max_length=50, verbose_name='Nome da tabela')),
+                ('nome_constraint', models.CharField(max_length=100, verbose_name='Nome da constraint')),
+                ('nome_model', models.CharField(max_length=50, verbose_name='Nome da model')),
+                ('tipo_constraint', models.CharField(max_length=50, verbose_name='Tipo da constraint')),
+            ],
+            options={
+                'verbose_name': 'Constraint removida',
+                'verbose_name_plural': 'Constraints removidas',
+            },
+        ),
+        migrations.AddField(
+            model_name='problemamigracao',
+            name='eh_importante',
+            field=models.BooleanField(default=False, verbose_name='É importante?'),
+        ),
+        migrations.AddField(
+            model_name='argumento',
+            name='constraint',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='base.Constraint'),
+        ),
+    ]
diff --git a/sapl/legacy/migration.py b/sapl/legacy/migration.py
index c4c7d0f77..60ab48e23 100644
--- a/sapl/legacy/migration.py
+++ b/sapl/legacy/migration.py
@@ -253,8 +253,10 @@ def recreate_constraints():
         if con.tipo_constraint == 'unique_together':
             nome_tabela = con.nome_tabela
             nome_constraint = con.nome_constraint
-            model = ContentType.objects.get(
-                model=con.nome_model.lower()).model_class()
+            # Pegando explicitamente o primeiro valor do filter,
+            # pois pode ser que haja mais de uma ocorrência
+            model = ContentType.objects.filter(
+                model=con.nome_model.lower())[0].model_class()
             args = [a.argumento for a in con.argumento_set.all()]
             for i in range(len(args)):
                 if isinstance(model._meta.get_field(args[i]),