Instead of trying to keep it as a script, make it a module and make the script a small wrapper around the module. Then you can use the same code from many places.