add checking for directories

This commit is contained in:
Jeniamakarchik
2020-02-25 12:58:24 +03:00
parent a080cb962b
commit 70575c803d
2 changed files with 36 additions and 5 deletions

35
src/util/check_dirs.py Normal file
View File

@@ -0,0 +1,35 @@
import argparse
import os
def parse_args():
parser = argparse.ArgumentParser(description="Utility for folders's clean up.")
parser.add_argument('-f', '--folders', type=str, nargs='*', help='Names of the folders to be cleaned.')
args = parser.parse_args()
return args
def check_dir(dir_path):
if not os.path.exists(dir_path):
try:
os.mkdir(dir_path)
except OSError as exc:
raise exc
if __name__ == '__main__':
folders = parse_args().folders
if not folders:
folders = ['docx', 'html', 'json', 'logs', 'config']
folder_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
folders = [os.path.join(folder_path, folder) for folder in folders]
try:
for folder in folders:
check_dir(folder)
except OSError as exc:
print(exc)
raise