SegmentTemplates and SegmentTimelines to SegmentTemplate and SegmentTimeline

Author: avishaycohen

src/mpd_parser/models/composite_tags.py

@@ -100,13 +100,9 @@ def segment_lists(self):
         ]
 
     @cached_property
-    def segment_templates(self):
-        return [
-            SegmentTemplate(member)
-            for member in self.element.xpath(
-                LOOKUP_STR_FORMAT.format(target="SegmentTemplate")
-            )
-        ]
+    def segment_template(self):
+        elements = self.element.xpath(LOOKUP_STR_FORMAT.format(target="SegmentTemplate"))
+        return SegmentTemplate(elements[0]) if elements else None
 
     @cached_property
     def asset_identifiers(self):
@@ -601,13 +597,9 @@ def segment_lists(self):
         ]
 
     @cached_property
-    def segment_templates(self):
-        return [
-            SegmentTemplate(member)
-            for member in self.element.xpath(
-                LOOKUP_STR_FORMAT.format(target="SegmentTemplate")
-            )
-        ]
+    def segment_template(self):
+        elements = self.element.xpath(LOOKUP_STR_FORMAT.format(target="SegmentTemplate"))
+        return SegmentTemplate(elements[0]) if elements else None
 
     @cached_property
     def sub_representations(self):
@@ -769,13 +761,9 @@ def segment_lists(self):
         ]
 
     @cached_property
-    def segment_templates(self):
-        return [
-            SegmentTemplate(member)
-            for member in self.element.xpath(
-                LOOKUP_STR_FORMAT.format(target="SegmentTemplate")
-            )
-        ]
+    def segment_template(self):
+        elements = self.element.xpath(LOOKUP_STR_FORMAT.format(target="SegmentTemplate"))
+        return SegmentTemplate(elements[0]) if elements else None
 
     @cached_property
     def representations(self):

src/mpd_parser/models/segment_tags.py

@@ -94,13 +94,11 @@ def start_number(self):
         return get_int_value(self.element.attrib.get("startNumber"))
 
     @cached_property
-    def segment_timelines(self):
-        return [
-            SegmentTimeline(member)
-            for member in self.element.xpath(
-                LOOKUP_STR_FORMAT.format(target="SegmentTimeline")
-            )
-        ]
+    def segment_timeline(self):
+        elements = self.element.xpath(
+            LOOKUP_STR_FORMAT.format(target="SegmentTimeline")
+        )
+        return SegmentTimeline(elements[0]) if elements else None
 
     @cached_property
     def bitstream_switchings(self):