mirror of
https://github.com/kevinveenbirkenbach/directory-content-scanner.git
synced 2024-11-24 13:51:04 +01:00
Accepted now all paths
This commit is contained in:
parent
100f5267f0
commit
fc19546378
65
scan.py
65
scan.py
@ -1,67 +1,66 @@
|
|||||||
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
|
return False
|
||||||
if file_filters:
|
|
||||||
if not any(file.endswith(file_type) for file_type in file_filters):
|
|
||||||
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):
|
||||||
|
"""Print the content of a file."""
|
||||||
|
try:
|
||||||
|
with open(file_path, 'r') as f:
|
||||||
|
content = f.read()
|
||||||
|
print(f"======== File Path: {file_path} ========")
|
||||||
|
print(content)
|
||||||
|
print("==================================\n")
|
||||||
|
except UnicodeDecodeError:
|
||||||
|
print(f"Warning: Could not read file due to encoding issues: {file_path}")
|
||||||
|
exit(1)
|
||||||
|
|
||||||
def scan_and_print(directory, file_filters, ignore_strings, ignore_hidden, verbose):
|
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):
|
for root, dirs, files in os.walk(directory):
|
||||||
filter_directories(dirs, ignore_strings, ignore_hidden, verbose)
|
filter_directories(dirs, ignore_strings, ignore_hidden)
|
||||||
|
|
||||||
for file in files:
|
for file in files:
|
||||||
if should_print_file(file, file_filters, ignore_strings, ignore_hidden):
|
if should_print_file(file, file_filters, ignore_strings, ignore_hidden):
|
||||||
file_path = os.path.join(root, file)
|
print_file_content(os.path.join(root, file))
|
||||||
try:
|
|
||||||
with open(file_path, 'r') as f:
|
|
||||||
content = f.read()
|
|
||||||
print(f"======== File Path: {file_path} ========")
|
|
||||||
print(content)
|
|
||||||
print("==================================\n")
|
|
||||||
except UnicodeDecodeError:
|
|
||||||
print(f"Warning: Could not read file due to encoding issues: {file_path}")
|
|
||||||
exit(1)
|
|
||||||
|
|
||||||
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()
|
||||||
|
Loading…
Reference in New Issue
Block a user