Add is_single_stage to ChapterDisplay

ebl/corpus/application/display_schemas.py

@@ -32,6 +32,7 @@ def make_line(self, data: dict, **kwargs) -> LineDisplay:
 class ChapterDisplaySchema(Schema):
     id_ = fields.Nested(ChapterIdSchema, required=True, data_key="id")
     text_name = fields.String(required=True, data_key="textName")
+    is_single_stage = fields.Boolean(required=True, data_key="isSingleStage")
     title = fields.List(fields.Nested(OneOfNoteLinePartSchema), dump_only=True)
     lines = fields.Nested(LineDisplaySchema, many=True, required=True)
 
@@ -40,5 +41,6 @@ def make_chapter(self, data: dict, **kwargs) -> ChapterDisplay:
         return ChapterDisplay(
             data["id_"],
             data["text_name"],
+            data["is_single_stage"],
             tuple(data["lines"]),
         )

ebl/corpus/domain/chapter_display.py

@@ -51,6 +51,7 @@ def of_line(line: Line) -> "LineDisplay":
 class ChapterDisplay:
     id_: ChapterId
     text_name: str
+    is_single_stage: bool
     lines: Sequence[LineDisplay]
 
     @property
@@ -62,5 +63,6 @@ def of_chapter(text: Text, chapter: Chapter) -> "ChapterDisplay":
         return ChapterDisplay(
             chapter.id_,
             text.name,
+            not text.has_multiple_stages,
             tuple(map(LineDisplay.of_line, chapter.lines)),
         )

ebl/corpus/domain/text.py

@@ -45,3 +45,7 @@ class Text:
     @property
     def id(self) -> TextId:
         return TextId(self.genre, self.category, self.index)
+
+    @property
+    def has_multiple_stages(self) -> bool:
+        return len({chapter.stage for chapter in self.chapters}) > 1

ebl/corpus/infrastructure/mongo_text_repository.py

@@ -112,7 +112,11 @@ def find_chapter_for_display(self, id_: ChapterId) -> ChapterDisplay:
             text = self.find(id_.text_id)
             chapters = self._chapters.aggregate(aggregate_chapter_display(id_))
             return ChapterDisplaySchema().load(
-                {**next(chapters), "textName": text.name}
+                {
+                    **next(chapters),
+                    "textName": text.name,
+                    "isSingleStage": not text.has_multiple_stages,
+                }
             )
         except NotFoundError as error:
             raise text_not_found(id_.text_id) from error

ebl/tests/corpus/test_chapter_display.py

@@ -36,6 +36,7 @@ def test_chapter_display_of_chapter() -> None:
     assert chapter_display == ChapterDisplay(
         chapter.id_,
         text.name,
+        not text.has_multiple_stages,
         tuple(LineDisplay.of_line(line) for line in chapter.lines),
     )
     assert chapter_display.title == make_title(chapter.lines[0].translation)

ebl/tests/corpus/test_chapter_display_schema

@@ -25,6 +25,8 @@ def to_dict(chapter: ChapterDisplay) -> dict:
                 "index": chapter.id_.text_id.index,
             },
         },
+        "textName": chapter.text_name,
+        "isSingleStage": chapter.is_single_stage,
         "lines": [
             {
                 "number": OneOfLineNumberSchema().dump(line.number),

ebl/tests/corpus/test_text.py

@@ -0,0 +1,23 @@
+import pytest
+
+from ebl.corpus.domain.stage import Stage
+from ebl.tests.factories.corpus import ChapterListingFactory, TextFactory
+
+
+@pytest.mark.parametrize(  # pyre-ignore[56]
+    "chapters,expected",
+    [
+        (tuple(), False),
+        (ChapterListingFactory.build_batch(2, stage=Stage.NEO_ASSYRIAN), False),
+        (
+            [
+                ChapterListingFactory.build(stage=Stage.NEO_ASSYRIAN),
+                ChapterListingFactory.build(stage=Stage.OLD_ASSYRIAN),
+            ],
+            True,
+        ),
+    ],
+)
+def test_has_multiple_stages(chapters, expected) -> None:
+    text = TextFactory.build(chapters=chapters)
+    assert text.has_multiple_stages == expected