def assert_contains_pdfs(self, directory):
"""Helper function that asserts a directory contains pdf files."""
self.assertNotEmpty([
fname for fname in file_util.list_files(directory)
if fname.endswith('.pdf')
])
def split_shop(basedir, fn, action_type):
shop_fp = dict()
if len(file_util.list_files(basedir)) != 0:
return
with open(fn) as fp:
for line in fp:
st, shop_id, day = parse_user_action(line)
if st:
write_shop(basedir, shop_fp, shop_id, day, action_type)
for _,fp in shop_fp.items():
fp.close()
def split_shop(basedir, fn, action_type):
shop_fp = dict()
if len(file_util.list_files(basedir)) != 0:
return
with open(fn) as fp:
for line in fp:
st, shop_id, day = parse_user_action(line)
if st:
write_shop(basedir, shop_fp, shop_id, day, action_type)
for _, fp in shop_fp.items():
fp.close()
def handle_stat(split_dir, stat_dir):
if len(file_util.list_files(stat_dir)) != 0:
return
file_list = file_util.list_files(split_dir)
for fn in file_list:
stat(fn, stat_dir)
def handle_stat(split_dir, stat_dir):
if len(file_util.list_files(stat_dir)) != 0:
return
file_list = file_util.list_files(split_dir)
for fn in file_list:
stat(fn, stat_dir)