def test_clear_file(self):
from pewtils.io import FileHandler
h = FileHandler("tests/files/temp", use_s3=False)
with closing(open("tests/files/temp/temp.txt", "wb")) as output:
output.write(b"test")
h.clear_file("temp", format="txt")
files = []
for file in h.iterate_path():
files.append(file)
self.assertNotIn("temp.txt", files)
self.assertEqual(len(files), 0)
os.rmdir("tests/files/temp")
h = FileHandler("tests/files/temp", use_s3=False)
key = h.get_key_hash("temp")
with closing(open("tests/files/temp/{}.txt".format(key),
"wb")) as output:
output.write(b"test")
h.clear_file("temp", format="txt", hash_key=True)
files = []
for file in h.iterate_path():
files.append(file)
self.assertNotIn("{}.txt".format(key), files)
self.assertEqual(len(files), 0)
os.rmdir("tests/files/temp")
def test_filehandler_clear_folder(self):
from pewtils.io import FileHandler
h = FileHandler("tests/files/temp", use_s3=False)
with closing(open("tests/files/temp/temp.txt", "wb")) as output:
output.write(b"test")
h.clear_folder()
files = []
for file in h.iterate_path():
files.append(file)
self.assertEqual(len(files), 0)
os.rmdir("tests/files/temp")
def test_filehandler_iterate_path(self):
from pewtils.io import FileHandler
h = FileHandler("tests/files", use_s3=False)
files = []
for file in h.iterate_path():
files.append(file)
files = [
f for f in files if not f.endswith(".pyc")
and f not in ["__pycache__", ".DS_Store"]
]
self.assertEqual(
files,
[
"subfolder",
"__init__.py",
"example.html",
"example_stripped_simple.html",
"json.json",
"example_stripped.html",
"py.py",
],
)