Restructure project directory, python files are now in their own folder
This commit is contained in:
353
bin/quickdel
353
bin/quickdel
@@ -1,179 +1,218 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
quickdel - delete any file matching a query
|
||||
#!/usr/bin/env bash
|
||||
|
||||
Dependencies
|
||||
============
|
||||
* fd
|
||||
* python-termcolor
|
||||
# Cleanup
|
||||
set -e
|
||||
trap 'exit 1' SIGINT
|
||||
|
||||
Command-Line Arguments
|
||||
======================
|
||||
* -d, --directories-only
|
||||
* -D, --directory TODO: implement this
|
||||
* -e, --empty-only
|
||||
* -E, --extension
|
||||
* -f, --files-only
|
||||
* -F, --force-directory-delete
|
||||
* -i, --no-ignore
|
||||
* -I, --no-ignore-vcs
|
||||
* -l, --links-only
|
||||
"""
|
||||
# ========== Source library ==========
|
||||
LIBDIR="/usr/share/file-scripts/"
|
||||
|
||||
import argparse
|
||||
import os
|
||||
import os.path
|
||||
import re
|
||||
import shutil
|
||||
import file_scripts.error as error
|
||||
import file_scripts.search as search
|
||||
|
||||
from termcolor import colored
|
||||
for f in "$LIBDIR"/*.sh; do
|
||||
source "${f}"
|
||||
done
|
||||
|
||||
# ========== Constants ==========
|
||||
fd_opts = ["--hidden"]
|
||||
# Matches 'y' or 'yes' only, ignoring case
|
||||
USER_RESPONSE_YES = r"^[Yy]{1}([Ee]{1}[Ss]{1})?$"
|
||||
RED=$'\e[1;31m'
|
||||
GREEN=$'\e[1;32m'
|
||||
BLUE=$'\e[1;34m'
|
||||
YELLOW=$'\e[1;33m'
|
||||
WHITE_BOLD=$'\e[1;37m'
|
||||
RESET=$'\e[0;0m'
|
||||
|
||||
declare -a extra_fd_opts
|
||||
declare -a typeopts
|
||||
|
||||
# ========== Functions ==========
|
||||
def color_file(filename):
|
||||
"""Return correct color code for filetype of filename.
|
||||
# ========== Helper functions ==========
|
||||
function help() {
|
||||
cat << HELPMESSAGE
|
||||
$(basename "$0") $MAJOR_VERSION.$MINOR_VERSION.$PATCH_VERSION
|
||||
|
||||
Example
|
||||
-------
|
||||
>>> color_file('Test File', 'red')
|
||||
'\x1b[31mTest String\x1b[0m'
|
||||
Usage: $(basename "$0") [-h] [-d] [-e] [-E ext] [-f] [-F] [-I] [-i] [-l] patterns [patterns ...]
|
||||
|
||||
:param filename: file to determine color output for
|
||||
:type filename: str
|
||||
:return: filename with ANSII escape codes for color
|
||||
:rtype: str
|
||||
"""
|
||||
if os.path.isdir(filename):
|
||||
return colored(filename, "blue")
|
||||
elif os.path.islink(filename):
|
||||
return colored(filename, "green")
|
||||
else:
|
||||
return filename
|
||||
positional arguments:
|
||||
patterns file matching patterns
|
||||
|
||||
options:
|
||||
-h, --help show this help message and exit
|
||||
-d, --directories-only
|
||||
filter results to directories
|
||||
-e, --empty-only filter results to empty files and directories
|
||||
-E ext, --extension ext
|
||||
file extension
|
||||
-f, --files-only filter results to files
|
||||
-F, --force-directory-delete
|
||||
do not ignore non-empty directories, delete anyways
|
||||
-I, --no-ignore-vcs do not ignore .gitignore
|
||||
-i, --no-ignore do not ignore .gitignore and .fdignore
|
||||
-l, --links-only filter results to symlinks
|
||||
HELPMESSAGE
|
||||
}
|
||||
|
||||
# ========== Main Script ==========
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument(
|
||||
"-d",
|
||||
"--directories-only",
|
||||
action="store_const",
|
||||
const=["--type", "directory"],
|
||||
dest="fd_extra_opts",
|
||||
help="filter results to directories",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-e",
|
||||
"--empty-only",
|
||||
action="store_const",
|
||||
const=["--type", "empty"],
|
||||
dest="fd_extra_opts",
|
||||
help="filter results to empty files and directories",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-E",
|
||||
"--extension",
|
||||
action="append",
|
||||
dest="extensions",
|
||||
help="file extension",
|
||||
metavar="ext",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-f",
|
||||
"--files-only",
|
||||
action="store_const",
|
||||
const=["--type", "file"],
|
||||
dest="fd_extra_opts",
|
||||
help="filter results to files",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-F",
|
||||
"--force-directory-delete",
|
||||
action="store_true",
|
||||
help="do not ignore non-empty directories, delete anyways",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-I",
|
||||
"--no-ignore-vcs",
|
||||
action="store_const",
|
||||
const="--no-ignore-vcs",
|
||||
dest="fd_extra_opts",
|
||||
help="do not ignore .gitignore",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-i",
|
||||
"--no-ignore",
|
||||
action="store_const",
|
||||
const="--no-ignore",
|
||||
dest="fd_extra_opts",
|
||||
help="do not ignore .gitignore and .fdignore",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-l",
|
||||
"--links-only",
|
||||
action="store_const",
|
||||
const=["--type", "symlink"],
|
||||
dest="fd_extra_opts",
|
||||
help="filter results to symlinks",
|
||||
)
|
||||
parser.add_argument("patterns", nargs="+", help="file matching patterns")
|
||||
# $1 is the output string, $2 is the color
|
||||
function color_output() {
|
||||
case "$2" in
|
||||
'red')
|
||||
printf "$RED%s\n" "$1"
|
||||
;;
|
||||
'green')
|
||||
printf "$GREEN%s\n" "$1"
|
||||
;;
|
||||
'blue')
|
||||
printf "$BLUE%s\n" "$1"
|
||||
;;
|
||||
'yellow')
|
||||
printf "$YELLOW%s\n" "$1"
|
||||
;;
|
||||
'white')
|
||||
printf "$WHITE_BOLD%s\n" "$1"
|
||||
;;
|
||||
'reset')
|
||||
printf "$RESET%s\n" "$1"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
args = parser.parse_args()
|
||||
# Color files blue and directories green
|
||||
function color_path() {
|
||||
if [[ -f "$1" ]]; then
|
||||
color_output "$1" 'blue'
|
||||
elif [[ -d "$1" ]]; then
|
||||
color_output "$1" 'green'
|
||||
fi
|
||||
}
|
||||
|
||||
if args.fd_extra_opts is not None:
|
||||
fd_opts.extend(args.fd_extra_opts)
|
||||
if args.extensions is not None:
|
||||
for ext in args.extensions:
|
||||
fd_opts.extend(["--extension", ext])
|
||||
# Delete path using correct command
|
||||
# Parameters:
|
||||
# - $1: path
|
||||
# - $2: force delete boolean flag
|
||||
function delete() {
|
||||
exit 1
|
||||
}
|
||||
|
||||
files = set()
|
||||
for p in args.patterns:
|
||||
files.update(
|
||||
search.find_files(opts=fd_opts, capture_text=True, pattern=p).splitlines()
|
||||
)
|
||||
files = sorted(files)
|
||||
while true; do
|
||||
case "${1}" in
|
||||
'-d' | '--directories-only')
|
||||
typeopts+=(--type directory)
|
||||
shift
|
||||
continue
|
||||
;;
|
||||
'-e' | '--empty-only')
|
||||
typeopts+=(--type empty)
|
||||
shift
|
||||
continue
|
||||
;;
|
||||
'-E' | '--extension')
|
||||
EXT="${2}"
|
||||
case "${EXT}" in
|
||||
"")
|
||||
exit 1
|
||||
;;
|
||||
-*)
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
extra_fd_opts+=('--extension' "$EXT")
|
||||
shift 2
|
||||
continue
|
||||
;;
|
||||
--extension=*)
|
||||
EXT="${1#*=}"
|
||||
case "${EXT}" in
|
||||
"")
|
||||
exit 1
|
||||
;;
|
||||
-*)
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
extra_fd_opts+=('--extension' "$EXT")
|
||||
shift
|
||||
continue
|
||||
;;
|
||||
'-f' | '--files-only')
|
||||
typeopts+=(--type file)
|
||||
shift
|
||||
continue
|
||||
;;
|
||||
'-F' | '--force-directory-delete')
|
||||
rm_force='--force'
|
||||
shift
|
||||
continue
|
||||
;;
|
||||
'-i' | '--no-ignore-vcs')
|
||||
extra_fd_opts+=('--no-ignore-vcs')
|
||||
shift
|
||||
continue
|
||||
;;
|
||||
'-I' | '--no-ignore')
|
||||
extra_fd_opts+=('--no-ignore')
|
||||
shift
|
||||
continue
|
||||
;;
|
||||
'-l' | '--links-only')
|
||||
typeopts+=(--type symlink)
|
||||
shift
|
||||
continue
|
||||
;;
|
||||
'-h' | '--help')
|
||||
help
|
||||
exit
|
||||
;;
|
||||
--)
|
||||
shift
|
||||
break
|
||||
;;
|
||||
-*)
|
||||
printf '%s\n' "Unknown option: ${1}" >&2
|
||||
exit 1
|
||||
;;
|
||||
*)
|
||||
break
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
if files == []:
|
||||
print("No results found, exiting")
|
||||
exit(error.E_NO_RESULTS)
|
||||
# Interpret options
|
||||
if [[ -z "$*" ]]; then
|
||||
help
|
||||
fi
|
||||
|
||||
# Pretty print all filenames
|
||||
for index, filename in enumerate([color_file(f) for f in files], 1):
|
||||
print(f"{index}. {filename}")
|
||||
# Padding line
|
||||
print()
|
||||
declare -a files pattern_results
|
||||
|
||||
try:
|
||||
user_response = input("Would you like to delete these files? [y/N]: ")
|
||||
except KeyboardInterrupt:
|
||||
exit(error.E_INTERRUPT)
|
||||
for pattern in "$@"; do
|
||||
readarray -d $'\n' -t pattern_results <<< "$(fd "${DEFAULT_FD_OPTS[@]}" "${extra_fd_opts[@]}" "${typeopts[@]}" "$pattern")"
|
||||
files+=("${pattern_results[@]}")
|
||||
done
|
||||
|
||||
if re.match(USER_RESPONSE_YES, user_response) is None:
|
||||
print("Operation cancelled")
|
||||
exit(error.E_USER_RESPONSE_NO)
|
||||
# If nothing was found
|
||||
if [[ -z "${files[*]}" ]]; then
|
||||
color_output 'No files found, exiting' 'yellow'
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Remove files first
|
||||
for f in [fi for fi in files if os.path.isfile(fi)]:
|
||||
os.remove(f)
|
||||
# Sort list of paths before printing
|
||||
readarray -t paths < <(printf '%s\n' "${files[@]}" | sort --unique)
|
||||
|
||||
# Check -f, --force-directory-delete option
|
||||
# shutil.rmtree forcibly removes directories
|
||||
# os.rmdir removes directories only if they are empty
|
||||
rmdir_func = shutil.rmtree if args.force_directory_delete else os.rmdir
|
||||
# Print list of files found matching criteria
|
||||
i=1
|
||||
for p in "${paths[@]}"; do
|
||||
printf '%s. %s\n' "$(color_output $i 'white')" "$(color_path "$p")"
|
||||
i=$((i + 1))
|
||||
done
|
||||
|
||||
for d in filter(os.path.isdir, files):
|
||||
try:
|
||||
rmdir_func(d)
|
||||
except OSError:
|
||||
print(
|
||||
f"{colored('Warning', 'yellow')}: {colored(d, 'blue')} is not empty, not deleting directory"
|
||||
)
|
||||
# Padding between files and prompt
|
||||
#color_output '' reset
|
||||
|
||||
print(colored("\nDeletions complete", "green"))
|
||||
read -r -n 1 -p 'Would you like to delete these files? [y/N]: ' user_response
|
||||
# Padding between prompt and output
|
||||
echo ''
|
||||
|
||||
if [[ "$user_response" =~ (y|Y) ]]; then
|
||||
for p in "${paths[@]}"; do
|
||||
if [[ -d "$p" ]]; then
|
||||
rm --recursive "$rm_force" --verbose -- "$p" || printf '%s %s\n' "$(color_output "Unable to remove path:" 'red')" "$(color_path "$p")"
|
||||
else
|
||||
rm "$rm_force" --verbose -- "$p" || printf '%s %s\n' "$(color_output "Unable to remove path:" 'red')" "$(color_path "$p")"
|
||||
fi
|
||||
done
|
||||
fi
|
||||
|
Reference in New Issue
Block a user