102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
#!/usr/bin/python
|
|
|
|
import datetime
|
|
import json
|
|
import random
|
|
import sys
|
|
|
|
annee = datetime.date.today().year
|
|
participants = [
|
|
'Maman',
|
|
'Papa',
|
|
'Mémé',
|
|
'Charlotte',
|
|
'Nico',
|
|
'Didi',
|
|
# 'Johnny',
|
|
'Ben',
|
|
'Ludo',
|
|
]
|
|
|
|
on_evite = {
|
|
'Ben': 'Ludo',
|
|
'Charlotte': 'Nico',
|
|
'Maman': 'Papa',
|
|
# 'Didi': 'Johnny',
|
|
}
|
|
|
|
annees_passees = {}
|
|
|
|
# On charge le résultat de l'année dernière
|
|
ancienne_annee = annee - 1
|
|
while True:
|
|
nom_fichier = f"{ancienne_annee}.json"
|
|
try:
|
|
with open(nom_fichier, "r", encoding="utf-8") as fd:
|
|
annee_passee = json.load(fd)
|
|
print(f"Résultat de l'année {ancienne_annee} chargé depuis le fichier {nom_fichier} :")
|
|
print("\n".join([f"- {offre} => {recois}" for offre, recois in annee_passee.items()]))
|
|
for offre, recois in annee_passee.items():
|
|
annees_passees[offre] = annees_passees.get(offre, []) + [recois]
|
|
ancienne_annee += -1
|
|
except FileNotFoundError:
|
|
print(f"Échec de chargement du tirage de l'an passé depuis le fichier {nom_fichier}.")
|
|
break
|
|
|
|
class EchecTirage(Exception):
|
|
pass
|
|
|
|
# On défine notre méthode de tirage
|
|
def tirage(nb_tentatives_max=99):
|
|
participants_offre = participants.copy()
|
|
participants_recois = participants.copy()
|
|
|
|
# On mélange les listes
|
|
random.shuffle(participants_offre)
|
|
random.shuffle(participants_recois)
|
|
|
|
result = dict()
|
|
for offre in participants_offre:
|
|
count = 0
|
|
while True:
|
|
count += 1
|
|
if count == nb_tentatives_max:
|
|
raise EchecTirage
|
|
recois = random.choice(participants_recois)
|
|
# Pas à soi même
|
|
if recois == offre:
|
|
continue
|
|
# On évite entre couple
|
|
if on_evite.get(offre) == recois or on_evite.get(recois) == offre:
|
|
continue
|
|
# On évite de faire comme les années passées
|
|
if recois in annees_passees.get(offre, []):
|
|
continue
|
|
# Trouvé !
|
|
result[offre] = recois
|
|
participants_recois.remove(recois)
|
|
break
|
|
return result
|
|
|
|
# On procède au tirage
|
|
print("On procède au tirage... Suspense !")
|
|
result = None
|
|
while not result:
|
|
try:
|
|
result = tirage()
|
|
except EchecTirage:
|
|
print("Échec du tirage, on recommence !")
|
|
result = None
|
|
|
|
# On affiche le résultat
|
|
print(f"Résultat du tirage pour l'année {annee} :")
|
|
for offre, recois in result.items():
|
|
print(f" - {offre} -> {recois}")
|
|
|
|
# On enregistre le tirage pour l'année passée
|
|
nom_fichier = f"{annee}.json"
|
|
with open(nom_fichier, "w", encoding="utf-8") as fichier:
|
|
fichier.write(json.dumps(result, indent=2, ensure_ascii=False))
|
|
print(f"Résultat enregistré dans le fichier {nom_fichier}.")
|
|
|
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4 expandtab
|