bookbot/main.py

from stats import count_words
from stats import count_characters
from stats import sort_by_count

import sys

def get_book_text(filepath):
    with open(filepath) as f:
        return f.read()

def main():
    if len(sys.argv) < 2:
        print("Usage: python3 main.py <path_to_book>")
        sys.exit(1)
    
    read_file = get_book_text(sys.argv[1])
    counted = count_words(read_file)
    counted_chars = count_characters(read_file)
    

    print("============ BOOKBOT ============")
    print("Analyzing book found at books/frankenstein.txt...\n----------- Word Count ----------")
    print(f"Found {counted} total words")
    print("--------- Character Count -------")
    
    sorted = sort_by_count(counted_chars)
    for item in sorted:
        if item["char"].isalpha():
            print(f"{item["char"]}: {item["num"]}")

    print("============= END ===============")

main()