def init_yappi():
    import yappi
    import atexit
    from random import randint
    import os

    BASE_PATH = '/var/www/appserv/log'
    os.makedirs(BASE_PATH, exist_ok=True)

    if 0:
        PATH_STAT = os.path.join(BASE_PATH, str(randint(0, 10000)))
        os.makedirs(PATH_STAT, exist_ok=True)
    else:
        PATH_STAT = BASE_PATH

    print('[YAPPI START] PATH_STAT: {}'.format(PATH_STAT))
    yappi.set_clock_type('wall')
    yappi.start()

    @atexit.register
    def dump_yappi():
        print('[YAPPI STOP]')
        yappi.stop()
        print('[YAPPI WRITE]')
        stats = yappi.get_func_stats()
        # for stat_type in ['pstat', 'callgrind', 'ystat']:
        for stat_type in ['pstat']:
            path = os.path.join(PATH_STAT, 'pants.{}'.format('pstat'))
            print('writing path: {}'.format(path))
            stats.save(path, type='pstat')
        # print('\n[YAPPI FUNC_STATS]')
        # print('writing pants.func_stats')
        # with open(PATH_STAT+'/pants.func_stats', 'w') as fh:
        #     stats.print_all(out=fh)
        # print('\n[YAPPI THREAD_STATS]')
        # print('writing pants.thread_stats')
        # tstats = yappi.get_thread_stats()
        # with open(PATH_STAT+'/pants.thread_stats', 'w') as fh:
        #     tstats.print_all(out=fh)
        print('[YAPPI OUT]')
init_yappi()