浏览代码

Verificar grupo

Verificar campo de grupo ['GRUPO']
Alejandro Rosales 1 年之前
父节点
当前提交
59b92549dc
共有 1 个文件被更改,包括 3 次插入0 次删除
  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')