Source code for pycallgraph.output.pickle

import pickle

from .output import Output




class PickleOutput(Output):

    def __init__(self, **kwargs):
        self.fp = None
        self.output_file = 'pycallgraph.pickle'
        Output.__init__(self, **kwargs)

    @classmethod
    def add_arguments(cls, subparsers, parent_parser, usage):
        defaults = cls()

        subparser = subparsers.add_parser(
            'pickle',
            help='Dump to a cPickle file for generation later',
            parents=[parent_parser], usage=usage,
        )

        subparser.add_argument(
            '-o', '--output-file', type=str, default=defaults.output_file,
            help='The generated cPickle file',
        )

        return subparser



    def done(self):
        with open(self.normalize_path(self.output_file), 'wb') as handle:
            pickle.dump(self.processor, handle, pickle.HIGHEST_PROTOCOL)