Browse Source

Verificar grupo

Verificar campo de grupo ['GRUPO']
Alejandro Rosales 1 năm trước cách đây
mục cha
commit
59b92549dc
1 tập tin đã thay đổi với 3 bổ sung0 xóa
  1. 3 0
      app.py

+ 3 - 0
app.py

@@ -109,6 +109,9 @@ def extract(username: str, password: str):
         historial_html_io = StringIO(f"<table>{historial_html}</table>")
         # Read the HTML table into a DataFrame
         df = pd.read_html(historial_html_io)[0]
+        
+        if 'GRUPO' not in df.columns:
+           raise KeyError("Column 'GRUPO' not found in the DataFrame")
 
         # Convert the DataFrame to JSON
         json_result = df[df['GRUPO'] != 'Promedio:'].to_json(orient='records')