def label_activity_geocollections(name):
"""Add geocollections to activity ``location`` fields.
``name`` is the name of an existing LCI database."""
assert name in databases, "{} not found".format(name)
assert ("world" in geocollections
), "Please run `bw2regionalsetup` before this function"
ecoinvent = ({x[1]
for x in Topography("ecoinvent").load()}
if "ecoinvent" in topocollections else set())
RoWs = ({x[1]
for x in Topography("RoW").load()}
if "RoW" in topocollections else set())
db = Database(name)
searchable = db.metadata.get("searchable")
if searchable:
db.make_unsearchable()
geocollections = set()
locations = {x["location"] for x in db}
assert "RoW" not in locations, "`RoW` found; use `rower` to label Rest-of-Worlds"
for act in pyprind.prog_bar(db):
if isinstance(act["location"], tuple):
geocollections.add(act["location"][0])
elif act["location"] in COUNTRIES:
geocollections.add("world")
elif act["location"] == "GLO":
geocollections.add("world")
elif act["location"] in RoWs:
act["location"] = ("RoW", act["location"])
act.save()
geocollections.add("RoW")
elif act["location"] in ecoinvent:
act["location"] = ("ecoinvent", act["location"])
act.save()
geocollections.add("ecoinvent")
else:
warnings.warn(
("Location {} in {} not understood; please add geocollection"
" manually, and add to databases[name]['geocollections']"
).format(act["location"], act.key))
if searchable:
db.make_searchable()
db.process()
db.metadata["regionalized"] = True
db.metadata["geocollections"] = sorted(geocollections)
databases.flush()
def __call__(cls, db_obj):
if isinstance(db_obj, str):
if db_obj in databases:
db_obj = Database(db_obj)
else:
raise ValueError(f"Given database '{db_obj}' doesn't exist")
elif not isinstance(db_obj, Database):
raise ValueError(f"Can't understand database input '{db_obj}'")
functional_flows = cls.find_functional_flows(db_obj)
cls.map_products(functional_flows)
cls.map_virtual_activities()
cls.label_functional_flows(functional_flows)
db_obj.metadata['multifunctional'] = True
# Flush metadata?
db_obj.process()
def create_database(self, name, data):
db = Database(name)
db.register()
db.write(data)
db.process()