Acrescentando entidade Projeto na app convenios

sigi/apps/casas/admin.py

@@ -4,7 +4,7 @@ from django.contrib.contenttypes import generic
 from sigi.apps.casas.forms import CasaLegislativaForm
 from sigi.apps.casas.models import CasaLegislativa
 from sigi.apps.contatos.models import Contato, Telefone
-from sigi.apps.convenios.models import Convenio, EquipamentoPrevisto, Anexo
+from sigi.apps.convenios.models import Projeto, Convenio, EquipamentoPrevisto, Anexo
 
 class ContatosInline(generic.GenericTabularInline):
     model = Contato
@@ -17,7 +17,7 @@ class TelefonesInline(generic.GenericTabularInline):
 
 class ConveniosInline(admin.TabularInline):
     model = Convenio
-    extra =1
+    extra = 1
 
 class CasaLegislativaAdmin(admin.ModelAdmin):
     form = CasaLegislativaForm

sigi/apps/convenios/admin.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 from django.contrib import admin
-from sigi.apps.convenios.models import Convenio, EquipamentoPrevisto, Anexo
+from sigi.apps.convenios.models import Projeto, Convenio, EquipamentoPrevisto, Anexo
 from sigi.apps.servicos.models import Servico
 
 class AnexosInline(admin.TabularInline):
@@ -25,7 +25,7 @@ class ConvenioAdmin(admin.ModelAdmin):
     change_list_template = 'convenios/change_list.html'
     fieldsets = (
         (None,
-            {'fields': ('casa_legislativa', 'num_processo_sf','tipo_convenio')}
+            {'fields': ('casa_legislativa', 'num_processo_sf','projeto')}
         ),
         ('Datas',
             {'fields': ('data_adesao', 'data_retorno_assinatura',
@@ -53,6 +53,7 @@ class EquipamentoPrevistoAdmin(admin.ModelAdmin):
     search_fields = ('convenio__id', 'equipamento__fabricante__nome',
                      'equipamento__modelo__modelo', 'equipamento__modelo__tipo__tipo')
 
+admin.site.register(Projeto)
 admin.site.register(Convenio, ConvenioAdmin)
 admin.site.register(EquipamentoPrevisto, EquipamentoPrevistoAdmin)
 admin.site.register(Anexo, AnexoAdmin)

sigi/apps/convenios/models.py

@@ -4,12 +4,13 @@ from django.db import models
 #from django.contrib.contenttypes import ContentType
 from django.contrib.contenttypes import generic
 
-class Convenio(models.Model):
-    CONVENIO_TIPO = (
-    ('PI', 'Programa Interlegis'),
-    ('PPL', 'Projeto Piloto de Modernização'),
-    ('PML', 'Projeto Modernização Legislativo')
-    )
+class Projeto(models.Model):
+    nome = models.CharField(max_length=50)
+        
+    def __unicode__(self):
+        return self.nome
+    
+class Convenio(models.Model):    
     casa_legislativa = models.ForeignKey(
         'casas.CasaLegislativa',
         verbose_name='Casa Legislativa'
@@ -25,9 +26,8 @@ class Convenio(models.Model):
         null=True,
         blank=True,
     )
-    tipo_convenio = models.CharField(
-        max_length=10,
-        choices=CONVENIO_TIPO
+    projeto = models.ForeignKey(
+        Projeto
     )
     data_retorno_assinatura = models.DateField(
         'data do retorno e assinatura',

sigi/sites.py

@@ -6,7 +6,7 @@ from sigi.apps.casas.admin import CasaLegislativa, CasaLegislativaAdmin
 from sigi.apps.contatos.admin import (UnidadeFederativa, UnidadeFederativaAdmin,
                                       Municipio, MunicipioAdmin, Telefone,
                                       TelefoneAdmin, Contato, ContatoAdmin)
-from sigi.apps.convenios.admin import (Convenio, ConvenioAdmin, EquipamentoPrevisto,
+from sigi.apps.convenios.admin import (Projeto, Convenio, ConvenioAdmin, EquipamentoPrevisto,
                                        EquipamentoPrevistoAdmin, Anexo, AnexoAdmin)
 from sigi.apps.inventario.admin import (Fornecedor, FornecedorAdmin, Fabricante,
                                         FabricanteAdmin, Equipamento,
@@ -50,6 +50,7 @@ default.register(Telefone, TelefoneAdmin)
 default.register(Contato, ContatoAdmin)
 
 # sigi.apps.convenios
+default.register(Projeto)
 default.register(Convenio, ConvenioAdmin)
 default.register(EquipamentoPrevisto, EquipamentoPrevistoAdmin)
 default.register(Anexo, AnexoAdmin)