Le script du concours pour le livre

Je me suis donné une heure pour écrire le script qui extraira de la boîte 'concourspython@gmail.com' les 5 gagnants au jeu jeudi prochain.

L'objectif n'était pas ici de faire un script bien léché, réutilisable ou organisé en classes, mais plutôt d'obtenir le résultat rapidement.

Grâce à libgmail et bien sur les modules standards comme email ou random, j'ai pu mettre au point ce script en moins de 70 lignes

Quel bonheur Python

import libgmail
import sys
import random
from email import message_from_string

# réponses bidons bien sûr ;)
results = ['1:a', '2:a', '3:a', '4:a', '5:a', '6:a', '7:a']

def qualifyContent(message_source):
    """ test if the mail wins """
    qualification = [False] * 7
    for line in message_source.strip().split():
        line = line.strip()
        if line in results:
            qualification[results.index(line)] = True
    return False not in qualification

def extractContent(message_source):
    """ get the mail body """
    message = message_from_string(message_source)
    payload = message.get_payload()
    if isinstance(payload, list) and len(payload) > 0:
        # trying first part
        payload = payload[0].get_payload()
    if not isinstance(payload, str):
        return None
    return payload

def getResults(login, password):
    """ retrieve mails from gmail and qualify them
    winners / losers / failures
    """
    ga = libgmail.GmailAccount(login, password)
    ga.login()
    failures = []
    losers = []
    winners = []
    folder = ga.getMessagesByFolder('inbox', allPages=True)
    current = 0
    for thread in folder:
        for msg in thread:
            if not current % 5:
                sys.stdout.write('%d' % current)
            else:
                sys.stdout.write('.')
            sys.stdout.flush()
            message = extractContent(msg.source)
            if message is None:
                failures.append(msg.source)
            elif qualifyContent(message):
                winners.append(msg.sender)
            else:
                losers.append(msg.sender)
            current += 1
    sys.stdout.write('\n')
    return winners, losers, failures, current

winners, losers, failures, num_players = getResults("concourspython@gmail.com",  "secret")
print '%d mails received' % num_players
print '%d winners' % len(winners)
print str(winners)
print '%d losers' % len(losers)
print str(losers)
print '%d failures' % len(failures)
print str(failures)
print 'picking up winners'
random.shuffle(winners)
final_winners = random.sample(winners, 5)
print str(final_winners)