2

87150d01 · 王晓庆 · 07a5522f · 87150d01
--- a/testcases.py
+++ b/testcases.py
@@ -83,16 +83,16 @@ if __name__ == '__main__':
    # 运行学生代码
    query_sql = parse_sql(sql_file)
-    cursor.execute(query_sql)
+    # cursor.execute(query_sql)
-    alldata = cursor.fetchall()
+    # alldata = cursor.fetchall()
-    print(alldata)
+    # print(alldata)
-    # df = pd.read_sql(query_sql, conn)
+    df = pd.read_sql(query_sql, conn)
    # 保存成csv格式，带表头，带每行前的序号，注意编码方式
    # csv方便用户查看
-    # df.to_csv(user_output_csv_file, index=True, header=True, encoding='gb18030')
+    df.to_csv(user_output_csv_file, index=True, header=True, encoding='gb18030')
    # txt用于和答案模板比较
-    # print(df)
+    print(df)
-    # df.to_csv(user_output_txt_file, index=False, header=True, encoding='UTF-8')
+    df.to_csv(user_output_txt_file, index=False, header=True, encoding='UTF-8')
    # 连接用完后记得关闭以释放资源
    conn.close()