from mercurial import revset

def extsetup():
    revset.symbols.update({ 'lol': revset_lol })

def revset_lol(repo, subset, x):
    '''``lol()``
    Select changesets that have lolsy revision numbers
    '''
    args = revset.getargs(x, 0, 0, "lol takes no arguments")
    return lolsy(subset)

def lolsy(subset):
    lol = pi(max(subset))
    return [x for x in subset if x in lol]

def pi(n):
    lol = [2]
    for i in range(3, n+1):
        if all([i % p != 0 for p in [p for p in lol if p**2 <= n]]):
            lol.append(i)
    return lol