add contractor on intervention model

GeotrekCE · Feb 8, 2024 · 5ec61b8 · 5ec61b8
1 parent 7dc7c9b
commit 5ec61b8
Show file tree

Hide file tree

Showing 16 changed files with 107 additions and 35 deletions.
diff --git a/geotrek/maintenance/filters.py b/geotrek/maintenance/filters.py
@@ -151,7 +151,7 @@ class InterventionFilterSet(AltimetryInterventionFilterSet, ZoningFilterSet, Str
     class Meta(StructureRelatedFilterSet.Meta):
         model = Intervention
         fields = StructureRelatedFilterSet.Meta.fields + [
-            'status', 'type', 'stake', 'subcontracting', 'project', 'on',
+            'status', 'type', 'stake', 'subcontracting', 'project', 'contractors', 'on',
         ]
 
     def __init__(self, *args, **kwargs):

diff --git a/geotrek/maintenance/forms.py b/geotrek/maintenance/forms.py
@@ -90,6 +90,7 @@ class InterventionForm(CommonForm):
             'height',
             'stake',
             'project',
+            'contractors',
             'access',
             'description',
             'material_cost',
@@ -104,7 +105,7 @@ class Meta(CommonForm.Meta):
         model = Intervention
         fields = CommonForm.Meta.fields + \
             ['structure', 'name', 'begin_date', 'end_date', 'status', 'disorders', 'type', 'description', 'subcontracting', 'length', 'width',
-             'height', 'stake', 'project', 'access', 'material_cost', 'heliport_cost', 'subcontract_cost', 'topology']
+             'height', 'stake', 'project', 'contractors', 'access', 'material_cost', 'heliport_cost', 'subcontract_cost', 'topology']
 
     def __init__(self, *args, target_type=None, target_id=None, **kwargs):
         super().__init__(*args, **kwargs)
@@ -204,9 +205,7 @@ class ProjectForm(CommonForm):
                     'constraint',
                     'global_cost',
                     'comments',
