Merge pull request #3244 from webkom/fix-companyInterest-csv-export

Validate company company name at csv export
webkom · Mar 14, 2023 · ba01477 · ba01477
2 parents cb02392 + ba11777
commit ba01477
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lego/apps/companies/views.py b/lego/apps/companies/views.py
@@ -196,6 +196,11 @@ def csv(self, *args, **kwargs):
             ]
         )
         for companyInterest in companyInterests:
+            company_name = (
+                companyInterest.company.name
+                if companyInterest.company
+                else companyInterest.company_name
+            )
             participant_range_start = companyInterest.participant_range_start
             participant_range_end = companyInterest.participant_range_end
             semesters = ", ".join(
@@ -260,7 +265,7 @@ def csv(self, *args, **kwargs):
             )
             writer.writerow(
                 [
-                    companyInterest.company_name,
+                    company_name,
                     companyInterest.contact_person,
                     companyInterest.mail,
                     companyInterest.phone,