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'))

运行结果：