kevinveenbirkenbach/directory-content-scanner
1
0
Fork 0
mirror of https://github.com/kevinveenbirkenbach/directory-content-scanner.git synced 2024-09-20 00:13:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

Accepted now all paths

Browse Source
This commit is contained in:
Kevin Veen-Birkenbach 2023-09-14 08:02:05 +02:00
parent 100f5267f0
commit fc19546378
1 changed files with 32 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
scan.py
View File

@ -1,40 +1,24 @@
import os import os
import argparse import argparse
def filter_directories(dirs, ignore_strings, ignore_hidden, verbose): def filter_directories(dirs, ignore_strings, ignore_hidden):
"""Filter out directories based on ignore criteria.""" """Filter out directories based on ignore criteria."""
if ignore_hidden: if ignore_hidden:
dirs[:] = [d for d in dirs if not d.startswith('.')] dirs[:] = [d for d in dirs if not d.startswith('.')]
dirs[:] = [d for d in dirs if not any(ig in d for ig in ignore_strings)] dirs[:] = [d for d in dirs if not any(ig in d for ig in ignore_strings)]
if verbose:
print(f"Filtered directories: {dirs}")
def should_print_file(file, file_filters, ignore_strings, ignore_hidden): def should_print_file(file, file_filters, ignore_strings, ignore_hidden):
"""Determine if a file should be printed based on filters.""" """Determine if a file should be printed based on filters."""
# Ignore hidden files
if ignore_hidden and file.startswith('.'): if ignore_hidden and file.startswith('.'):
return False return False
if file_filters and not any(file.endswith(file_type) for file_type in file_filters):
# Filter by file type
if file_filters:
if not any(file.endswith(file_type) for file_type in file_filters):
return False return False
# Ignore files based on ignore_strings
if any(ignore_str in file for ignore_str in ignore_strings): if any(ignore_str in file for ignore_str in ignore_strings):
return False return False
return True return True
def print_file_content(file_path):
def scan_and_print(directory, file_filters, ignore_strings, ignore_hidden, verbose): """Print the content of a file."""
for root, dirs, files in os.walk(directory):
filter_directories(dirs, ignore_strings, ignore_hidden, verbose)
for file in files:
if should_print_file(file, file_filters, ignore_strings, ignore_hidden):
file_path = os.path.join(root, file)
try: try:
with open(file_path, 'r') as f: with open(file_path, 'r') as f:
content = f.read() content = f.read()
@ -45,23 +29,38 @@ def scan_and_print(directory, file_filters, ignore_strings, ignore_hidden, verbo
print(f"Warning: Could not read file due to encoding issues: {file_path}") print(f"Warning: Could not read file due to encoding issues: {file_path}")
exit(1) exit(1)
def handle_directory(directory, file_filters, ignore_strings, ignore_hidden, verbose):
"""Handle scanning and printing for directories."""
for root, dirs, files in os.walk(directory):
filter_directories(dirs, ignore_strings, ignore_hidden)
for file in files:
if should_print_file(file, file_filters, ignore_strings, ignore_hidden):
print_file_content(os.path.join(root, file))
elif verbose: elif verbose:
print(f"Skipped file: {file}") print(f"Skipped file: {file}")
def handle_file(file_path, file_filters, ignore_strings, ignore_hidden):
"""Handle scanning and printing for individual files."""
if should_print_file(os.path.basename(file_path), file_filters, ignore_strings, ignore_hidden):
print_file_content(file_path)
def main(): def main():
parser = argparse.ArgumentParser(description="Scan directories and print file contents.") parser = argparse.ArgumentParser(description="Scan directories and print file contents.")
parser.add_argument("directories", nargs='+', help="List of directories to scan.") parser.add_argument("paths", nargs='+', help="List of files or directories to scan.")
parser.add_argument("-f", "--filetype", nargs='+', default=[], help="Filter by file types (e.g., .txt .log).") parser.add_argument("-f", "--filetype", nargs='+', default=[], help="Filter by file types (e.g., .txt .log).")
parser.add_argument("-i", "--ignore", nargs='+', default=[], help="Ignore files and folders containing these strings.") parser.add_argument("-i", "--ignore", nargs='+', default=[], help="Ignore files and folders containing these strings.")
parser.add_argument("--ignore-hidden", action='store_true', help="Ignore hidden directories.") parser.add_argument("--ignore-hidden", action='store_true', help="Ignore hidden directories and files.")
parser.add_argument("-v", "--verbose", action='store_true', help="Enable verbose mode.") parser.add_argument("-v", "--verbose", action='store_true', help="Enable verbose mode.")
args = parser.parse_args() args = parser.parse_args()
for directory in args.directories: for path in args.paths:
if not os.path.isdir(directory): if os.path.isdir(path):
print(f"Error: {directory} is not a valid directory.") handle_directory(path, args.filetype, args.ignore, args.ignore_hidden, args.verbose)
elif os.path.isfile(path):
handle_file(path, args.filetype, args.ignore, args.ignore_hidden)
else:
print(f"Error: {path} is neither a valid file nor a directory.")
exit(1) exit(1)
scan_and_print(directory, args.filetype, args.ignore, args.ignore_hidden, args.verbose)
if __name__ == "__main__": if __name__ == "__main__":
main() main()