Python 关于类函数设计的一点总结
代码1
#!/usr/bin/env python
#-*-encoding:utf-8-*-
__author__ = 'shouke'
import os
class MyTestClass:
def __init__(self):
self.file_list_for_dirpath = []
# 获取指定目录下的文件
def get_files_in_dirpath(self, dirpath):
if not os.path.exists(dirpath):
print('路径：%s不存在，退出程序' % dirpath)
exit()
for name in os.listdir(dirpath):
full_path = os.path.join(dirpath, name)
if os.path.isdir(full_path):
self.get_files_in_dirpath(full_path)
else:
self.file_list_for_dirpath.append(full_path)
def get_file_list_for_dirpath(self):
return self.file_list_for_dirpath
obj = MyTestClass()
obj.get_files_in_dirpath('E:\mygit\AutoTestPlatform/UIAutotest')
print(obj.get_file_list_for_dirpath())
运行结果：
说明：
如上，get_files_in_dirpath函数目的是为了获取指定目录下的文件，按常理是函数中定义个变量，存放结果，最后直接return这个变量就可以了，但是因为涉及子目录的遍历，函数中通过self.get_files_in_dirpath对函数进行再次调用，这样一来，便无法通过简单的return方式返回结果了。
个人觉得比较不合理的方式就是按上面的，“强行”在类中定义个类属性来存放这个结果，然后再定义个函数，返回这个结果，感觉这样设计不太好，还会增加代码逻辑的模糊度。
那咋办？个人觉得比较合理的解决方案，可以使用嵌套函数。如下：
代码2
#!/usr/bin/env python
#-*-encoding:utf-8-*-
__author__ = 'shouke'
import os
class MyTestClass:
def __init__(self):
pass
# 获取指定目录下的文件
def get_files_in_dirpath(self, dirpath):
file_list_for_dirpath = []
def collect_files_in_dirpath(dirpath):
nonlocal file_list_for_dirpath
if not os.path.exists(dirpath):
print('路径：%s不存在，退出程序' % dirpath)
exit()
for name in os.listdir(dirpath):
full_path = os.path.join(dirpath, name)
if os.path.isdir(full_path):
collect_files_in_dirpath(full_path)
else:
file_list_for_dirpath.append(full_path)
collect_files_in_dirpath(dirpath)
return file_list_for_dirpath
obj = MyTestClass()
print(obj.get_files_in_dirpath('E:\mygit\AutoTestPlatform/UIAutotest'))
运行结果：
