ducminh-phan/reformat-gherkin

from typing import Optional, Tuple

from ._base import prepare
from .location import LocationMixin
from .table_row import TableRow
from .tag import Tag


@prepare
class Examples(LocationMixin):
    keyword: str
    name: str
    tags: Tuple[Tag, ...]
    description: str
    table_body: Tuple[TableRow, ...]
    table_header: Optional[TableRow] = None

    def __iter__(self):
        yield from self.tags

        yield self

        if self.table_header is not None:
            yield self.table_header

        yield from self.table_body