文件夹创建
创建文件夹函数提供的功能和扩展，如下：
当文件夹不存在时，创建文件夹；
当文件夹存在时，根据参数，是否删除文件夹；
实现：
def mkdir_if_not_exist(dir_name, is_delete=False):
"""
创建文件夹
:param dir_name: 文件夹
:param is_delete: 是否删除
:return: 是否成功
"""
try:
if is_delete:
if os.path.exists(dir_name):
shutil.rmtree(dir_name)
print u'[INFO] 文件夹 "%s" 存在, 删除文件夹.' % dir_name
if not os.path.exists(dir_name):
os.makedirs(dir_name)
print u'[INFO] 文件夹 "%s" 不存在, 创建文件夹.' % dir_name
return True
except Exception as e:
print '[Exception] %s' % e
return False
复制代码
文件读取
文件读取函数提供的功能和扩展，如下：
实现便捷地文件读取功能；
当参数mode是one时，读取1行；
当参数mode是more时，读取多行；
实现：
def read_file(data_file, mode='more'):
"""
读文件, 原文件和数据文件
:return: 单行或数组
"""
try:
with open(data_file, 'r') as f:
if mode == 'one':
output = f.read()
return output
elif mode == 'more':
output = f.readlines()
return map(str.strip, output)
else:
return list()
except IOError:
return list() |
