Browse Source

Fix issue with H2O factories

JoostSijm 5 years ago
parent
commit
36dd37dcbf
1 changed files with 5 additions and 1 deletions
  1. 5 1
      app/api.py

+ 5 - 1
app/api.py

@@ -50,10 +50,14 @@ def parse_factories(html):
     factories = []
     for factory_tree in factories_tree:
         columns = factory_tree.find_all('td')
+        if columns[1].contents[4].name == 'span':
+            resource_type = TYPES[columns[1].contents[4]['class'][0]]
+        else:
+            resource_type = None
         factories.append({
             'id': factory_tree['user'],
             'name': columns[1].contents[0].strip(),
-            'resource_type': TYPES[columns[1].contents[4]['class'][0]],
+            'resource_type': resource_type,
             'region_name': columns[1].contents[2],
             'level': columns[2].string,
             'workers': re.sub(r'\/[0-9]*$', '', columns[3].string),