bw2package_fixed.py

import bw2data, bw2io

BIOSPHERE_FLOW_DATABASE_LABELS = ["ecoinvent-3.9.1-biosphere"]

class Purolator(bw2io.package.BW2Package):
    @classmethod
    def _create_obj(cls, data):
        instance = data["class"](data["name"])

        if data["name"] not in instance._metadata:
            instance.register(**data["metadata"])
        else:
            instance.backup()
            instance.metadata = data["metadata"]

        flows = {
            (obj['database'], obj['code']) 
            for label in BIOSPHERE_FLOW_DATABASE_LABELS
            for obj in bw2data.Database(label)
        }
        
        if isinstance(instance, bw2data.Method):
            data['data'] = [line for line in data['data'] if line[0] in flows]
        
        instance.write(data["data"])
        return instance