main.py

import sys
from stats import get_num_words, get_char_distribution, sort_distribution


def get_book_text(path):
    with open(path) as f:
        file_contents = f.read()
        return file_contents


def main(book):
    content = get_book_text(book)
    num_words = get_num_words(content)
    char_distrib = get_char_distribution(content)
    sorted = sort_distribution(char_distrib)
    
    print("============ BOOKBOT ============")
    print(f"Analyzing book found at {book}...")
    print("----------- Word Count ----------")
    print(f"Found {num_words} total words")
    print("--------- Character Count -------")
    for d in sorted:
        if d["char"].isalpha():
            print(f"{d["char"]}: {d["num"]}")
    print("============= END ===============")
    

if len(sys.argv) != 2:
    print("Usage: python3 main.py <path_to_book>")
    sys.exit(1)

book = sys.argv[1]
main(book)