How to assign dynamic variables calling from a function in python