Преглед на файлове

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