diff options
| -rw-r--r-- | stats/process_salelog/utils.py | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/stats/process_salelog/utils.py b/stats/process_salelog/utils.py index fd4e432..4c59f5f 100644 --- a/stats/process_salelog/utils.py +++ b/stats/process_salelog/utils.py @@ -49,15 +49,19 @@ class ItemDB: self.itemdb_file = ElementTree(file="../data/items.xml") for item in self.itemdb_file.getroot(): - if item.get('id') > 500: - item_struct = Item() - item_struct.name = item.get('name') - if item.get('weight'): - item_struct.weight = item.get('weight') - if item.get('type'): - item_struct.type = item.get('type') - item_struct.description = item.get('description') - self.item_names[int(item.get('id'))] = item_struct + if item.get('name'): + file2 = ElementTree(file="../" + item.get('name')) + for item2 in file2.getroot(): + if item2.get('name'): + file3 = ElementTree(file="../" + item2.get('name')) + for item3 in file3.getroot(): + item_struct = Item() + item_struct.name = item3.get('name') + item_struct.weight = int(item3.get('weight', 0)) + if item3.get('type'): + item_struct.type = item3.get('type') + item_struct.description = item3.get('description') + self.item_names[int(item3.get('id'))] = item_struct def getItem(self, item_id): return self.item_names[item_id] |