Make a vim command act on a visual selection OR the whole buffer

Question

I have this vim function/command that formats json

function! JSON()
  %!python -m json.tool
  setlocal ft=json
endfunction

command! Json call JSON()