{WIP} Get header from each auctiations #3

codevance · May 19, 2018 · ca36c22 · ca36c22
1 parent ae33ae8
commit ca36c22
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/comprasnet/__init__.py b/comprasnet/__init__.py
@@ -76,11 +76,15 @@ def get_data_auctions_pages(self, data):
 
             current_result['cidade'] = cidade
             current_result['uf'] = uf
+            headers = [element for element in td.b.contents
+                       if 'Código da UASG' not in element]
+            headers = [element.strip() for element in headers
+                       if isinstance(element, str)]
+            current_result['cabecalho'] = headers
 
             for line in str(td).split("<br/>"):
 
                 if 'digo da UASG' in line:
-
                     try:
                         current_result['cabecalho'] = header
                         codigo_da_uasg_chave = line.split(":")[0]