Improve error handling
This commit is contained in:
parent
c0b9924ed3
commit
6f2726f423
2 changed files with 15 additions and 6 deletions
|
@ -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
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in a new issue