소스 검색

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])