浏览代码

Now handling degenerate case

andreagus 8 年之前
父节点
当前提交
6eb07270d5
共有 1 个文件被更改,包括 9 次插入3 次删除
  1. 9 3
      converter.py

+ 9 - 3
converter.py

@@ -11,7 +11,13 @@ with open(file_name_csv, 'wb') as csvfile:
     for neighbor in root.iter('DATI'):
         parent = neighbor.attrib.get("ISTANTE")
         parent = parent[:-4]
-        vm = neighbor[0].text
-        vmin = neighbor[1].text
-        vmax = neighbor[2].text
+        number = len(neighbor.getchildren())
+        if number > 1:
+            vm = neighbor[0].text
+            vmin = neighbor[1].text
+            vmax = neighbor[2].text
+        else:
+            vm = "NA"
+            vmin = "NA"
+            vmax = "NA"
         spamwriter.writerow([parent, vm, vmin, vmax])