main.py

from stats import count_word_in_document, count_word_occurences, sort_dict
import sys

def get_book_text(filepath: str) -> str:
	text_content = ""

	with open(filepath) as f:
		text_content = f.read()

	return text_content


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

	path_to_book = sys.argv[1]
	

	print("============ BOOKBOT ============")	
	# return the text content
	text = get_book_text(path_to_book)

	print(f"Analysing book found at {path_to_book}...")
	
	# return number of word in document
	num_words = count_word_in_document(text)
	print("----------- Word Count ----------")
	print(f"Found {num_words} total words")

	# get the number of occurence for each charactere
	chars_dict = count_word_occurences(text)

	# sort them by occurences
	sorted_chars = sort_dict(chars_dict)

	print("--------- Character Count -------")
	for item in sorted_chars:
		print(f"'{item['char']}: {item['num']}'")


if __name__ == "__main__":
	main()