usuarios_por_id.py

"""
    >>> usuarios = [Usuario('usuario %d' % i, i) for i in range(10)]
    >>> pegar_usuarios(usuarios, [1, 2, 3])
    [<Usuario 1>, <Usuario 2>, <Usuario 3>]
"""


class Usuario(object):
    def __init__(self, nome, id):
        self.id = id
        self.nome = nome
    def __repr__(self):
        return '<Usuario %d>' % self.id


def pegar_usuarios(usuarios, ids):
    return [u for u in usuarios if u.id in ids]