update LegalEntity model

oscar_invoices/migrations/0005_legalentity_company_number_and_more.py

@@ -0,0 +1,23 @@
+# Generated by Django 4.2.11 on 2024-03-23 19:54
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('oscar_invoices', '0004_date_banking_details'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='legalentity',
+            name='company_number',
+            field=models.CharField(blank=True, max_length=20, null=True, verbose_name='Company identification number'),
+        ),
+        migrations.AlterField(
+            model_name='legalentity',
+            name='vat_number',
+            field=models.CharField(blank=True, max_length=20, null=True, verbose_name='VAT identification number'),
+        ),
+    ]

oscar_invoices/templates/oscar_invoices/invoice.html

@@ -133,7 +133,12 @@ <h1>{% trans "Original invoice" %} #{{ invoice.number }}</h1><br>
                                 </td>
                                 <td>
                                     {{ legal_entity.business_name }}<br>
-                                    {% trans "VAT ID" %}: {{ legal_entity.vat_number }}<br>
+                                    {%  if legal_entity.vat_number %}
+                                      {% trans "VAT ID" %}: {{ legal_entity.vat_number }}<br>
+                                    {% endif %}
+                                    {%  if legal_entity.company_number %}
+                                      {% trans "Company ID" %}: {{ legal_entity.company_number }}<br>
+                                    {% endif %}
                                     <address>
                                         {{ legal_entity_address.summary }}<br/>
                                         {% if legal_entity_address.phone_number %}

tests/factories.py

@@ -6,7 +6,7 @@
 
 class LegalEntityFactory(factory.django.DjangoModelFactory):
     business_name = 'Test Company'
-    vat_number = 'test-vat-number'
+    company_number = 'test-company-number'
 
     class Meta:
         model = get_model('oscar_invoices', 'LegalEntity')