jeanGaston
9e2f761544
- draw.py/file_io.py/main.py: wire in exclusion pairs so the draw can avoid configured giver/receiver combinations. - utils.py: add get_last_n_years helper. - env.py: load configuration from environment variables (via .env) instead of hardcoding real SMTP/CSV secrets in a tracked file; add .env.example as the template and gitignore .env. - test_draws.py: drop hardcoded personal path, use env.py config instead. - add requirements.txt for the upcoming web GUI/container work. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
54 lines
2.5 KiB
Python
54 lines
2.5 KiB
Python
from random import choice
|
|
|
|
def draw_names(current_participants, history_data, draws_per_person, exclusions, max_attempts=5):
|
|
"""
|
|
Perform the Secret Santa draw considering past years' draws and exclusions.
|
|
:param current_participants: This year's participant list.
|
|
:param history_data: Historical draw data to avoid repeats.
|
|
:param draws_per_person: Number of people each participant should give gifts to.
|
|
:param exclusion_file: Path to the exclusion file.
|
|
:param max_attempts: Maximum number of retry attempts before loosening exclusions.
|
|
:return: The new draw results.
|
|
"""
|
|
participants = [p[0] for p in current_participants] # Get participant names
|
|
already_drawn = [] # Track who has been drawn
|
|
new_draw = [] # Store new draw results
|
|
|
|
for attempt in range(max_attempts):
|
|
print(attempt)
|
|
new_draw.clear()
|
|
already_drawn.clear()
|
|
success = True # Track if draw is successful in this attempt
|
|
|
|
for i, giver in enumerate(participants):
|
|
email = current_participants[i][1]
|
|
# Collect previous recipients to avoid drawing the same person again
|
|
previous_recipients = {recipient for record in history_data if record[0] == giver for recipient in record[2:]}
|
|
|
|
# Create a set of available participants excluding the giver, previous recipients, and exclusions
|
|
available_participants = set(participants) - {giver} - previous_recipients
|
|
available_participants -= {receiver for receiver in participants if (giver, receiver) in exclusions}
|
|
new_recipients = []
|
|
|
|
# Ensure there are enough available participants for the draw
|
|
if len(available_participants) < draws_per_person:
|
|
success = False
|
|
break
|
|
|
|
# Select recipients for the current giver
|
|
while len(new_recipients) < draws_per_person:
|
|
selected = choice(list(available_participants))
|
|
if already_drawn.count(selected) < draws_per_person:
|
|
new_recipients.append(selected)
|
|
already_drawn.append(selected)
|
|
available_participants.discard(selected)
|
|
|
|
new_draw.append([giver, email] + new_recipients)
|
|
|
|
# If the draw was successful, break out of attempts
|
|
if success:
|
|
return new_draw
|
|
print(f"Attempt {attempt + 1} failed, retrying...")
|
|
|
|
# If all attempts fail, raise an error
|
|
raise ValueError("Unable to complete a valid draw after maximum attempts.") |