Merge branch 'main' of https://github.com/brightway-lca/bw_simapro_csv

bw_simapro_csv/brightway.py

@@ -55,6 +55,15 @@ def substitute_unspecified(s: Union[str, None]) -> Union[str, None]:
     return s
 
 
+def allocation_as_manual_property(exc: dict) -> dict:
+    """If allocation field is present, add it as manual property as well"""
+    if "allocation" in exc:
+        if "properties" not in exc:
+            exc["properties"] = {}
+        exc["properties"]["manual_allocation"] = exc["allocation"]
+    return exc
+
+
 def lci_to_brightway(spcsv: SimaProCSV, missing_string: str = "(unknown)") -> dict:
     """Turn an extracted SimaPro CSV extract into metadata that can be imported into Brightway.
 
@@ -170,7 +179,7 @@ def lci_to_brightway(spcsv: SimaProCSV, missing_string: str = "(unknown)") -> di
         if "Products" in process.blocks:
             for edge in process.blocks["Products"].parsed:
                 process_dataset["exchanges"].append(
-                    edge | {"type": "production", "functional": True}
+                    allocation_as_manual_property(edge | {"type": "production", "functional": True})
                 )
         elif "Waste treatment" in process.blocks:
             for edge in process.blocks["Waste treatment"].parsed:

tests/integration/test_brightway_export_allocation.py

@@ -0,0 +1,78 @@
+import datetime
+from pprint import pprint
+
+from bw_simapro_csv import SimaProCSV
+
+
+def test_basic_header_extraction(fixtures_dir):
+    given = SimaProCSV(fixtures_dir / "allocation.csv").to_brightway()
+    expected = {
+        "created": "2016-10-12T22:54:47",
+        "name": "Bobs_burgers",
+        "simapro_csv_version": None,
+        "simapro_libraries": ["Methods"],
+        "simapro_project": "Bobs_burgers",
+        "simapro_version": "8.2.0.0",
+    }
+    for key, value in expected.items():
+        assert value == given["database"][key]
+
+    allocated = given["processes"][1]
+    expected = {
+        "code": "ReCenter000033915302504",
+        "comment": "Stuff happened, and then we enjoyed life outside the office",
+        "data_entry": "Some people and stuff like that",
+        "data_generator": "Somebody",
+        "database": "Bobs_burgers",
+        "location": None,
+        "name": "Rice, at farm (WFLDB 3.0)/kg/IN",
+        "publication_date": datetime.date(2015, 6, 8),
+        "references": [],
+        "simapro_project": "Bobs_burgers",
+        "tags": {
+            "category_type": "material",
+            "simapro_type": "Unit process",
+            "system_description": "ECOSPOLD IMPORT",
+        },
+        "type": "multifunctional",
+    }
+    for key, value in expected.items():
+        assert allocated[key] == value
+
+    assert set(expected).union({"exchanges"}) == set(allocated)
+
+    expected = {
+        'allocation': 95.8,
+        'amount': 6250.0,
+        'category': r'_WFLDB 3.0\Plant products\Arable\Rice',
+        'comment': 'INDIA',
+        'functional': True,
+        'line_no': 211,
+        'name': 'Rice, at farm (WFLDB 3.0)/IN U',
+        'properties': {'manual_allocation': 95.8},
+        'type': 'production',
+        'unit': 'kg',
+        'waste_type': 'not defined'
+    }
+    assert allocated["exchanges"][0] == expected
+
+    expected = {
+        'allocation': 4.2,
+        'amount': 3125.0,
+        'category': r'_WFLDB 3.0\Plant products\Arable\Rice',
+        'comment': 'The amount of straw is calculated '
+                   'from the straw to grain-ratio of 1 '
+                   'and a straw harvest rate of 50%. '
+                   'Economic allocation is based on the '
+                   'assumption that grains account for '
+                   '92% and straw for 8% of the price.',
+        'functional': True,
+        'line_no': 212,
+        'name': 'Rice straw, at farm (WFLDB 3.0)/IN U',
+        'properties': {'manual_allocation': 4.2},
+        'type': 'production',
+        'unit': 'kg',
+        'waste_type': 'not defined'
+    }
+    assert allocated["exchanges"][1] == expected
+    assert len(allocated["exchanges"])