how to get the duration of audio python

1import wave
2import contextlib
3fname = '/tmp/test.wav'
4with contextlib.closing(wave.open(fname,'r')) as f:
5    frames = f.getnframes()
6    rate = f.getframerate()
7    duration = frames / float(rate)
8    print(duration)