Class TableGroupBlueprint

pydbml/parser/blueprints.py:304–328

@dataclass
class TableGroupBlueprint(Blueprint):
    name: str
    items: List[str]
    comment: Optional[str] = None
    note: Optional[NoteBlueprint] = None
    color: Optional[str] = None

    def build(self) -> 'TableGroup':
        if not self.parser:
            raise RuntimeError('Parser is not set')
        items = []
        for table_name in self.items:
            components = table_name.split('.')
            schema, table = components if len(components) == 2 else ('public', components[0])
            table_obj = self.parser.locate_table(schema, table)
            if table_obj in items:
                raise ValidationError(f'Table "{table}" is already in group "{self.name}"')
            items.append(table_obj)
        return TableGroup(
            name=self.name,
            items=items,
            comment=self.comment,
            note=self.note.build() if self.note else None,
            color=self.color
        )