Improve error handling

This commit is contained in:
Edgar P. Burkhart 2024-04-10 13:26:36 +02:00
parent c0b9924ed3
commit 6f2726f423
Signed by: edpibu
GPG Key ID: 9833D3C5A25BD227
2 changed files with 15 additions and 6 deletions

View File

@ -73,13 +73,18 @@ def main():
) )
logger.info("Vérification") logger.info("Vérification")
assert args.lists.is_dir(), f"ROOT_PATH={ROOT_PATH} n'est pas un dossier" if not args.lists.is_dir():
logger.error(f"Le dossier <{args.lists}> n'existe pas.")
return
if not args.blacklists.is_dir(): if not args.blacklists.is_dir():
logger.warning(f"Le dossier <{args.blacklists}> n'existe pas, il va être créé.")
args.blacklists.mkdir() args.blacklists.mkdir()
if not args.results.is_dir(): if not args.results.is_dir():
logger.warning(f"Le dossier <{args.results}> n'existe pas, il va être créé.")
args.results.mkdir() args.results.mkdir()
if args.number < 1:
assert args.number >= 1, "Nombre de musiques < 1" logger.error("Le nombre de musiques est inférieur à 1.")
return
# Lecture des fichiers musique dans ROOT_PATH # Lecture des fichiers musique dans ROOT_PATH
# Faire un dossier différent pour les gens qui ne jouent pas # Faire un dossier différent pour les gens qui ne jouent pas

View File

@ -2,6 +2,7 @@ import logging
import random import random
import re import re
from datetime import datetime from datetime import datetime
import sys
logger = logging.getLogger("musik.list") logger = logging.getLogger("musik.list")
@ -32,9 +33,12 @@ def generate_list(args):
] ]
_musiks = list(filter(lambda _m: _m not in blacklist, _raw_musiks)) _musiks = list(filter(lambda _m: _m not in blacklist, _raw_musiks))
assert ( if len(_musiks) < args.number:
len(_musiks) >= args.number logger.error(
), f"{_u} a {len(_musiks)} musique(s) non black-listée au lieu de {args.number}" f"{_u} a {len(_musiks)} musique(s) non black-listée(s)"
f"au lieu de {args.number}"
)
sys.exit()
logger.debug("Ajout des musiques à la liste") logger.debug("Ajout des musiques à la liste")
musik_list += random.sample(_musiks, args.number) musik_list += random.sample(_musiks, args.number)