整理部分代码

.gitignore

@@ -1 +1,3 @@
 venv
+__pycache__/
+

data_verification.py

@@ -25,7 +25,7 @@ async def create_upload_file(file: UploadFile = File(...)):
     # print(file.filename)
     contents = await file.read()
 
-    savename = "/data/download/" + file.filename
+    savename = "download_cache/" + file.filename
     # savename = "uploadfile/" + file.filename
     with open(savename, "wb") as f:
         f.write(contents)
@@ -42,8 +42,9 @@ async def create_upload_file(file: UploadFile = File(...)):
 
     # 读取前5行，正常应该有字段名了
     row_range = sheet[1:5]
-    for r in row_range:
-        for c in r:
+    for i, r in enumerate(row_range):
+        for j, c in enumerate(r):
+            print(f"第{i + 1 }行，第{j}列，值：{c.value}")
             if "户主编号" == c.value:
                 title_row_num = c.row
     # 获取字段名对应的列
@@ -74,25 +75,27 @@ def read_data(ws, start_row, end_row, title_dict):
 
 
 def check_poverty_causes(ws, row_num, title_dict):
-    poverty_causes = ws[f"{title_dict['主要致贫原因']}{row_num}"].value
+    main_reason = "主要致贫原因"
+    poverty_causes = ws[f"{title_dict[main_reason]}{row_num}"].value
     # 致贫原因列表
     imageTypeList = ["因病", "因学", "因残", "因自然灾害", "因意外事故", "因产业项目失败", "因务工就业不稳", "缺劳动力", "因房", "因水", "其他（填写备注）"]
     if poverty_causes not in imageTypeList:
-        ws[f"{title_dict['主要致贫原因']}{row_num}"].comment = Comment(
+        ws[f"{title_dict[main_reason]}{row_num}"].comment = Comment(
             text="21.监测对象致（返）贫风险非最新设计的风险类型", author="system"
         )
         yellow_fill = PatternFill(patternType="solid", fgColor="FFFF00")
-        ws[f"{title_dict['主要致贫原因']}{row_num}"].fill = yellow_fill
+        ws[f"{title_dict[main_reason]}{row_num}"].fill = yellow_fill
 
 
 def check_identitycard_length(ws, row_num, title_dict):
-    identitycard = ws[f"{title_dict['户主证件号码']}{row_num}"].value
+    info_number = "户主证件号码"
+    identitycard = ws[f"{title_dict[info_number]}{row_num}"].value
     if len(identitycard) not in [15, 18, 20, 22]:
-        ws[f"{title_dict['户主证件号码']}{row_num}"].comment = Comment(
+        ws[f"{title_dict[info_number]}{row_num}"].comment = Comment(
             text="31.监测对象家庭成员证件号码位数异常（证件号码非15、18、20、22位）", author="system"
         )
         yellow_fill = PatternFill(patternType="solid", fgColor="FFFF00")
-        ws[f"{title_dict['户主证件号码']}{row_num}"].fill = yellow_fill
+        ws[f"{title_dict[info_number]}{row_num}"].fill = yellow_fill
 
 
 if __name__ == "__main__":