"""
Authors:
Consulted:
Date:
Purpose: Looping recursion lab practice.
"""

import os

FAMILY = [
    "Louise",
    ["Richard", ["Sarah"], ["Peter", ["Aaron"]], ["John"], ["Ross"]],
    ["Steven", ["Chris"], ["Michael", ["Mac"], ["Claire"], ["Frances"]]],
    ["Jeannie", ["Anna"], ["Claire"]],
]


def countDescendants(familyTree):
    """
    Returns the total number of people listed in the given tree of
    descendants. Each entry in the tree is a list where the first element
    is a string naming the person. The remaining elements are sub-trees
    for each of that person's children. If someone has no descendants,
    then their list will just have their name.
    """
    count = 1
    for subTree in familyTree[1:]:
        count += countDescendants(subTree)
    return count


print("Should be 2:")
print(countDescendants(FAMILY[1][2]))
print("Should be 6:")
print(countDescendants(FAMILY[1]))
print("Should be 16:")
print(countDescendants(FAMILY))


def revealHiddenFiles(directory):
    """
    Recursively traverses the directories under the given one, printing
    out each hidden file it finds. Hidden files are just files (not
    directories) whose name starts with '.'.

    Directories and files are inspected in alphabetical order.
    """
    # TODO: Implement me


print()
print("Should have one .DS_Store in each directory, plus .numbers:")
revealHiddenFiles('testdir')
print("\nCorrect result:")
print("""\
testdir/.DS_Store
testdir/.hidden/.DS_Store
testdir/.numbers
testdir/pset/.DS_Store
testdir/pset/scene/.DS_Store
testdir/pset/shrub/.DS_Store
testdir/pset/shrub/images/.DS_Store
testdir/remember/.DS_Store""")


def listHiddenFiles(directory):
    """
    Works like revealHiddenFiles except returns a list of file paths,
    rather than printing them.
    """
    # TODO: Implement me


print()
print("Should have one .DS_Store in each directory, plus .numbers:")
print(listHiddenFiles('testdir'))
print("\nCorrect result:")
print(
    "['testdir/.DS_Store', 'testdir/.hidden/.DS_Store', 'testdir/.numbers',"
  + " 'testdir/pset/.DS_Store', 'testdir/pset/scene/.DS_Store',"
  + " 'testdir/pset/shrub/.DS_Store', 'testdir/pset/shrub/images/.DS_Store',"
  + " 'testdir/remember/.DS_Store']"
)