## 帮阿雪写的一个小程序
---------------------------------------------------------------------------------------------------
上大学的时候,总是会由很多表格需要同学们去搞,尤其是刚开学的那个时候,显然是很烦躁,
阿雪刚开学的时候,作为班干部,表示有时候刚录表不是很熟悉经常会弄到很晚,甚至还会弄错,
这就让我很是触动,所以想帮她搞一搞,顺便增强一下我们的友谊/hhhhhh
---------------------------------------------------------------------------------------------------
思路:
第一步: 信息还是要填的,以往是同学们发给班委,然后慢慢来搞,但现在不一样的,我要求同学们把他们要填的信息写在记事本里面,内容之间用中文逗号隔开,然后在发给我就行。
第二步: 阿雪只要做的事情是把每个小朋友发的txt文本点击下载就行了,这个显然很简单的事情。
第三步: QQ个人接受的文件一般都会在 (FileRecv)这个文件夹内,每个朋友QQ存放的位置不一样我这里就不详细说明了,我自己存放的位置是:(r'D://qq//qq文档//‘我的qq号'/FileRecv//) 我们把这个路径就姑且叫做qq_path吧!这个路径我们需要事先写入代码内,我这里定义的函数是copy_file()内:具体思想对了就行。
第四步: 整个程序没啥操作难度,根据提示填入指令就行,都是批量处理,函数中我设置了时间间隔,我个人感觉这样能让程序有个停顿感,更友好一点吧。
函数库:
- os库: 用于创建文件夹,修改路径,确定路径,文件遍历等操作
- csv库: 用于表格的录入time库: 用于程序停顿感
- pandas库: 用于csv和xlsx格式之间的转换
- shutil库: 用于文件的复制
复制函数:把qq接受的文件全部转移到我们指定的位置便于操作
def copy_file(): # 将qq地址下的文本文件转移到文本存放处待用 qq_path = r'D://qq//qq文档//qq//FileRecv//' # 文本在qq的位置 for filename in os.listdir(qq_path): txt_path = os.path.join(qq_path, filename) # 拼接这二个地址 进行判断 if txt_path[-4:] == '.txt': if txt_path[-5:-4] in [str(i) for i in range(10)]: # 取后缀判断 取学号判断 是否是我们需要的txt shutil.copy(txt_path, patht) # 进行复制 os.remove(txt_path) # 删除在qq地址的文件 避免下次录入 文本过多 elif txt_path[-6:-4] in [str(i) for i in range(10, 66)]: # 班级人数个人设定 shutil.copy(txt_path, patht) # 进行复制 os.remove(txt_path) # 删除在qq地址的文件 避免下次录入 文本过多 return qq_path
信息的录入:遍历文件夹,逐一读入文本,将得到的信息放入列表
def luru(): # 把文本以学号保存 文本里面信息用中文逗号分隔 qq_path = copy_file() # 复制文本到patht listss = [] # 录入信息列表 filenamepath = [] # 文件名称地址 pathtxt = r'D://测试文件//文本存放处//' for filename in os.listdir(pathtxt): # 遍历文本存放处所有的文件 filenamepath.append(os.path.join(pathtxt, filename)) print("当前目录{}存放着{}个文件:".format(pathtxt, len(filenamepath))) if len(filenamepath) == 0: print('请确保文件正确存放在{}处:'.format(qq_path)) for i in range(len(filenamepath)): # 遍历这个列表 获得绝对地址 with open(filenamepath[i], "r", encoding='utf-8') as fo: for j in fo: listss.append(j.split(',')) return listss
在这里我加了一个小判断,如果录入文本信息为0,那么可能qq接受信息地址没有填对,路径没有很好的确定,导致这个bug
表格的录入:这个就很简单了,利用os库和csv库
def wubiaotou(): name = input("重命名该表格名称:") with open(pathb + name + ".csv", "a+", newline='') as fo: num = input("请输入表头,并用中文逗号分隔开:") headers = num.split(",") # headers = ["国籍","民族","省份"] fr_csv = csv.writer(fo) fr_csv.writerow(headers) fr_csv.writerows(lists) pathnamew = pathb + name + '.csv' time.sleep(0.5) successful(pathnamew) time.sleep(2) demo()
删除功能:象征性的写了一下,为了把功能完善一点。
def del_s(path_, fg): os.chdir(path_) # 进入path_存放处地址 biaonumber = len(os.listdir(path_)) print("当前目录下{}存在{}个文件:".format(path_, biaonumber)) if biaonumber == 0: print("无需删除,当前目录下存在{}个文件".format(biaonumber)) else: delnum = input("请输入要删除的文件名称,并用中文逗号隔开,若输入all则全部删除:").split(",") if delnum[0] == 'all': # 当索引第一个为all则删除全部 for filename in os.listdir(path_): # 遍历这个文件夹 os.remove(os.path.join(path_, filename)) time.sleep(0.5) print("已经删除{}目录下所有文件!".format(path_)) else: # 删除输入的表格名称 for i in range(len(delnum)): fn = str(delnum[i]) + fg os.remove(os.path.join(path_ + fn)) time.sleep(0.5) print("已经删除{}目录下{}个文件!".format(path_, len(delnum))) time.sleep(2) demo()
整个代码不是很难理解,仔细看看应该都能明白,相信你们,奥力给,干了兄弟们!
全部代码如下:
# -*- coding : utf-8 -*- # @Time : 2020/3/2 18:38 # @author : 沙漏在下雨 # @Software : PyCharm import os import csv import time import pandas as pd import shutil path = r'D://测试文件//' if not os.path.exists(path): # 创建这个存放文件的大目录 os.mkdir(path) patht = path + "文本存放处//" if not os.path.exists(patht): # 创建这个存放文本的文件夹 os.mkdir(patht) pathb = path + "表格存放处//" if not os.path.exists(pathb): # 创建这个存放表格的文件夹 os.mkdir(pathb) def copy_file(): # 将qq地址下的文本文件转移到文本存放处待用 qq_path = r'D://qq//qq文档//qq//FileRecv//' # 文本在qq的位置 for filename in os.listdir(qq_path): txt_path = os.path.join(qq_path, filename) # 拼接这二个地址 进行判断 if txt_path[-4:] == '.txt': if txt_path[-5:-4] in [str(i) for i in range(10)]: # 取后缀判断 取学号判断 是否是我们需要的txt shutil.copy(txt_path, patht) # 进行复制 os.remove(txt_path) # 删除在qq地址的文件 避免下次录入 文本过多 elif txt_path[-6:-4] in [str(i) for i in range(10, 66)]: shutil.copy(txt_path, patht) # 进行复制 os.remove(txt_path) # 删除在qq地址的文件 避免下次录入 文本过多 return qq_path def luru(): # 把文本以学号保存 文本里面信息用中文逗号分隔 qq_path = copy_file() # 复制文本到patht listss = [] # 录入信息列表 filenamepath = [] # 文件名称地址 pathtxt = r'D://测试文件//文本存放处//' for filename in os.listdir(pathtxt): # 遍历文本存放处所有的文件 filenamepath.append(os.path.join(pathtxt, filename)) print("当前目录{}存放着{}个文件:".format(pathtxt, len(filenamepath))) if len(filenamepath) == 0: print('请确保文件正确存放在{}处:'.format(qq_path)) for i in range(len(filenamepath)): # 遍历这个列表 获得绝对地址 with open(filenamepath[i], "r", encoding='utf-8') as fo: for j in fo: listss.append(j.split(',')) return listss lists = luru() def youbiaotou(): print("请将带有表头的表格事先放入{}处待用:".format(pathb)) names = input("请输入该表格名称:") pathnamey = pathb + names + ".xlsx" # 获得原xlsx的地址 # 使用pandas 讲 xlsx 转换为 csv 格式 便于后续操作 date_xlsx = pd.read_excel(pathnamey, index_col=0) pathnamec = pathb + names + '.csv' date_xlsx.to_csv(pathnamec, encoding='utf_8_sig') # 防止转换时候编码错误 with open(pathnamec, "a+", encoding='utf-8', newline='') as fo: fw_csv = csv.writer(fo) fw_csv.writerows(lists) # 写入多行 time.sleep(0.5) successful(pathnamec) time.sleep(2) demo() def successful(pathname): # 录入成功标识 print("表格制作成功!".center(38, '-')) print("表格位置:{}".format(pathname)) print("--" * 22) def wubiaotou(): name = input("重命名该表格名称:") with open(pathb + name + ".csv", "a+", newline='') as fo: num = input("请输入表头,并用中文逗号分隔开:") headers = num.split(",") # headers = ["国籍","民族","省份"] fr_csv = csv.writer(fo) fr_csv.writerow(headers) fr_csv.writerows(lists) pathnamew = pathb + name + '.csv' time.sleep(0.5) successful(pathnamew) time.sleep(2) demo() def helpss(): # 帮助信息 helptxt = ('说明:\t该程序主要作用是实现简单的自动录入表格功能\n\n' '格式:\t文本信息写入记事本并用中文逗号将内容隔开,命名规则为学生学号\n\n' '操作:\t只需要事先获得本台机器腾讯QQ文件保存地址即(FileRecv地址)' '将其事先写入代码内\n\n' '备注:\t程序自动将文本打包放入文本存放处(具体位置:{0})进行后续操作\n\n' '\t若调用模块有表头录入功能,请事先将待用表格放入({1})处待用\n\n' '\t最后表格将放入表格存放处(具体位置:{2})进行查看\n\n' '\t同时程序功能还实现了批量删除文本和表格功能供给选择\n\n' '\t若存在相同名称表格的不同格式' '则优先删除(.csv)格式的表格\n\n'.format(patht, pathb, pathb)) with open(path + '帮助信息.txt', "w", encoding='utf-8') as fo: fo.write(helptxt) if os.path.exists(path + '帮助信息.txt'): # newname = '帮助信息.txt' # os.rename(path+'helptxt.txt',path+newname) os.system(path + '帮助信息.txt') demo() def del_s(path_, fg): os.chdir(path_) # 进入path_存放处地址 biaonumber = len(os.listdir(path_)) print("当前目录下{}存在{}个文件:".format(path_, biaonumber)) if biaonumber == 0: print("无需删除,当前目录下存在{}个文件".format(biaonumber)) else: delnum = input("请输入要删除的文件名称,并用中文逗号隔开,若输入all则全部删除:").split(",") if delnum[0] == 'all': # 当索引第一个为all则删除全部 for filename in os.listdir(path_): # 遍历这个文件夹 os.remove(os.path.join(path_, filename)) time.sleep(0.5) print("已经删除{}目录下所有文件!".format(path_)) else: # 删除输入的表格名称 for i in range(len(delnum)): fn = str(delnum[i]) + fg os.remove(os.path.join(path_ + fn)) time.sleep(0.5) print("已经删除{}目录下{}个文件!".format(path_, len(delnum))) time.sleep(2) demo() def demo(): # 目录因素 fix = '-' * 44 title = '文本录入表格系统'.center(38, '-') helps = '使用说明请输入数值 1'.center(40, ' ') fit1 = '模板无表头请输数值 2'.center(40, ' ') fit2 = '模板有表头请输数值 3'.center(40, ' ') del1 = '删除文本请输入数值 4'.center(40, ' ') del2 = '删除表格请输入数值 5'.center(40, ' ') fix = '-' * 44 c = ['fix', 'title', 'helps', 'fit1', 'fit2', 'del1', 'del2', 'fix'] for i in c: print('{}'.format(eval(i))) control = input("请输入你的数值:") if control == '1': helpss() elif control == '2': wubiaotou() elif control == '3': youbiaotou() elif control == '4': fg = '.txt' del_s(patht, fg) elif control == '5': fg = '.csv' del_s(pathb, fg) else: print("输入选项错误!") demo()
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!