【统计代码行数】如何判断一个字符串是否已经在txt文档里？

xjtu_wong

想实现统计文件夹内总共的代码行数。实现将代码名称与行数这一字典保存成文件时发现没法判断某一字符串是否在文本文件里，应该如何解决呢？

1. # -*- coding: utf-8 -*-
   
2. """
   
3. Created on Tue Jul 23 10:01:46 2019
   
4. 
   
5. @author: u
   
6. """
   
7. 
   
8. import glob
   
9. import datetime
   
10. import re
    
11. 
    
12. nowTime=datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')#现在
    
13. file_name1 = r'Total_code_count.txt'
    
14. file_name2 = r'Code_count_details.txt'
    
15. files = glob.glob(r'D:\FishC_learning\*.py')
    
16. len_dict = {}
    
17. 
    
18. total_count = 0
    
19. for each_py in files:
    
20.      this_count = 0
    
21.      with open(each_py,'r', encoding='UTF-8') as py:
    
22.             for each_line in py:
    
23.                    this_count += 1
    
24.             len_dict[each_py] = this_count
    
25.      total_count += this_count
    
26.      
    
27. #print(len_dict.items())
    
28. print(total_count)
    
29. 
    
30. with open(file_name1,'a+') as f1:
    
31.        strings =nowTime + '时刻本文件夹的代码总数为：' + str(total_count) + '\n'
    
32.        print(strings)
    
33.        f1.writelines(strings)
    
34. 
    
35. with open(file_name2,'r+') as f2:
    
36.        for each_py in len_dict.keys():
    
37.               if each_py not in f2.read():
    
38.                      strings2 = str(each_py) + ' 的代码数为：' + str(len_dict[each_py]) + '\n'
    
39.                      print(strings2)
    
40.                      f2.writelines(strings2)

复制代码

xjtu_wong

就在最下面的代码段里：

1. for each_py in len_dict.keys():
   
2.               if each_py not in f2.read():

复制代码

这个功能失效了，尝试用find方法也不行。
用正则表达式传出什么/F错误，从来没见过。。。
有毒

DT_Nelson

用os模块中的os.listdir()函数可以得到一个路径下所有的文件名称
然后用正则表达式就没问题了