forked from LiveCarta/BookConverter
36 lines
892 B
Python
36 lines
892 B
Python
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
|