Add subjects from relations to vocabulary

acdh-oeaw · Jul 2, 2024 · 66ed898 · 66ed898
1 parent 1223075
commit 66ed898
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/apis_ontology/management/commands/transfer_subjects_to_vocab.py b/apis_ontology/management/commands/transfer_subjects_to_vocab.py
@@ -1,7 +1,9 @@
 from django.core.management.base import BaseCommand
 
 from apis_ontology.models import Subject, Work
+from apis_core.relations.models import Relation
 import re
+from tqdm.auto import tqdm
 
 DELIMITERS = [",", ";", "+", "/"]
 PATTERN = "|".join(map(re.escape, DELIMITERS))
@@ -16,16 +18,28 @@ def split_subjects(old_subject_field):
                 si.strip().capitalize() for si in re.split(PATTERN, old_subject_field)
             ]
 
-        for w in Work.objects.all():
+        for w in tqdm(Work.objects.all()):
             if w.subject:
                 subject_list = split_subjects(w.subject)
                 for si in subject_list:
                     new_sub, _ = Subject.objects.get_or_create(name=si)
                     w.skip_history_when_saving = True
                     w.subject_vocab.add(new_sub)
 
+        rels_updated = 0
+        for rel in tqdm(Relation.objects.all()):
+            if hasattr(rel, "subject_of_teaching"):
+                if rel.subject_of_teaching:
+                    subject_list = split_subjects(rel.subject_of_teaching)
+                    for si in subject_list:
+                        new_sub, _ = Subject.objects.get_or_create(name=si)
+                        rel.skip_history_when_saving = True
+                        rel.subject_vocab.add(new_sub)
+                        rels_updated += 1
+
         self.stdout.write(
             self.style.SUCCESS(
                 f"Subject vocabulary now contains {len(Subject.objects.all())} subjects."
             )
         )
+        print(f"{rels_updated} relations were updated with subject as a vocabulary")