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")
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

View File

@ -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)