forked from LiveCarta/BookConverter
add checking for directories
This commit is contained in:
35
src/util/check_dirs.py
Normal file
35
src/util/check_dirs.py
Normal 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
|
||||
Reference in New Issue
Block a user