Python decorator with a function that receives a parameter