feat(brevo): créer les objets dans Brevo lors de leur création en BDD

clevercloud/cron.json

@@ -11,7 +11,6 @@
     "25 7 * * 1 $ROOT/clevercloud/siaes_update_count_fields.sh",
     "30 7 * * 1 $ROOT/clevercloud/siaes_update_super_badge_field.sh",
     "50 7 * * 1 $ROOT/clevercloud/companies_update_users_and_count_fields.sh",
-    "55 7 * * 1 $ROOT/clevercloud/crm_brevo_sync_companies.sh",
     "0 7 * * 2 $ROOT/clevercloud/siaes_send_completion_reminder_emails.sh",
     "30 7 * * * $ROOT/clevercloud/tenders_send_author_list_of_super_siaes_emails.sh",
     "0 8 * * * $ROOT/clevercloud/siaes_send_user_request_reminder_emails.sh",
@@ -20,4 +19,4 @@
     "0 9 * * * $ROOT/clevercloud/tenders_send_siae_contacted_reminder_emails.sh",
     "10 9 * * * $ROOT/clevercloud/tenders_send_siae_interested_reminder_emails.sh",
     "*/5 8-15 * * 1-5 $ROOT/clevercloud/tenders_send_validated.sh"
-]
+]

lemarche/companies/migrations/0006_company_brevo_company_id.py

@@ -0,0 +1,19 @@
+# Generated by Django 4.2.17 on 2025-02-06 14:36
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("companies", "0005_company_linkedin_buyer_count"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="company",
+            name="brevo_company_id",
+            field=models.CharField(
+                blank=True, help_text="Identifiant de l'entreprise dans Brevo", null=True, verbose_name="ID Brevo"
+            ),
+        ),
+    ]

lemarche/companies/models.py

@@ -3,6 +3,8 @@
 from django.template.defaultfilters import slugify
 from django.utils import timezone
 from django_better_admin_arrayfield.models.fields import ArrayField
+from django.db.models.signals import post_save
+from django.dispatch import receiver
 
 from lemarche.utils.constants import ADMIN_FIELD_HELP_TEXT, RECALCULATED_FIELD_HELP_TEXT
 
@@ -52,6 +54,13 @@ class Company(models.Model):
     created_at = models.DateTimeField(verbose_name="Date de création", default=timezone.now)
     updated_at = models.DateTimeField(verbose_name="Date de modification", auto_now=True)
 
+    brevo_company_id = models.CharField(
+        verbose_name="ID Brevo",
+        help_text="Identifiant de l'entreprise dans Brevo",
+        blank=True,
+        null=True,
+    )
+
     objects = models.Manager.from_queryset(CompanyQuerySet)()
 
     class Meta:
@@ -72,3 +81,11 @@ def save(self, *args, **kwargs):
         """Generate the slug field before saving."""
         self.set_slug()
         super().save(*args, **kwargs)
+
+
+@receiver(post_save, sender=Company)
+def create_company_in_brevo(sender, instance, created, **kwargs):
+    if created:
+        from lemarche.utils.apis.api_brevo import create_brevo_company_from_company
+
+        create_brevo_company_from_company(instance)

lemarche/companies/tests/test_models.py

@@ -0,0 +1,49 @@
+from unittest.mock import patch, MagicMock
+
+from django.test import TestCase
+
+from lemarche.companies.factories import CompanyFactory
+
+
+class CompanySignalTest(TestCase):
+    @patch("lemarche.utils.apis.api_brevo.create_company")
+    def test_create_company_in_brevo_signal(self, mock_create_company):
+        """Test that creating a new company triggers the Brevo sync"""
+        # Create a new company
+        company = CompanyFactory()
+
+        # Verify create_company was called once with the company instance
+        mock_create_company.assert_called_once_with(company)
+
+        # Create another company
+        company2 = CompanyFactory()
+        self.assertEqual(mock_create_company.call_count, 2)
+        mock_create_company.assert_called_with(company2)
+
+        # Update existing company
+        company.name = "Updated Name"
+        company.save()
+
+        # Call count should still be 2 since we only sync on creation
+        self.assertEqual(mock_create_company.call_count, 2)
+
+    @patch("lemarche.utils.apis.api_brevo.sib_api_v3_sdk.CompaniesApi")
+    def test_company_attributes_sent_to_brevo(self, mock_companies_api):
+        """Test the attributes sent to Brevo when creating a company"""
+        # Setup the mock response
+        mock_response = MagicMock()
+        mock_response.id = 12345
+        mock_instance = mock_companies_api.return_value
+        mock_instance.companies_post.return_value = mock_response
+
+        # Create a company
+        company = CompanyFactory(name="Test Company", website="https://example.com")
+
+        # Get the Body instance that was passed to companies_post
+        args, kwargs = mock_instance.companies_post.call_args
+        body_obj = args[0]
+
+        self.assertEqual(body_obj.name, "Test Company")
+        self.assertEqual(body_obj.attributes, {"domain": "https://example.com", "app_id": company.id, "siae": False})
+
+        self.assertEqual(company.brevo_company_id, 12345)

lemarche/crm/management/commands/crm_brevo_sync_companies.py

@@ -30,34 +30,16 @@ def handle(self, recently_updated: bool, **options):
         self.stdout.write("Script to sync Siaes with Brevo CRM (companies)...")
 
         # Step 1: build the queryset
-        siaes_qs = Siae.objects.all()
+        siaes_qs = Siae.objects.filter(brevo_company_id__isnull=True)
         self.stdout.write(f"Sync Siae > Brevo: we have {Siae.objects.count()} siaes")
         # Update only the recently updated siaes
         if recently_updated:
             siaes_qs = siaes_qs.filter(updated_at__gte=two_weeks_ago)
             self.stdout.write(f"Sync Siae > Brevo: {siaes_qs.count()} recently updated")
 
-        # Step 2: Add the 90-day limited annotations
-        siaes_qs = siaes_qs.with_tender_stats(since_days=90)
-
-        # Step 3: loop on the siaes
+        # Step 2: loop on the siaes
         for index, siae in enumerate(siaes_qs):
-            new_extra_data = {
-                "completion_rate": siae.completion_rate if siae.completion_rate is not None else 0,
-                "tender_received": siae.tender_email_send_count_annotated,
-                "tender_interest": siae.tender_detail_contact_click_count_annotated,
-            }
-
-            # extra_data update if needed
-            if siae.extra_data.get("brevo_company_data") != new_extra_data:
-                siae.extra_data.update(
-                    {
-                        "brevo_company_data": new_extra_data,
-                    }
-                )
-                siae.save(update_fields=["extra_data"])
-
-            api_brevo.create_or_update_company(siae)
+            api_brevo.create_brevo_company_from_siae(siae)
             if (index % 10) == 0:  # avoid API rate-limiting
                 time.sleep(1)
             if (index % 500) == 0: