@pysnooper.snoop("test.log", depth=2)

@pysnooper.snoop("test.log", variables=('K', 'P'))

import pysnooper


K = 0


def (i):
    global K
    K += 1
    j = i ** 2
    return j


def g(i):
    global K
    K += 1
    j = i + f(i)
    return j



@pysnooper.snoop("test.main.log", depth=1)
def main():
    for i in range(4):
        print(g(i))


if __name__ == '__main__':
    main()