Source code for bw2regional.xtables

from .loading import Loading
from .meta import extension_tables, geocollections
from .validate import xtable_validator




class ExtensionTable(Loading):


    _metadata = extension_tables



    validator = xtable_validator



    matrix = "xtable_matrix"


    @property


    def filename(self):
        return super(ExtensionTable, self).filename.replace(".loading", ".xtable")




    def write_to_map(self, *args, **kwargs):
        raise NotImplementedError




    def import_from_map(self, mask=None):
        from .utils import get_pandarus_map

        geocollection = extension_tables[self.name].get("geocollection")
        xt_field = extension_tables[self.name].get("xt_field")

        if not geocollection:
            raise ValueError("No geocollection for this extension table")
        if geocollections[geocollection].get('kind') == 'raster':
            raise ValueError("This function is only for vectors.")

        map_obj = get_pandarus_map(geocollection)
        data = []

        if xt_field is None:
            raise ValueError("No `xt_field` field name specified")

        id_field = geocollections[geocollection].get("field")
        if not id_field:
            raise ValueError(
                "Geocollection must specify ``field`` field name for unique feature ids"
            )

        for feature in map_obj:
            label = feature["properties"][id_field]
            value = float(feature["properties"][xt_field])
            if mask is not None and value == mask:
                continue
            data.append((value, (geocollection, label)))

        self.write(data)