5402285zz 发表于 2021-7-30 18:09:46

课后作业30 第四题

def content_find(target):
      all_file = os.listdir(os.curdir())
      list1 =[]
      for each_file in all_file:
            if os.path.splitext(each_file) == 'txt':
                  list1.append(os.getcwd(each_file) + os.sep + os.linesep)
            else:
                  if os.path.isdir(each_file):
                        os.chdir(os.getcwd(each_file))
                        return list1 + content_find(target)
                        os.chdir(os.curdir())


第一步中列出所有含txt的文件路径中,可不可以用递归,可以的话求大佬知道,实在不知道咋写了,不可以的话请大佬指点一下,这种情况下为啥用不来.

大马强 发表于 2021-7-30 19:28:40

这是我以前写的,你先看看,我去回顾下
import os
import easygui as eg


def mq(path, code_count, count_type, c_type):
    os.chdir(path)
    for eachfile in os.listdir(os.getcwd()):
      if os.path.isdir(eachfile):
            mq(eachfile, code_count, count_type, c_type)
            os.chdir(os.pardir)
      else:
            (f_name, f_extension) = os.path.splitext(eachfile)
            if f_extension in count_type:
                count = 0
                with open(eachfile) as fp:
                  try:
                        for each_line in fp:
                            count += 1
                  except UnicodeDecodeError:
                        pass
                if f_extension in code_count.keys():
                  code_count += count
                  c_type += 1
                else:
                  code_count.setdefault(f_extension, count)
                  c_type.setdefault(f_extension, 1)


eg.msgbox('欢迎使用大马强代码统计软件', '\@.@/', ok_button='继续')
Dir = eg.diropenbox(title='请选择要统计的目录')
eg.msgbox('正在统计代码行数...\n请等待10分钟', '提示', ok_button='妈耶!GKD!')

code_count = dict()
c_type = dict()
count_type = ['.py', '.c', '.cpp']
mq(Dir, code_count, count_type, c_type)

total = 0# 改进函数列表推导式
for each in list(code_count.keys()):
    total += code_count

with open('统计.txt', 'w', encoding='utf-8') as fp:
    for j in list(code_count.keys()):
      fp.write('【%s】源文件%d个,源代码%d\n' % (j, c_type, code_count))
with open('统计.txt', encoding='utf-8') as fp:
    msg = '您当前已经积累了%d行代码,完成进度:%.2f%% \n离10w行代码还有%d行,请继续努力!' % (
      total, total/1000, 100000-total)# 百分数表示【%%】
    eg.textbox(msg, '统计结果', text=fp.read())

大马强 发表于 2021-7-30 20:21:28

本帖最后由 大马强 于 2021-7-30 20:30 编辑

import os
def count_txt(cur_path):

    # 改变当前工作位置
    os.chdir(cur_path)
    for each in os.listdir(cur_path):

      if os.path.isdir(each):
            # 递归调用,进入这个目录
            count_txt(each)
            # 目录遍历完毕,返回将工作区改为父目录,返回到上一层
            os.chdir(os.pardir)

      else:
            # 条件匹配
            if os.path.splitext(each)[-1] == ".txt":
                print(each)


# 当期前位置
cur_path = os.getcwd()
count_txt(cur_path)

大马强 发表于 2021-7-30 20:30:04

你这个就是递归了呀
看这里
return list1 + content_find(target)
就是递归调用了content_find函数

5402285zz 发表于 2021-7-30 21:10:15

大马强 发表于 2021-7-30 20:30
你这个就是递归了呀
看这里



大佬可以给个联系方式吗?真心想跟您学习.

5402285zz 发表于 2021-7-30 21:11:04

大马强 发表于 2021-7-30 20:30
你这个就是递归了呀
看这里



.txt
Traceback (most recent call last):
File "D:\phython\课后作业30 4.py", line 15, in <module>
    list2 = content_find(target)
File "D:\phython\课后作业30 4.py", line 10, in content_find
    os.chdir(os.getcwd(each_file))
TypeError: nt.getcwd() takes no arguments (1 given)


出现了这样的错误

大马强 发表于 2021-7-30 21:18:14

我也是入门菜鸡,加我qq1432659208

大马强 发表于 2021-7-30 21:20:16

s.chdir(os.getcwd(each_file))
改成
s.chdir(os.getcwd())
试试,我记得应该不带参数的

大马强 发表于 2021-7-30 22:51:43

import os


def content_find(target, path, file_list):

    os.chdir(path)
    all_file = os.listdir(os.getcwd())
    for each_file in all_file:
      # 先判断是否为目录
      if os.path.isdir(each_file):
            content_find(target, each_file, file_list)
            os.chdir(os.pardir)

      else:
            # 再判断是否符合要求
            # 分离会有两个值[文件名,拓展名] 用 -1 索引取最后一个元素
            if os.path.splitext(each_file)[-1] == target:
                # list1.append(os.getcwd(each_file) + os.sep + os.linesep)
                # 取文件名
                file_list.append(each_file)


cur_path = os.getcwd()
print("您当前的位置是:", cur_path)
target = input("请问要统计什么类型的文件:[格式:.txt,.py,.c]")
list2 = []
content_find(target, cur_path, list2)
for each_file in list2:
    print(each_file)

诶嘿,我改了改你的代码,你看看
页: [1]
查看完整版本: 课后作业30 第四题