Erro ao tentar gravar informações do cProfile no arquivo

Question

Estou tentando carregar um perfil cProfile, fazer algumas classificações e ajustes e, em seguida, enviar os resultados para um arquivo.Com base no documentação, pensei que poderia simplesmente passar um objeto de arquivo e o print_stats função redirecionaria para esse fluxo.

Aqui está o código que estou tentando usar:

import sys,pstats
s = open('output.txt', 'w')
p = pstats.Stats('profile.dat', s)

e aqui está o erro resultante:

TypeError: Cannot create or construct a <class pstats.Stats at 0xbaa870> object from '<open file 'output.txt', mode 'w' at 0xb2ef60>''

Devo acrescentar também que quando não passo um objeto para o parâmetro stream, a saída aparece normalmente no terminal.

Answer 1

Olhando para o código fonte, você teria que passar o arquivo como um stream argumento de palavra-chave (não está claro para mim por que foi implementado assim...), como:

p = pstats.Stats('profile.dat', stream = s)

Veja abaixo o comentário embutido e o if "stream" in kwds linha.

class Stats:
    """..."""
    def __init__(self, *args, **kwds):
        # I can't figure out how to explictly specify a stream keyword arg
        # with *args:
        #   def __init__(self, *args, stream=sys.stdout): ...
        # so I use **kwds and sqauwk if something unexpected is passed in.
        self.stream = sys.stdout
        if "stream" in kwds:
            self.stream = kwds["stream"]
            del kwds["stream"]
        if kwds:
            keys = kwds.keys()
            keys.sort()
            extras = ", ".join(["%s=%s" % (k, kwds[k]) for k in keys])
            raise ValueError, "unrecognized keyword args: %s" % extras
        if not len(args):
            arg = None
        else:
            arg = args[0]
            args = args[1:]
        self.init(arg)
        self.add(*args)