def copy_to_temp_directory(path_folder: str) -> str:
"""copy directory to OS temp directory
Arguments:
path_folder {str} -- directory path
Returns:
str -- temp directory path
"""
# generate hash for temp folder
hash_folder = get_hash()
# get os temp directory
temp_dir = tools.get_temp_directory()
# create temp directory
temp_dir = os.path.join(temp_dir, hash_folder)
try:
# create directory
tools.create_directory(temp_dir)
# copy directory to OS temp directory
tools.copy_directory(path_folder, temp_dir)
except IOError as error:
print(error)
return False
return temp_dir
def copy_to_organized_directory(dir_src: str, dir_dst: str) -> bool:
"""copy source directory to organized directory
Get content from source directory and copy to a new directory
into new folders with the extentions name
Arguments:
dir_src {str} -- source directory
dir_dst {str} -- destination directory
Raises:
IOError: path dir_src not found
IOError: path dir_dst not found
Returns:
bool -- if successful return True
"""
if not checks.check_exists(dir_src):
raise IOError("Directory '{0}' not found.".format(dir_src))
if not checks.check_exists(dir_dst):
raise IOError("Directory '{0}' not found.".format(dir_dst))
# copy files temp to new directory
list_files = get_files(dir_src)
for file in list_files:
basename = os.path.basename(file)
search = re.search(r"\.([A-Za-z0-9]+)$", basename)
if search:
extension = search.group(1)
# check = [ext for ext in extensions if search.group(1) == ext]
directory_extension = os.path.join(dir_dst, extension)
if not checks.check_exists(directory_extension):
tools.create_directory(directory_extension)
# build target directory
target_directory = os.path.join(directory_extension, basename)
# copy file
tools.copy_file(file, target_directory)
return True
def copy_to_temp_directory(path_folder: str) -> str:
"""copy directory to OS temp directory
Arguments:
path_folder {str} -- directory path
Returns:
str -- temp directory path
"""
if not path_folder:
return None
if not os.path.exists(path_folder):
return None
random_number = gen_random_bit_number(256)
# generate hash for temp folder
hash_folder = get_hash(random_number)
# get os temp directory
temp_dir = tools.get_temp_directory()
# create temp directory
temp_dir = os.path.join(temp_dir, hash_folder)
try:
# create directory
tools.create_directory(temp_dir)
# copy directory to OS temp directory
tools.copy_directory(path_folder, temp_dir)
except IOError as error:
print(error)
return None
return temp_dir
def run(path_folder, create_copy=True):
"""main script run
Arguments:
path_folder {str} -- folder to organize
Keyword Arguments:
create_copy {bool} -- create a folder copy (default: {True})
Raises:
IOError: if folder doesn't exists
"""
print(f"\n[Junkfile] - Parameters: ")
print(f"[Junkfile] - path_folder: {path_folder}")
print(f"[Junkfile] - create_copy: {create_copy}\n")
print("[Junkfile] - executing...")
# check if path exits
if not checks.check_exists(path_folder):
raise Exception("Please choose a directory.")
# check if is not inthe blacklist directory
if path_folder in get_blacklist_directories():
raise Exception("Directory may not be selected. Try another directory")
# remove last slash
path_folder = sanitize_path(path=path_folder)
'''directory contextmanager
check if path has permission to change, set as current directory and
change to old current directory at the end
'''
with current_directory_handler(path_folder):
# copy folder to to OS temp directory
temp_dir = copy_to_temp_directory(path_folder=path_folder)
# make path
new_directory = build_directory_path_with_date(path_folder=path_folder)
# create a new folder
if not tools.create_directory(new_directory):
raise Exception(
f"new_directory could not be created. new_directory: '{new_directory}'"
)
# copy to new folder
copy_to_organized_directory(dir_src=temp_dir, dir_dst=new_directory)
# delete folders
if checks.check_exists(temp_dir):
tools.delete_directory(temp_dir)
# check if is a copy
if not create_copy:
tools.delete_directory(path_folder)
tools.move_directory(new_directory, path_folder)
if create_copy:
print(f"[Junkfile] - Was created new directory: {new_directory}")
print("[Junkfile] - executed!")
return True
def test_create_directory(test_input, expected_input):
result = tools.create_directory(test_input)
assert result == expected_input, f"Returns {result} instead {expected_input}"
tools.delete_directory(testdata.directory_create[0][0])