book_generator/stats_chapitres.py
2024-09-06 11:47:54 +02:00

57 lines
1.8 KiB
Python

import re
from collections import defaultdict
regex_chapitre = r'\*\* (.+)'
fichier_livre = 'livre.org'
# Ouvrir le fichier livre.org
with open("livre.org", "r") as livre:
content = livre.read()
# Définir la fonction pour séparer les mots d'une ligne
def split_words(line):
return re.split('[\s]+', line)
# Initialisation du dictionnaire pour stocker le nombre de mots par chapitre
chapters_word_count = defaultdict(int)
# Parcours des lignes du fichier
# Parcourir chaque ligne du fichier livre.org
chapitre = '(chapitre not found)'
for ligne in content.strip().split("\n"):
# Rechercher le titre du chapitre
match_chapitre = re.search(regex_chapitre, ligne)
if match_chapitre:
chapitre = re.sub( ":title:", "", match_chapitre.group(1))
print(chapitre)
words = split_words(ligne)
chapters_word_count[chapitre] += len(words)
def draw_progress_bar(percent: float, target: int) -> str:
# Calcul du nombre total de caractères nécessaire pour représenter 100%
max_length = len("") * (target / 100)
# Calcul de la longueur de la barre de progression
length = int((max_length * percent) // 100)
# Création de la chaîne représentant la barre de progression
progress_bar = "[" + ("" * int(length)) + ("_" * (int(max_length) - int(length))) + "]"
# Affichage de la barre de progression
# print(f"\rProgression : {percent}%, {progress_bar}", end="\r")
return progress_bar
objectif_mots=500
sum_mots=0
# Afficher le résultat
print("Nombre de mots par chapitre : ")
count_chapitres=0
for count, value in sorted(chapters_word_count.items(), key=lambda item: item[0]):
sum_mots+=value
count_chapitres+=1
print(f"{value} mots \t\t \t{count} \t\t{draw_progress_bar(value, objectif_mots)} \t\t ")
print(f"\n Total : \n\t {sum_mots} mots.\n\t {count_chapitres} chapitres.")