From 8c9395d6a0e07ea64a7ed88a466c5befb83178e3 Mon Sep 17 00:00:00 2001
From: 阳洁 <2077506045@qq.com>
Date: 星期五, 28 七月 2023 18:42:59 +0800
Subject: [PATCH] 测评系统自动测试
---
comm/read_data.py | 36 ++++++++++++++++++++++++++++++++++++
1 files changed, 36 insertions(+), 0 deletions(-)
diff --git a/comm/read_data.py b/comm/read_data.py
new file mode 100644
index 0000000..58d3a3e
--- /dev/null
+++ b/comm/read_data.py
@@ -0,0 +1,36 @@
+import codecs
+import csv
+from openpyxl import load_workbook
+
+
+def read_csv(file_name):
+ data_list = list()
+ with codecs.open(file_name, encoding="utf-8") as f:
+ for row in csv.DictReader(f, skipinitialspace=True):
+ data_list.append(row)
+
+ return data_list
+
+
+def read_excel(file_name, sheet_name):
+ # 鎵撳紑Excel鏂囦欢锛屽姞杞絯orkbook
+ workbook = load_workbook(filename=file_name)
+ # 鏍规嵁鍚嶇О鑾峰彇worksheet
+ worksheet = workbook.get_sheet_by_name(sheet_name)
+ # 鑾峰彇鎵�鏈夋暟鎹�
+ rows = worksheet.rows
+
+ # 灏嗙涓�琛屾暟鎹綔涓哄瓧鍏哥殑閿�
+ header = []
+ for cell in next(rows):
+ header.append(cell.value)
+
+ # 浠庣浜岃璧烽亶鍘嗘瘡涓�琛屾暟鎹紝骞跺瓨鍌ㄥ埌瀛楀吀涓�
+ data_list = []
+ for row in rows:
+ data = {}
+ for i, cell in enumerate(row):
+ data[header[i]] = cell.value
+ data_list.append(data)
+
+ return data_list
--
Gitblit v1.9.1