-
-                    css_class="span6"),
-                Div('project_owner',
+                    'project_owner',
                     'project_manager',
                     'contractors',
                     Fieldset(_("Fundings")),

diff --git a/geotrek/maintenance/locale/de/LC_MESSAGES/django.po b/geotrek/maintenance/locale/de/LC_MESSAGES/django.po
@@ -8,7 +8,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: PACKAGE VERSION\n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-12-11 15:38+0000\n"
+"POT-Creation-Date: 2024-02-07 12:56+0000\n"
 "PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
 "Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
 "Language-Team: LANGUAGE <[email protected]>\n"
@@ -161,6 +161,9 @@ msgstr ""
 msgid "Subcontract cost"
 msgstr ""
 
+msgid "Contractors"
+msgstr ""
+
 msgid "3D Length"
 msgstr ""
 
@@ -266,9 +269,6 @@ msgstr ""
 msgid "Domain"
 msgstr ""
 
-msgid "Contractors"
-msgstr ""
-
 msgid "Project owner"
 msgstr ""
 
@@ -281,6 +281,9 @@ msgstr ""
 msgid "Projects"
 msgstr ""
 
+msgid "Interventions contractors"
+msgstr ""
+
 msgid "Period"
 msgstr ""
 

diff --git a/geotrek/maintenance/locale/en/LC_MESSAGES/django.po b/geotrek/maintenance/locale/en/LC_MESSAGES/django.po
@@ -8,7 +8,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: PACKAGE VERSION\n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-12-11 15:38+0000\n"
+"POT-Creation-Date: 2024-02-07 12:56+0000\n"
 "PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
 "Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
 "Language-Team: LANGUAGE <[email protected]>\n"
@@ -161,6 +161,9 @@ msgstr ""
 msgid "Subcontract cost"
 msgstr ""
 
+msgid "Contractors"
+msgstr ""
+
 msgid "3D Length"
 msgstr ""
 
@@ -266,9 +269,6 @@ msgstr ""
 msgid "Domain"
 msgstr ""
 
-msgid "Contractors"
-msgstr ""
-
 msgid "Project owner"
 msgstr ""
 
@@ -281,6 +281,9 @@ msgstr ""
 msgid "Projects"
 msgstr ""
 
+msgid "Interventions contractors"
+msgstr ""
+
 msgid "Period"
 msgstr ""
 

diff --git a/geotrek/maintenance/locale/es/LC_MESSAGES/django.po b/geotrek/maintenance/locale/es/LC_MESSAGES/django.po
@@ -8,7 +8,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: PACKAGE VERSION\n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-12-11 15:38+0000\n"
+"POT-Creation-Date: 2024-02-07 12:56+0000\n"
 "PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
 "Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
 "Language-Team: LANGUAGE <[email protected]>\n"
@@ -161,6 +161,9 @@ msgstr ""
 msgid "Subcontract cost"
 msgstr ""
 
+msgid "Contractors"
+msgstr ""
+
 msgid "3D Length"
 msgstr ""
 
@@ -266,9 +269,6 @@ msgstr ""
 msgid "Domain"
 msgstr ""
 
-msgid "Contractors"
-msgstr ""
-
 msgid "Project owner"
 msgstr ""
 
@@ -281,6 +281,9 @@ msgstr ""
 msgid "Projects"
 msgstr ""
 
+msgid "Interventions contractors"
+msgstr ""
+
 msgid "Period"
 msgstr ""
 

diff --git a/geotrek/maintenance/locale/fr/LC_MESSAGES/django.po b/geotrek/maintenance/locale/fr/LC_MESSAGES/django.po
@@ -7,7 +7,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: PACKAGE VERSION\n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-12-11 15:38+0000\n"
+"POT-Creation-Date: 2024-02-07 12:56+0000\n"
 "PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
 "Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
 "Language-Team: LANGUAGE <[email protected]>\n"
@@ -163,6 +163,9 @@ msgstr "Coût héliportage"
 msgid "Subcontract cost"
 msgstr "Coût sous-traitance"
 
+msgid "Contractors"
+msgstr "Prestataires"
+
 msgid "3D Length"
 msgstr "Longueur 3D"
 
@@ -268,9 +271,6 @@ msgstr "Commentaires"
 msgid "Domain"
 msgstr "Domaine"
 
-msgid "Contractors"
-msgstr "Prestataires"
-
 msgid "Project owner"
 msgstr "Maître d'ouvrage"
 
@@ -283,6 +283,9 @@ msgstr "Financeurs"
 msgid "Projects"
 msgstr "Chantiers"
 
+msgid "Interventions contractors"
+msgstr "Prestataires des interventions"
+
 msgid "Period"
 msgstr "Période"
 

diff --git a/geotrek/maintenance/locale/it/LC_MESSAGES/django.po b/geotrek/maintenance/locale/it/LC_MESSAGES/django.po
@@ -8,7 +8,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: PACKAGE VERSION\n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-12-11 15:38+0000\n"
+"POT-Creation-Date: 2024-02-07 12:56+0000\n"
 "PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
 "Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
 "Language-Team: LANGUAGE <[email protected]>\n"
@@ -161,6 +161,9 @@ msgstr ""
 msgid "Subcontract cost"
 msgstr ""
 
+msgid "Contractors"
+msgstr ""
+
 msgid "3D Length"
 msgstr ""
 
@@ -266,9 +269,6 @@ msgstr ""
 msgid "Domain"
 msgstr ""
 
-msgid "Contractors"
-msgstr ""
-
 msgid "Project owner"
 msgstr ""
 
@@ -281,6 +281,9 @@ msgstr ""
 msgid "Projects"
 msgstr ""
 
+msgid "Interventions contractors"
+msgstr ""
+
 msgid "Period"
 msgstr ""
 

diff --git a/geotrek/maintenance/locale/nl/LC_MESSAGES/django.po b/geotrek/maintenance/locale/nl/LC_MESSAGES/django.po
@@ -8,7 +8,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: PACKAGE VERSION\n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-12-11 15:38+0000\n"
+"POT-Creation-Date: 2024-02-07 12:56+0000\n"
 "PO-Revision-Date: YEAR-MO-DA HO:MI+ZONE\n"
 "Last-Translator: FULL NAME <EMAIL@ADDRESS>\n"
 "Language-Team: LANGUAGE <[email protected]>\n"
@@ -161,6 +161,9 @@ msgstr ""
 msgid "Subcontract cost"
 msgstr ""
 
+msgid "Contractors"
+msgstr ""
+
 msgid "3D Length"
 msgstr ""
 
@@ -266,9 +269,6 @@ msgstr ""
 msgid "Domain"
 msgstr ""
 
-msgid "Contractors"
-msgstr ""
-
 msgid "Project owner"
 msgstr ""
 
@@ -281,6 +281,9 @@ msgstr ""
 msgid "Projects"
 msgstr ""
 
+msgid "Interventions contractors"
+msgstr ""
+
 msgid "Period"
 msgstr ""
 

diff --git a/geotrek/maintenance/migrations/0026_intervention_contractors.py b/geotrek/maintenance/migrations/0026_intervention_contractors.py
@@ -0,0 +1,18 @@
+# Generated by Django 3.2.23 on 2024-02-05 09:27
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('maintenance', '0025_remove_intervention_date'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='intervention',
+            name='contractors',
+            field=models.ManyToManyField(blank=True, related_name='interventions', to='maintenance.Contractor', verbose_name='Contractors'),
+        ),
+    ]
diff --git a/geotrek/maintenance/models.py b/geotrek/maintenance/models.py
@@ -23,7 +23,7 @@
 from geotrek.zoning.mixins import ZoningPropertiesMixin
 
 from mapentity.models import DuplicateMixin
-
+from django.contrib.postgres.aggregates import ArrayAgg
 
 if 'geotrek.signage' in settings.INSTALLED_APPS:
     from geotrek.signage.models import Blade
