Commit: a34e66bbcad3e13452d20b1bc4fc5f587feb5ad7 - factorio_calc (hg)

Just a simple, and painful to use calculator for the game Factorio written in Python

Commit MetaInfo

• Revision: a34e66bbcad3e13452d20b1bc4fc5f587feb5ad7 (tree)
• Time: 2017-10-03 11:56:13
• Author: Eric Hopper <hopper@omni...>
• Commiter: Eric Hopper

Log Message

Implement a way to save itemdb (but not yet load) as XML.

Change Summary

• modified: factorio_calc.py
• modified: item-db.pickle

Incremental Difference

diff -r e1e9d8116ea8 -r a34e66bbcad3 factorio_calc.py
--- a/factorio_calc.py	Wed Aug 09 15:49:29 2017 -0700
+++ b/factorio_calc.py	Mon Oct 02 19:56:13 2017 -0700
@@ -4,6 +4,8 @@
 from fractions import Fraction as _F
 import sys
 from collections import namedtuple
+import re
+
 
 class ProductionItem:
     __slots__ = ('_name', '_time', '_ingredients', '_produced', '__weakref__')
@@ -75,6 +77,42 @@
                     return item
         raise KeyError(name)
 
+    @staticmethod
+    def _itemId(item):
+        idstr = item._name.lower()
+        idstr = re.sub(r'\s+', '_', idstr)
+        return idstr
+
+    @staticmethod
+    def _itemAsXML(item, item_idmap):
+        if item not in item_idmap:
+            cur_id = ItemSet._itemId(item)
+            item_idmap[item] = (cur_id, False)
+            for _, ingredient in item._ingredients:
+                yield from ItemSet._itemAsXML(ingredient, item_idmap)
+            if item._produced is not None:
+                yield f'  <item id="{cur_id}" name="{item._name}" ' \
+                      f'time="{item._time}" ' \
+                      f'produced="{item._produced}">\n'
+                for count, ingredient in item._ingredients:
+                    ingredient_id = item_idmap[ingredient][0]
+                    yield f'    <ingredient idref="{ingredient_id}" ' \
+                          f'count="{count}" />\n'
+                yield '  </item>\n'
+            else:
+                yield f'  <item id="{cur_id}" name="{item._name}" />\n'
+            item_idmap[item] = (cur_id, True)
+        elif not item_idmap[item][1]:
+            raise RuntimeError(f"Circular reference detected '{item._name}'")
+
+    def asXML(self):
+        yield '<?xml version="1.0" encoding="utf-8" standalone="no" ?>\n'
+        yield '<factorio_calc_item_db>\n'
+        item_idmap = {}
+        for item in self:
+            yield from self._itemAsXML(item, item_idmap)
+        yield '</factorio_calc_item_db>\n'
+
 _mod_dir = _osp.dirname(__file__)
 db_fname = _osp.join(_mod_dir, 'item-db.pickle')
 

diff -r e1e9d8116ea8 -r a34e66bbcad3 item-db.pickle
Binary file item-db.pickle has changed