编写python程序，查找指定后缀文件并压缩打包

第十三次py课

2.查找出工作目录下的所有 Python 程序文件（以.py 结尾的文件），然后将所有 Python程序复制到新建文件夹 my_python 下，最后把 my_python 文件夹进行压缩，将压缩后的文件命名为 all_python.zip，并存储在计算机桌面。
要求：查找文件函数递归调用以查找其子文件夹中的文件；统计当前工作目录下的.py文件。
提示：使用os、shutil模块中的相关函数完成。
示例如下：

代码如下：

（请不要白嫖作业，可以的话点一下赞或者关注qwq）

import os,shutil
my = os.path.abspath(r'D:\\study3')
cp = os.path.abspath(r'D:\\my_python')

if not os.path.exists(cp):
os.makedirs(cp)
t = 0
q = 0
a = input('请输入需要查找到的文件扩展名：')

if os.path.exists(my):
for root, dirs, files in os.walk(my):
for file in files :
src_file = os.path.join(root, file)
if a in src_file:
q += 1
print('\n'
'==============共找到%d文件==============='
'\n' % (q))
if os.path.exists(my):
for root, dirs, files in os.walk(my):
for file in files :
src_file = os.path.join(root, file)
if a in src_file:
t = t + 1
print(t,src_file)
shutil.copy(src_file, cp)

shutil.make_archive('all_python','zip',base_dir='D:\\my_python')
shutil.move('D:\\study3\\Chap13\\all_python.zip','C:\\Users\\10161\\Desktop')