RomanNum3ral
/
python-forensics-handbook
mirror of https://github.com/chapinb/python-forensics-handbook.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Updated recursion example to provide iterator

This commit is contained in:
Chapin Bryce 2019-06-16 17:50:38 -04:00
parent 9270858828
commit 89bf6cbda6
1 changed files with 10 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
sections/section_01/recursion_example.py
View File

@ -81,10 +81,6 @@ def list_directory(path):
print(f"\t{entry}") print(f"\t{entry}")
def iterate_files(path): def iterate_files(path):
"""Iterate over the `path` recursively."""
num_files = 0
num_py_files = 0
num_folders = 0
# Though `os.walk()` exposes a list of directories in the # Though `os.walk()` exposes a list of directories in the
# current `root`, it is rarely used since we are generally # current `root`, it is rarely used since we are generally
# interested in the files found within the subdirectories. # interested in the files found within the subdirectories.
@ -92,25 +88,24 @@ def iterate_files(path):
# DO NOT NAME `dirs` as `dir` since `dir` is a reserved word! # DO NOT NAME `dirs` as `dir` since `dir` is a reserved word!
for root, dirs, files in os.walk(os.path.abspath(path)): for root, dirs, files in os.walk(os.path.abspath(path)):
# Both `dirs` and `files` are lists containing all entries # Both `dirs` and `files` are lists containing all entries
# at the current `root`. This means we can quickly count # at the current `root`.
# the number of files and folders by taking the `len()`.
num_folders += len(dirs)
num_files += len(files)
for fentry in files: for fentry in files:
# To effectively reference a file, you should include # To effectively reference a file, you should include
# the below line which creates a full path reference # the below line which creates a full path reference
# to the specific file, regardless of how nested it is # to the specific file, regardless of how nested it is
file_entry = os.path.join(root, fentry) file_entry = os.path.join(root, fentry)
# We can then hand `file_entry` off to other functions. # We can then hand `file_entry` off to other functions.
if file_entry.endswith('py'): yield file_entry
num_py_files += 1
print(f"Number of folders: {num_folders}")
print(f"Number of files: {num_files}")
print(f"Number of Python files: {num_py_files}")
if __name__ == "__main__": if __name__ == "__main__":
abspath = os.path.abspath abspath = os.path.abspath
print(f"Listing {abspath('.')}") print(f"Listing {abspath('.')}")
list_directory('.') list_directory('.')
print(f"Recurively counting files in {abspath('../../')}") print(f"\nRecurively counting files in {abspath('../../')}")
iterate_files('../../') num_py_files = 0
for file_entry in iterate_files('../../'):
if file_entry.endswith('.py'):
num_py_files += 1
print(f"\t{num_py_files} python files found "
f"in {abspath('../../')}")