最近在学习excel,突发奇想给自己找些题库练练
代码很简单,需要的人不可以直接复制代码,因为文件路径不一样
主要功能就是每天给自己生成自己想练习的题目数量,然后直接生成文件
重新上传了V2.0版本的代码
1.增加了可以循环写入的函数
2.Excel有两种后缀,改进了判断后缀名称写入
需要Excel题库的可以私信联系我
import datetime
import random
import time
import os
# 定义一个判断在写入时是否有同名文件存在,有的话,就+1,反复调用
def exists():
global i
if os.path.isfile(f'D:\\{today}练习{i + 1}.xlsx'):
i += 1
exists()
elif os.path.isfile(f'D:\\{today}练习{i + 1}.xls'):
i += 1
exists()
else:
write()
# 定义一个写入函数,因为源文件名称有两种后缀,所以,分情况写入
def write():
global choice
if choice.endswith('.xlsx'):
with open(f'D:\\{today}练习{i + 1}.xlsx', 'wb')as f2:
f2.write(result)
f2.flush()
elif choice.endswith('.xls'):
with open(f'D:\\{today}练习{i + 1}.xls', 'wb')as f2:
f2.write(result)
f2.flush()
# 获取今天的时间并格式化输出
today = datetime.datetime.today().strftime('%Y-%m-%d')
# 使用os模块列出题库列表
ku_list = os.listdir(r'D:\Excel训练库') # 题库路径
# 循环5次,需要几套题,在这里改数字
for i in range(5):
# 随机选择题库列表中的一个文件
choice = random.choice(ku_list)
print(choice) # 这里可以选择不打印,主要看选中了哪个文件
# 使用二进制读取选中的文件
with open(f'D:\Excel训练库\\{choice}', 'rb')as fp:
result = fp.read()
time.sleep(0.4) # 这里可以选择不打印,没什么太大作用
exists()