summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRawng <rawng.github@gmail.com>2014-10-11 11:28:57 -0400
committerRawng <rawng.github@gmail.com>2014-10-11 11:28:57 -0400
commitbcdc59583b7a109eced76fb50869b5ad47123895 (patch)
tree56dfda2b8f2ff27aa959cc43b9116a7ff949a9d4
parent4d494f21474684590ae4e475fb0e0c0fcea47daf (diff)
downloadmanamarket-bcdc59583b7a109eced76fb50869b5ad47123895.tar.gz
manamarket-bcdc59583b7a109eced76fb50869b5ad47123895.tar.bz2
manamarket-bcdc59583b7a109eced76fb50869b5ad47123895.tar.xz
manamarket-bcdc59583b7a109eced76fb50869b5ad47123895.zip
Closes #6.
-rw-r--r--stats/process_salelog/utils.py22
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]