diff --git a/musik/__main__.py b/musik/__main__.py index 17e9c44..7060489 100644 --- a/musik/__main__.py +++ b/musik/__main__.py @@ -73,13 +73,18 @@ def main(): ) 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(): + logger.warning(f"Le dossier <{args.blacklists}> n'existe pas, il va être créé.") args.blacklists.mkdir() if not args.results.is_dir(): + logger.warning(f"Le dossier <{args.results}> n'existe pas, il va être créé.") args.results.mkdir() - - assert args.number >= 1, "Nombre de musiques < 1" + if args.number < 1: + logger.error("Le nombre de musiques est inférieur à 1.") + return # Lecture des fichiers musique dans ROOT_PATH # Faire un dossier différent pour les gens qui ne jouent pas diff --git a/musik/list.py b/musik/list.py index 2cf0a83..166d2a4 100644 --- a/musik/list.py +++ b/musik/list.py @@ -2,6 +2,7 @@ import logging import random import re from datetime import datetime +import sys logger = logging.getLogger("musik.list") @@ -32,9 +33,12 @@ def generate_list(args): ] _musiks = list(filter(lambda _m: _m not in blacklist, _raw_musiks)) - assert ( - len(_musiks) >= args.number - ), f"{_u} a {len(_musiks)} musique(s) non black-listée au lieu de {args.number}" + if len(_musiks) < args.number: + logger.error( + 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") musik_list += random.sample(_musiks, args.number)