Parse Mixer facilities

trailmap/parse_input.py

@@ -144,3 +144,22 @@ def find_sep_commod(facility, facility_archetype):
             facility_out_commods.extend(out_commods)
 
     return facility_in_commods, facility_out_commods
+
+
+def find_mixer_commod(facility, facility_archetype):
+    '''Searches for commodities within a Mixer facility, which uses a
+    different xml schema than other Cyclus archetypes
+    '''
+    out_tags = ['out_commod']
+    facility_in_commods = []
+    facility_out_commods = []
+
+    for archetype_var in facility.find('.config/' + facility_archetype):
+        if archetype_var.tag == 'in_streams':
+            for commod in archetype_var.findall('./stream/commodities/item/commodity'):
+                facility_in_commods.append(commod.text)
+        out_commods = find_commod(archetype_var, out_tags)
+        if out_commods is not None:
+            facility_out_commods.extend(out_commods)
+
+    return facility_in_commods, facility_out_commods