| from shutil import copyfile | |
| from os import listdir | |
| from os.path import isfile, join | |
| import os | |
| import datetime | |
| def get_all_folders(folder): | |
| if check_file_exists(folder) == False: | |
| print("Cannot find the folder ", folder) | |
| return [] | |
| subfolders = [f for f in os.listdir(folder) if not isfile(join(folder,f))] | |
| return subfolders | |
| def get_all_files(folder): | |
| if check_file_exists(folder) == False: | |
| print("Cannot find the folder ", folder) | |
| return [] | |
| ori_files = [f for f in listdir(folder) if isfile(join(folder, f))] | |
| return ori_files | |
| def create_folder(folder): | |
| if not os.path.exists(folder): | |
| os.mkdir(folder) | |
| def create_folders(folder_list): | |
| for f in folder_list: | |
| create_folder(f) | |
| def replace_file_ext(fname, new_ext): | |
| ext_pos = fname.find(".") | |
| if ext_pos != -1: | |
| return fname[0:ext_pos] + "."+ new_ext | |
| else: | |
| print("Please check " + fname) | |
| def check_file_exists(fname, verbose=True): | |
| try: | |
| if not os.path.exists(fname) or get_file_size(fname) == 0: | |
| if verbose: | |
| print("file {} does not exists! ".format(fname)) | |
| return False | |
| except: | |
| print("File {} has some issue! ".format(fname)) | |
| return False | |
| return True | |
| def delete_file(fname): | |
| if check_file_exists(fname): | |
| os.remove(fname) | |
| def get_file_size(fname): | |
| return os.path.getsize(fname) | |
| def get_folder_size(folder): | |
| return sum(os.path.getsize(folder + f) for f in listdir(folder) if isfile(join(folder, f))) | |
| def get_cur_time_stamp(): | |
| return datetime.datetime.now().strftime("%d-%B-%I-%M-%p") |