def format_layer(self, layers, base_id): anns = {} for (start, end, n_pg), page_ref in zip(layers["pages"], layers["pages_ref"]): uuid = self.get_unique_id() span = Span(start=start, end=end) page = Page(span=span, imgnum=n_pg, reference=page_ref) anns[uuid] = page layer = Layer(annotation_type=LayerEnum("Pagination"), annotations=anns) result = { LayerEnum("Pagination").value: json.loads(layer.json(exclude_none=True)) } return result
def save_layer(self, base_name: str, layer_name: LayerEnum, layer: Layer): layer_fn = self._mkdir( self.layers_path / base_name) / f"{layer_name.value}.yml" dump_yaml(json.loads(layer.json()), layer_fn)