Skip to content

Commit 99a8878

Browse files
committed
added clean_up function. os swappd for pathlib
1 parent 758f54c commit 99a8878

File tree

1 file changed

+12
-7
lines changed

1 file changed

+12
-7
lines changed

utils/cazy_functions.py

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
1-
import os
1+
import pathlib
22
import zipfile
33

44
# Unzip zipped file
5-
def unzip(zip_file:str, output_dir:str):
6-
with zipfile.ZipFile(zip_file, 'r') as zip_ref:
7-
zip_ref.extractall(output_dir)
5+
def unzip(zip_file:pathlib.PosixPath, output_dir:pathlib.PosixPath):
6+
with zipfile.ZipFile(str(zip_file), 'r') as zip_ref:
7+
zip_ref.extractall(str(output_dir))
88

99
# Make directory if it doesn't exist
10-
def dir_exists(directory:str):
11-
if os.path.exists(directory) is False:
12-
return os.makedirs(directory)
10+
def dir_exists(directory:pathlib.PosixPath):
11+
if pathlib.Path.is_dir(directory) is False:
12+
return pathlib.Path.mkdir(directory)
1313
return None
1414

1515
# Convert each value in combined_list to an integer if possible
@@ -18,3 +18,8 @@ def num_convert(num_string:str):
1818
return int(num_string)
1919
except ValueError:
2020
return None
21+
22+
# Delete all temp files created by wget
23+
def clean_up(cwd:pathlib.PosixPath):
24+
for file in cwd.glob('**/*.tmp'):
25+
pathlib.Path.unlink(file)

0 commit comments

Comments
 (0)