Merge pull request #56 from seasidesparrow/subtitle_fix.20240228

correctly captures subtitles
adsabs · Feb 28, 2024 · 61f793c · 61f793c
2 parents 2fc718b + 03bfa4a
commit 61f793c
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/adsmanparse/translator.py b/adsmanparse/translator.py
@@ -69,12 +69,17 @@ def _get_title(self):
             title_ln = title.get('langNative', None)
             if title_en:
                 self.output['title'] = title_en
-            else:
-                if title_tn:
-                    self.output['title'] = title_tn
-                    self.output['language'] = title_ln
+            elif title_tn:
+                self.output['title'] = title_tn
+                self.output['language'] = title_ln
             if subtitle:
-                self.output['title'] += ": " + subtitle
+                subtitle_en = subtitle.get('textEnglish', None)
+                subtitle_tn = subtitle.get('textNative', None)
+                subtitle_ln = subtitle.get('langNative', None)
+                if subtitle_en:
+                    self.output['title'] += ": " + subtitle_en
+                elif subtitle_tn:
+                    self.output['title'] += ": " + subtitle_tn
 
 
     # INDIVIDUAL NAME