@@ -50,6 +50,8 @@ class Intervention(ZoningPropertiesMixin, AddPropertyMixin, GeotrekMapEntityMixi
     material_cost = models.FloatField(default=0.0, blank=True, null=True, verbose_name=_("Material cost"))
     heliport_cost = models.FloatField(default=0.0, blank=True, null=True, verbose_name=_("Heliport cost"))
     subcontract_cost = models.FloatField(default=0.0, blank=True, null=True, verbose_name=_("Subcontract cost"))
+    contractors = models.ManyToManyField('Contractor', related_name="interventions", blank=True,
+                                         verbose_name=_("Contractors"))
 
     # AltimetryMixin for denormalized fields from related topology, updated via trigger.
     length = models.FloatField(editable=True, default=0.0, null=True, blank=True, verbose_name=_("3D Length"))
@@ -549,8 +551,13 @@ def interventions_csv_display(self):
         return [str(i) for i in self.interventions.existing()]
 
     @property
-    def contractors_display(self):
-        return [str(c) for c in self.contractors.all()]
+    def intervention_contractors(self):
+        return self.interventions.aggregate(
+            intervention_contractors=ArrayAgg('contractors__contractor', distinct=True))['intervention_contractors']
+
+    @classproperty
+    def intervention_contractors_verbose_name(cls):
+        return _("Interventions contractors")
 
     @property
     def founders_display(self):

diff --git a/geotrek/maintenance/templates/maintenance/intervention_detail.html b/geotrek/maintenance/templates/maintenance/intervention_detail.html
@@ -95,6 +95,12 @@
                 {% else %} <span class="none">{% trans "None" %}</span>
                 {% endif %}
             </td>
+        </tr>
+            <tr>
+            <th>{{ intervention|verbose:"contractors" }}</th>
+            <td>
+                {% valuelist intervention.contractors.all %}
+            </td>
         </tr>
         <tr>
             <th>{% trans "Related object" %}</th>

diff --git a/geotrek/maintenance/templates/maintenance/project_detail.html b/geotrek/maintenance/templates/maintenance/project_detail.html
@@ -59,6 +59,12 @@
                 {% valuelist project.contractors.all %}
             </td>
         </tr>
+        <tr>
+            <th>{{ project|verbose:"intervention_contractors" }}</th>
+            <td>
+                {% valuelist project.intervention_contractors %}
+            </td>
+        </tr>
         <tr>
             <th>{% trans "Fundings" %}</th>
             <td>

diff --git a/geotrek/maintenance/templates/maintenance/sql/post_20_views.sql b/geotrek/maintenance/templates/maintenance/sql/post_20_views.sql
@@ -25,6 +25,7 @@ SELECT a.id,
        j.nb_days AS "Cost",
        (a.material_cost+a.heliport_cost+a.subcontract_cost)::float AS "Total cost",
        h.name AS "Project",
+       m.contractor AS "Contractors",
        CASE
            WHEN k.app_label = 'core' THEN 'Tronçons'
            WHEN k.app_label = 'infrastructure' THEN 'Aménagements'
@@ -38,6 +39,13 @@ SELECT a.id,
        a.date_update AS "Update date",
        a.geom_3d
 FROM maintenance_intervention a
+LEFT JOIN
+    (SELECT array_to_string(array_agg(a.contractor), ', '::text, '_'::text) contractor,
+            intervention_id
+     FROM maintenance_contractor a
+     JOIN maintenance_intervention_contractors b ON a.id = b.contractor_id
+     JOIN maintenance_intervention c ON b.intervention_id = c.id
+     GROUP BY intervention_id) m ON a.id = m.intervention_id
 LEFT JOIN maintenance_interventiontype b ON a.type_id = b.id
 LEFT JOIN maintenance_interventionstatus c ON a.status_id = c.id
 LEFT JOIN core_stake d ON a.stake_id = d.id

diff --git a/geotrek/maintenance/tests/test_models.py b/geotrek/maintenance/tests/test_models.py
@@ -17,7 +17,7 @@
     FundingFactory, InfrastructureInterventionFactory,
     InfrastructurePointInterventionFactory, InterventionDisorderFactory,
     InterventionFactory, InterventionJobFactory, ManDayFactory, ProjectFactory,
-    SignageInterventionFactory)
+    SignageInterventionFactory, ContractorFactory)
 from geotrek.outdoor.tests.factories import CourseFactory, SiteFactory
 from geotrek.signage.tests.factories import BladeFactory, SignageFactory
 
@@ -361,3 +361,11 @@ def test_trails_property(self):
         project.interventions.add(intervention_blade)
         project.interventions.add(intervention_course)
         self.assertQuerysetEqual(list(project.trails), ['trail_1', 'trail_2', 'trail_signage'], ordered=False, transform=str)
+
+    def test_intervention_contractors(self):
+        project = ProjectFactory.create()
+        contractor1 = ContractorFactory.create(contractor="contractor1")
+        contractor2 = ContractorFactory.create(contractor="contractor2")
+        intervention = InterventionFactory.create(project=project)
+        intervention.contractors.set([contractor1, contractor2])
+        self.assertEqual(project.intervention_contractors, ["contractor1", "contractor2"])