制作无课表（一只Python菜鸟写的小程序）

苏丛

准备工作：安装xlrd, xlwt模块, 一些个人无课表

操作方法：将个人无课表与小程序放在同一个文件夹里，运行程序

注意事项：个人无课表的格式要符合模板的， 文件名为:名字.xlsx (eg 苏丛.xlsx)

模板如图


代码如下

1. import os
   
2. import xlrd
   
3. import xlwt
   
4. 
   
5. #打开当前目录
   
6. all_files = os.listdir(os.curdir)
   
7. #用来收集.xlsx文件的列表
   
8. file = []
   
9. target = '无课'
   
10. loc_dict = {}
    
11. 
    
12. #获得无课的坐标和课表所有人
    
13. def get_loc():
    
14.     #收集.xlsx文件
    
15.     for each_file in all_files:
    
16.         if os.path.splitext(each_file)[1] == '.xlsx':
    
17.             file.append(each_file)
    
18. 
    
19.     #收集坐标和名字数据
    
20.     for each_file in file:
    
21.         name = os.path.splitext(each_file)[0]
    
22.         workbook = each_file
    
23.         #打开文件
    
24.         with xlrd.open_workbook(workbook) as wb:
    
25.             #打开第一个表格
    
26.             sheet = wb.sheets()[0]
    
27. 
    
28.             #遍历表格，查找目标
    
29.             for row_num in range(sheet.nrows):
    
30.                 for col_num in range(sheet.ncols):
    
31.                     #记录坐标和名字数据
    
32.                     if sheet.cell_value(row_num, col_num) == target:
    
33.                         try:
    
34.                             loc_dict[row_num, col_num] += ' ' + name
    
35.                         except KeyError:
    
36.                             loc_dict[row_num, col_num] = name
    
37. 
    
38. def get_frame(sheet):
    
39.     #获得课表框架
    
40.     with xlrd.open_workbook(file[0]) as frame:
    
41.         sheet = frame.sheets()[0]
    
42.         frame_col = sheet.col_values(0)
    
43.         frame_row = sheet.row_values(0)
    
44. 
    
45.     return frame_col, frame_row
    
46. 
    
47. def write_name(sheet):
    
48.     #写入姓名
    
49.     for each_loc in loc_dict:
    
50.         #四个参数分别是纵坐标，横坐标，名字
    
51.         sheet.write(each_loc[0], each_loc[1], loc_dict[each_loc])
    
52. 
    
53. #写入框架
    
54. def write_frame(frame_col, frame_row, sheet):
    
55.     loc = 0
    
56.     for each_row in frame_row:
    
57.         sheet.row(0).write(loc, each_row)
    
58.         loc += 1
    
59.         
    
60.     loc = 0
    
61.     #这里不知道为什么不能跟上面写入一行一样地写入一列。
    
62.     for each_col in frame_col:
    
63.         sheet.write(loc, 0, each_col)
    
64.         loc += 1
    
65. 
    
66. def write_workbook():
    
67.     #创建一个新的课表
    
68.     new_wb = xlwt.Workbook()
    
69.     sheet = new_wb.add_sheet('无课表', cell_overwrite_ok = True)
    
70. 
    
71.     frame_col, frame_row = get_frame(sheet)
    
72. 
    
73.     write_frame(frame_col, frame_row, sheet)
    
74. 
    
75.     write_name(sheet)
    
76. 
    
77.     new_wb.save('无课表.xls')
    
78. 
    
79. if __name__ == '__main__':
    
80.     get_loc()
    
81.     write_workbook()
    
82.    
    

复制代码

还有一些问题：
不能保存为.xlsx文件
我在想加一个检测功能：检测出不符合模板的文件，然后生成一个txt文件。但还没想好怎么写。

还请在座的各位大佬不吝赐教！
有任何做的不好的都提出来，小毛病大毛病都行，或者有哪些不好的编程习惯都可以指出来
嘻嘻嘻

小庆子

赞

苏丛

小庆子 发表于 2018-5-8 09:27
赞