Give the user some choice with a python script.